Update app.py

app.py

@@ -8,8 +8,10 @@ import pandas as pd
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+import os
 from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool, PythonInterpreterTool, SpeechToTextTool #, ImageCaptioningTool
 
 class BasicAgent():
@@ -18,7 +20,7 @@ class BasicAgent():
         
         # Configuration du modèle Azure GPT-4o-mini
         self.model = LiteLLMModel(
-            model_id="azure/gpt-4o-mini",
+            model_id="azure/gpt-4o",
             api_key=os.getenv("api_key_4o"),
             api_base=os.getenv("base_url_4o"),
             api_version="2025-01-01-preview"
@@ -28,7 +30,7 @@ class BasicAgent():
         tools = [
             DuckDuckGoSearchTool(),          # Recherche web (Wikipedia)
             VisitWebpageTool(),              # Visite de pages web
-            PythonInterpreterTool(),         # Calculs, Excel, traitement de données
+            PythonInterpreterTool(),                # Calculs, Excel, traitement de données
             SpeechToTextTool(),              # Speech to text
             #ImageCaptioningTool(),           # Reconnaissance d'image
         ]
@@ -41,13 +43,27 @@ class BasicAgent():
         
         print("Alfred agent ready with tools!")
     
+    def extract_final_answer(self, response):
+        """Extrait la réponse finale du response de l'agent"""
+        if isinstance(response, str):
+            # Si c'est déjà une string, la retourner directement
+            return response.strip()
+        
+        # Si c'est un objet avec des attributs, essayer d'extraire le contenu
+        if hasattr(response, 'content'):
+            return str(response.content).strip()
+        elif hasattr(response, 'text'):
+            return str(response.text).strip()
+        else:
+            # Convertir en string par défaut
+            return str(response).strip()
+    
     def __call__(self, question: str) -> str:
         print(f"Alfred received question (first 50 chars): {question[:50]}...")
         
         try:
             prompt = f"""
             You are Alfred, an intelligent assistant with access to multiple tools.
-
             To answer questions:
             1. ANALYZE the question to identify the type of task.
             2. USE the appropriate tool:
@@ -82,9 +98,10 @@ class BasicAgent():
             
         except Exception as e:
             print(f"Erreur lors du traitement: {e}")
-            return e
+            return "Je ne peux pas répondre à cette question."
     
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,