Update app.py

app.py

@@ -6,9 +6,13 @@ import pandas as pd
 import logging
 from typing import Optional, List
 from smolagents import (
+                MCPClient,
                 CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, GoogleSearchTool,
                 VisitWebpageTool,
                 LiteLLMModel,AzureOpenAIServerModel, WikipediaSearchTool,PythonInterpreterTool)
+from mcp import StdioServerParameters
+
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -64,10 +68,26 @@ class BasicAgent:
             api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
             api_version="2024-12-01-preview"    #2024-02-15-preview
         )
-        
-        # Initialize tools
+
+        server_parameters = StdioServerParameters(
+                        command="uvx",  # Using uvx ensures dependencies are available
+                        args=["--quiet", "pubmedmcp@0.1.3"],
+                        env={"UV_PYTHON": "3.12", **os.environ},
+                        )
+    
+    
+        try:
+            self.mcp_client = MCPClient(server_parameters)  
+            # tools = mcp_client.get_tools()
+            # agent = CodeAgent(tools=tools, model=model, add_base_tools=True)
+            # agent.run("Donner la definition de mesure pour un etudiant de licence 3 en mathematique")
+        except:
+            print("🥵 🥵erreur lors de la connexion au serveur mcp")
+            
+            # Initialize tools
         self.tools = [
             WikipediaSearchTool(),
+            self.mcp_client.get_tools()
                         # DuckDuckGoSearchTool(),
             # PythonInterpreterTool(),
                       # VisitWebpageTool()
@@ -192,6 +212,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print("Agent did not produce any answers to submit.")
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
+    # se deconnecter du serveur mcp
+    self.mcp_client.disconnect()
     # 4. Prepare Submission 
     submission_data = {"username": username.strip(), "agent_code": agent_code, "answers": answers_payload}
     status_update = f"Agent finished. Submitting {len(answers_payload)} answers for user '{username}'..."