Update app.py

app.py

@@ -18,37 +18,37 @@ from openai import OpenAI
 
 class BasicAgent:
     def __init__(self):
-        print("Tool-based Agent initialized")
+        print("GAIA Smart Agent initialized")
         self.client = OpenAI()
 
-    # Simple safe calculator
-    def calculate(self, expression):
-        try:
-            return eval(expression)
-        except:
-            return None
-
     def clean_answer(self, text):
+        """
+        GAIA scoring requires exact answers.
+        This removes explanations and keeps only the final value.
+        """
 
         text = text.strip()
 
+        # remove phrases
         text = re.sub(r"(?i)final answer[:\- ]*", "", text)
         text = re.sub(r"(?i)answer[:\- ]*", "", text)
 
+        # keep only last line
+        text = text.split("\n")[-1]
+
         return text.strip()
 
     def ask_llm(self, question):
 
         prompt = f"""
-You are solving a GAIA benchmark task.
-
-You can:
-- reason step by step
-- perform calculations if needed
+Solve the question carefully.
 
-If a math expression appears, write it clearly.
+Think step by step internally but output ONLY the final answer.
 
-Return ONLY the final answer.
+Rules:
+- No explanation
+- No sentences
+- Only the final value
 
 Question:
 {question}
@@ -56,45 +56,26 @@ Question:
 
         response = self.client.chat.completions.create(
             model="gpt-4o-mini",
-            messages=[{"role": "user", "content": prompt}],
+            messages=[
+                {"role": "system", "content": "You are a precise problem solver."},
+                {"role": "user", "content": prompt}
+            ],
             temperature=0
         )
 
         return response.choices[0].message.content.strip()
 
-    def try_calculation(self, question):
-
-        # find simple math expressions
-        match = re.findall(r"[\d\.\+\-\*\/\(\) ]+", question)
-
-        for m in match:
-            m = m.strip()
-            if len(m) > 2:
-                result = self.calculate(m)
-                if result is not None:
-                    return result
-
-        return None
-
     def __call__(self, question: str) -> str:
 
         print("Question:", question)
 
-        # try calculator first
-        calc_result = self.try_calculation(question)
-
-        if calc_result is not None:
-            print("Used calculator:", calc_result)
-            return str(calc_result)
-
-        # otherwise ask LLM
-        answer = self.ask_llm(question)
+        raw_answer = self.ask_llm(question)
 
-        answer = self.clean_answer(answer)
+        final_answer = self.clean_answer(raw_answer)
 
-        print("Final answer:", answer)
+        print("Final Answer:", final_answer)
 
-        return answer
+        return final_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """