Update agent.py

agent.py

@@ -1,4 +1,4 @@
-# agent_v34.py (wersja oparta na stabilnym V26 + precyzyjne poprawki logiczne)
+# agent_v35.py (logika komutatywna, warunkowe filtrowanie, poprawka chess image)
 import os
 import re
 import io
@@ -75,9 +75,11 @@ class GaiaAgent:
             try:
                 df = pd.read_excel(io.BytesIO(content), engine="openpyxl")
                 df.columns = [c.lower() for c in df.columns]
-                df['sales'] = pd.to_numeric(df['sales'], errors='coerce')
-                df = df[df['category'].str.lower() == 'food']
-                return f"${df['sales'].sum():.2f}"
+                if 'category' in df.columns and 'sales' in df.columns:
+                    df['sales'] = pd.to_numeric(df['sales'], errors='coerce')
+                    food_df = df[df['category'].str.lower() == 'food']
+                    return f"${food_df['sales'].sum():.2f}"
+                return "[MISSING REQUIRED COLUMNS]"
             except:
                 return "$0.00"
         try:
@@ -85,6 +87,28 @@ class GaiaAgent:
         except:
             return ""
 
+    def handle_commutativity(self, question):
+        try:
+            table = {}
+            lines = question.splitlines()
+            header = []
+            for line in lines:
+                if line.strip().startswith("|*"):
+                    header = line.strip().split("|")[2:]
+                elif line.strip().startswith("|") and line.count("|") >= 6:
+                    parts = line.strip().split("|")[1:-1]
+                    key, values = parts[0], parts[1:]
+                    table[key] = values
+            S = list(table.keys())
+            non_commutative = set()
+            for i in S:
+                for j in S:
+                    if table[i][S.index(j)] != table[j][S.index(i)]:
+                        non_commutative.update([i, j])
+            return ", ".join(sorted(non_commutative))
+        except:
+            return ""
+
     def format_answer(self, raw, question):
         q = question.lower()
         raw = raw.strip().strip("\"'")
@@ -100,8 +124,7 @@ class GaiaAgent:
             return ", ".join(items)
 
         if "commutative" in q:
-            items = sorted(set(re.findall(r"[abcde]", raw)))
-            return ", ".join(items)
+            return self.handle_commutativity(question)
 
         if "first name" in q:
             return raw.split()[0]
@@ -136,9 +159,9 @@ class GaiaAgent:
         file_bytes, file_type = (None, None)
         if task_id:
             file_bytes, file_type = self.fetch_file(task_id)
+
         context = self.handle_file(file_bytes, file_type, question) if file_bytes else self.extract_web_context(question)
 
-        # fallback: use direct search prompt
         if not context.strip():
             prompt_map = {
                 "youtube": "transcript of video site:youtube.com",