Update app.py

app.py

@@ -18,6 +18,7 @@ class BasicAgent:
     def __init__(self):
         from smolagents import CodeAgent, LiteLLMModel, DuckDuckGoSearchTool, PythonInterpreterTool
         import os
+
         model = LiteLLMModel(
             model_id="groq/llama-3.3-70b-versatile",
             api_key=os.environ["GROQ_API_KEY"],
@@ -25,17 +26,36 @@ class BasicAgent:
         self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool(), PythonInterpreterTool()],
             model=model,
-            max_steps=8,
+            max_steps=6,
         )
 
+    def _clean(self, text: str) -> str:
+        text = str(text).strip()
+        for prefix in ["FINAL ANSWER:", "Final answer:", "Answer:", "The answer is"]:
+            if text.startswith(prefix):
+                text = text[len(prefix):].strip()
+        return text.strip().strip('"').rstrip(".")
+
     def __call__(self, question: str) -> str:
-        try:
-            answer = self.agent.run(
-                f"Answer with ONLY the final answer, nothing else.\n\nQuestion: {question}"
-            )
-            return str(answer).strip().rstrip(".")
-        except Exception:
-            return "unknown"
+        import traceback, time
+        prompt = (
+            "You are answering a benchmark question. Your response is graded by EXACT string match.\n"
+            "Output ONLY the answer itself: no explanation, no sentence, no units unless the question asks for them, "
+            "no trailing period, no quotes.\n"
+            "If the answer is a number, write just the number. If it is a name, write just the name.\n\n"
+            f"Question: {question}"
+        )
+        for attempt in range(2):
+            try:
+                answer = self.agent.run(prompt)
+                cleaned = self._clean(answer)
+                if cleaned:
+                    return cleaned
+            except Exception as e:
+                print(f"[AGENT ERROR] attempt {attempt+1} | {type(e).__name__}: {e}")
+                traceback.print_exc()
+                time.sleep(3)
+        return "unknown"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """