Update app.py

app.py

@@ -100,8 +100,19 @@ Always end with just the direct answer the question asks for."""
         
         try:
             # Let the CodeAgent handle everything with ReAct pattern
-            # The LLM is smart enough to format answers properly for GAIA
-            result = self.agent.run(question)
+            # We'll include formatting instructions in our prompt
+            formatted_question = f"""Please answer this question with just the direct answer needed (no prefixes like "The answer is"):
+
+{question}
+
+Important: 
+- For numbers, provide just the number
+- For names/words, provide just the name/word  
+- For yes/no questions, answer just "Yes" or "No"
+- Use web search when you need current information
+- Write Python code for calculations or data processing"""
+
+            result = self.agent.run(formatted_question)
             
             final_answer = str(result).strip()
             print(f"Agent answer: {final_answer}")
@@ -118,6 +129,7 @@ Always end with just the direct answer the question asks for."""
         print("Smolagents not available - using basic fallback")
         return "Smolagents required but not available"
 
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,