Update app.py

app.py

@@ -66,33 +66,53 @@ class BasicAgent:
         message=f"""
             Solve this GAIA benchmark task carefully.
 
-            Rules:
-            - Use web search only when necessary.
-            - Use python if useful.
-            - Use read_files if an attachment exists.
-            - Think step by step internally.
-            - NEVER reveal reasoning.
-            - NEVER output thoughts.
-            - NEVER explain your answer.
-            - Return ONLY the exact final answer.
-            - If numberic, return only the number.
-            - If text, return only the text.
-            - If comma-separated list requested, output ONLY the list.
-            - If unsure, still attempt the best possible answer.
-            - Never say "I don't know".
-            - Never refuse.
+            CRITICAL RULES:
+            1. Return ONLY the exact answer.
+            2. Never explain reasoning.
+            3. Never write thoughts.
+            4. Never say "After searching"
+            5. Never include "Final Answer:"
+            6. Never output sentences unless requested.
+            7. If they ask for:
+                -a number: return only the number
+                -first name: return only first name
+                -surname: return only surname
+                -chess move: return only algebraic notation
+                -comma-separated list: only the list
+            8. If an attachment exists, ALWAYS read it.
+            9. Use web search for factual questions.
+            10. If uncertain, still give best concise answer.
+            11. NEVER include any extra text before or after the answer.
+
+            BAD:
+            "The answer is 485"
+
+            Good:
+            485
+            
             Question: {question}
             Attachment path: {file_path}
         """
         result = self.agent.run(message)
         if result is None:
             return ""
+            
         result = str(result).strip()
+        
         if "Final Answer" in result:
             result = result.split("Final Answer:")[-1].strip()
-        if "Thoughts" in result:
+
+        if "Here is the final answer" in result:
             lines = result.splitlines()
             result = lines[-1].strip()
+        
+        if "Thoughts:" in result:
+            result = result.split("Thoughts:")[-1].strip()
+
+
+
+        result = result.split("\n")[0].strip()
+        result = result.strip('"').strip("'")
         return result
     
 def run_and_submit_all( profile: gr.OAuthProfile | None):