update

app.py

@@ -6,7 +6,7 @@ from gradio import ChatMessage
 from smolagents.gradio_ui import stream_to_gradio
 
 from agents.all_agents import get_master_agent
-from llm import get_default_model
+from llm import ANTHROPIC_MODEL_IDS, get_anthropic_model
 
 
 gr.set_static_paths(paths=["images/"])
@@ -23,8 +23,9 @@ def resize_image(image):
     return image
 
 
-def chat_interface_fn(input_request, history: List[ChatMessage], gallery):
-    agent = get_master_agent(get_default_model())
+def chat_interface_fn(input_request, history: List[ChatMessage], gallery, anthropic_api_key, anthropic_model_id):
+    model = get_anthropic_model(anthropic_model_id, anthropic_api_key)
+    agent = get_master_agent(model)
     if gallery is None:
         gallery = []
     else:
@@ -123,6 +124,26 @@ 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>
+        Don't hesitate to like this other Space or reach out to me on <a href="https://www.linkedin.com/in/sbucaille/">LinkedIn</a> if you have any questions or feedback!<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 +151,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