Smol adaptations to avoid reloading the server

app.py

@@ -2,26 +2,32 @@ import gradio as gr
 import os
 
 from mcp import StdioServerParameters
-from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
+from smolagents import InferenceClientModel, CodeAgent, MCPClient
 
+HF_MCP_URL = "https://alihmaou-mcp-tools.hf.space/gradio_api/mcp/sse"
 
-try:
-    mcp_client = MCPClient(
-        {"url": "https://alihmaou-mcp-tools.hf.space/gradio_api/mcp/sse"} # This is the MCP Server we created in the previous section
-    )
-    tools = mcp_client.get_tools()
+mcp_client = MCPClient({"url": HF_MCP_URL})
+model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
+tools = mcp_client.get_tools()
+agent = CodeAgent(tools=tools, model=model)
+
+def run_agent(message, history):
+    return str(agent.run(message))
 
-    model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
-    agent = CodeAgent(tools=[*tools], model=model)
+def reload_tools():
+    global tools, agent
+    tools = mcp_client.get_tools()
+    agent = CodeAgent(tools=tools, model=model)
+    return "\n".join(f"- **{t.name}**: {t.description}" for t in tools)
 
-    demo = gr.ChatInterface(
-        fn=lambda message, history: str(agent.run(message)),
-        type="messages",
-        examples=["Combien de nombres premiers entre 0 et 500 ?"],
-        title="MCP Tools tester",
-        description="MCP tools tester connected to https://alihmaou-mcp-tools.hf.space/gradio_api/mcp/sse",
-    )
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column(scale=2):
+            chatbot = gr.ChatInterface(fn=run_agent)
+        with gr.Column(scale=1):
+            tool_list = gr.Markdown(value=reload_tools())
+            reload_btn = gr.Button("🔄 Recharger les outils MCP")
+            reload_btn.click(fn=reload_tools, outputs=tool_list)
 
-    demo.launch()
-finally:
-    mcp_client.disconnect()
+demo.launch()
+mcp_client.disconnect()