Update agent.py

agent.py

@@ -1,4 +1,4 @@
-# agent_v29.py
+# agent_v30.py
 import os
 import re
 import io
@@ -28,7 +28,7 @@ class GaiaAgent:
         response = self.client.chat.completions.create(
             model=model,
             messages=[
-                {"role": "system", "content": "You are a precise assistant. Return only a short factual answer. Format appropriately. Never guess."},
+                {"role": "system", "content": "Return only a short factual answer. Format it properly. Never guess."},
                 {"role": "user", "content": prompt.strip() + "\nAnswer:"}
             ],
             temperature=0.0,
@@ -51,7 +51,7 @@ class GaiaAgent:
     def ask_image(self, image_bytes, question):
         image_b64 = base64.b64encode(image_bytes).decode("utf-8")
         messages = [
-            {"role": "system", "content": "Answer with only the correct chess move in algebraic notation."},
+            {"role": "system", "content": "Return only the winning move in chess algebraic notation (e.g., Qd1). No explanation."},
             {
                 "role": "user",
                 "content": [
@@ -68,9 +68,10 @@ class GaiaAgent:
             df = pd.read_excel(io.BytesIO(file_bytes), engine="openpyxl")
             df.columns = [col.lower() for col in df.columns]
             if 'category' in df.columns and 'sales' in df.columns:
-                food_df = df[df['category'].str.contains('food', case=False)]
+                df['sales'] = pd.to_numeric(df['sales'], errors='coerce')
+                food_df = df[df['category'].str.lower().str.contains('food')]
                 total = food_df['sales'].sum()
-                return f"${total:.2f}"
+                return f"${total:.2f}" if not pd.isna(total) else "$0.00"
         except Exception:
             pass
         return "$0.00"
@@ -78,6 +79,7 @@ class GaiaAgent:
     def extract_answer(self, raw, question):
         q = question.lower()
         raw = raw.strip().strip("\"'").strip()
+        raw = re.sub(r"^[-•\s]*", "", raw)
 
         if "studio albums" in q:
             try:
@@ -91,8 +93,9 @@ class GaiaAgent:
             return match.group(1) if match else raw
 
         if "vegetables" in q or "ingredients" in q:
-            list_raw = re.findall(r"[a-zA-Z]+(?: [a-zA-Z]+)?", raw)
-            return ", ".join(sorted(set(i.lower() for i in list_raw)))
+            unwanted = {"pure", "extract", "granulated", "sugar", "juice", "vanilla", "ripe", "fresh", "whole", "bean", "pinch", "cups", "salt", "water"}
+            terms = [t.lower() for t in re.findall(r"[a-zA-Z]+(?: [a-zA-Z]+)?", raw)]
+            return ", ".join(sorted(set(t for t in terms if t.split()[0] not in unwanted)))
 
         if "usd with two decimal places" in q:
             match = re.search(r"\$?([0-9]+(?:\.[0-9]{1,2})?)", raw)
@@ -132,12 +135,11 @@ class GaiaAgent:
                     return self.ask(f"Video summary: {summary}\n\n{question}")
 
             if "malko competition" in question.lower():
-                search = self.get_web_info("list of Malko Competition winners after 1977 and their nationalities")
+                search = self.get_web_info("malko competition winners after 1977 yugoslavia site:wikipedia.org")
                 return self.ask(f"Web result: {search}\n\n{question}")
 
             if "commutative" in question:
-                table_text = question.strip()
-                return self.ask(f"Analyze the following table for non-commutative pairs:\n{table_text}\nList only the elements involved in alphabetical order, comma separated.")
+                return self.ask(f"Based on this table, which elements show the operation is not commutative?\n{question}\nList them comma-separated, alphabetically.")
 
             if file_bytes and "image" in ctype:
                 raw = self.ask_image(file_bytes, question)
@@ -156,4 +158,4 @@ class GaiaAgent:
         except Exception as e:
             return f"[ERROR: {e}]"
 
-        return self.extract_answer(raw, question)
+        return self.extract_answer(raw, question)