Update app and llm

app.py

@@ -123,6 +123,25 @@ with gr.Blocks() as demo:
         </div>
         """,
     )
+    gr.Markdown(
+        """
+        ## Update 17/06/2025
+        This Space was originally a Hackathon submission, funded with Anthropic Free Credits.<br>
+        Due to the high popularity of the Space, unfortunately I can't fund personally the credits anymore.<br>
+        I have added below the ability to add your own Anthropic API Key and select the model to use.<br>
+        """
+    )
+    anthropic_api_key = gr.Textbox(label="Anthropic API Key")
+    anthropic_model_id = gr.Dropdown(label="Anthropic Model", choices=ANTHROPIC_MODEL_IDS)
+    gr.Markdown(
+        """
+        ## Future plans
+        I plan to continue developing this Space on a more personal space here : https://huggingface.co/spaces/stevenbucaille/ScouterAI <br>
+        This Space will be powered with ZeroGPU and have more LLM options.<br>
+        Stay tuned!
+        <br>
+        """
+    )
     output_gallery = gr.Gallery(label="Images generated by the agent (do not put images)", type="pil", format="png")
     textbox = gr.MultimodalTextbox()
     gr.ChatInterface(
@@ -130,7 +149,7 @@ with gr.Blocks() as demo:
         type="messages",
         multimodal=True,
         textbox=textbox,
-        additional_inputs=[output_gallery],
+        additional_inputs=[output_gallery, anthropic_api_key, anthropic_model_id],
         additional_outputs=[output_gallery],
     )
 

llm.py

@@ -1,25 +1,24 @@
-import os
-
 from smolagents import LiteLLMModel
 
 
-LOCAL_LLM_SETTINGS = {
-    "api_base": "http://127.0.0.1:1234/v1",
-    "api_key": "api-key",
-    "model_id": "gemma-3-12b-it-qat",
-}
-
-ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
-
-assert ANTHROPIC_API_KEY is not None, "ANTHROPIC_API_KEY is not set"
+ANTHROPIC_MODEL_IDS = [
+    "claude-opus-4-20250514",
+    "claude-sonnet-4-20250514",
+    "claude-3-7-sonnet-latest",
+    "claude-3-5-haiku-latest",
+    "claude-3-5-sonnet-latest",
+    "claude-3-5-sonnet-20240620",
+    "claude-3-opus-latest",
+    "claude-3-sonnet-20240229",
+    "claude-3-haiku-20240307",
+]
 
 
-def get_default_model():
+def get_anthropic_model(model_id, anthropic_api_key):
+    if model_id not in ANTHROPIC_MODEL_IDS:
+        raise ValueError(f"Model {model_id} not found in Anthropic model IDs")
     model = LiteLLMModel(
-        # model_id="claude-3-7-sonnet-20250219",
-        model_id="claude-3-5-haiku-latest",
-        api_key=os.getenv("ANTHROPIC_API_KEY"),
-        # reasoning_effort="low",
+        model_id=model_id,
+        api_key=anthropic_api_key,
     )
-    print("Loaded LLM model")
     return model