Update app.py

app.py

@@ -11,6 +11,8 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, VisitWebpageTool, PythonCodeTool, SpeechToTextTool, ImageCaptioningTool
+
 class BasicAgent():
     def __init__(self):
         print("MyCustomAgent with SmolaAgent initialized.")
@@ -24,46 +26,65 @@ class BasicAgent():
         )
         
         # Outils disponibles
-        search_tool = DuckDuckGoSearchTool()
-        webpage_tool = VisitWebpageTool()
+        tools = [
+            DuckDuckGoSearchTool(),          # Recherche web (Wikipedia)
+            VisitWebpageTool(),              # Visite de pages web
+            PythonCodeTool(),                # Calculs, Excel, traitement de données
+            SpeechToTextTool(),              # Speech to text
+            ImageCaptioningTool(),           # Reconnaissance d'image
+        ]
         
         # Créer l'agent Alfred avec les outils
         self.alfred = CodeAgent(
-            tools=[search_tool, webpage_tool], 
-            model=self.model,
-            additional_authorized_imports=["pandas", "numpy"],
-            max_steps = 15
+            tools=tools, 
+            model=self.model
         )
         
-        print("Alfred agent ready with search, webpage, and Python tools!")
+        print("Alfred agent ready with tools!")
     
     def __call__(self, question: str) -> str:
-        """
-        Permet d'appeler l'agent directement avec une question
-        """
+        print(f"Alfred received question (first 50 chars): {question[:50]}...")
+        
         try:
-            # Ajouter des instructions dans la question pour améliorer les réponses
-            enhanced_question = f"""
-            {question}
+            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:
+                - For mathematical calculations: use PythonCodeTool
+                - To read Excel/CSV files: use PythonCodeTool with pandas
+                - For information retrieval: use DuckDuckGoSearchTool, then VisitWebpageTool if needed
+                - To analyze images: use ImageCaptioningTool
+                - For speech-to-text: use SpeechToTextTool
+                - For data processing: use PythonCodeTool
+                - RETURN only the final answer, clearly and concisely.
+            
+            If you cannot answer, simply say: "I cannot answer this question."
             
-            Instructions:
-            - If this involves data analysis, use Python code to calculate results
-            - If data files are mentioned try access it
-            - Search for additional context if needed
-            """
+            Example answers:
+                - For a calculation: "42"
+                - For a date: "March 15, 2023"
+                - For a name: "Albert Einstein"
+                - For a number: "1,234,567"
+                - For a list: "b, e"
+            
+            Here is the question to process: {question}"""
+
+            # Utiliser Alfred pour traiter la question
+            response = self.alfred.run(prompt)
+
+            # IMPORTANT: Extraire seulement la réponse finale
+            # Pas de "FINAL ANSWER" ou formatage supplémentaire
+            final_answer = self.extract_final_answer(response)
+            
+            print(f"Alfred returning answer: {final_answer}")
+            return final_answer
             
-            result = self.alfred.run(enhanced_question)
-            return str(result)
         except Exception as e:
-            if "content filtering" in str(e).lower() or "content policy" in str(e).lower():
-                # Essayer avec une reformulation plus neutre
-                neutral_question = f"Please analyze and provide information about: {question}"
-                try:
-                    result = self.alfred.run(neutral_question)
-                    return str(result)
-                except:
-                    return f"Unable to process this question due to content restrictions. Please rephrase."
-            return f"Erreur lors du traitement de la question: {str(e)}"
+            print(f"Erreur lors du traitement: {e}")
+            return "Je ne peux pas répondre à cette question."
+    
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """