Update app.py

app.py

@@ -440,6 +440,52 @@ def solve_directly_with_python(question: str) -> str | None:
     return None
 
 
+def solve_commutativity_table(question: str) -> str | None:
+    q = question.lower()
+
+    if "|---" not in question:
+        return None
+
+    if "commutative" not in q and "commutativity" not in q:
+        return None
+
+    lines = [
+        line.strip()
+        for line in question.splitlines()
+        if line.strip().startswith("|")
+    ]
+
+    if len(lines) < 3:
+        return None
+
+    header = [x.strip() for x in lines[0].strip("|").split("|")]
+    cols = header[1:]
+
+    table = {}
+
+    for line in lines[2:]:
+        cells = [x.strip() for x in line.strip("|").split("|")]
+        if len(cells) != len(cols) + 1:
+            continue
+
+        row = cells[0]
+        values = cells[1:]
+        table[row] = dict(zip(cols, values))
+
+    for a in cols:
+        for b in cols:
+            if a == b:
+                continue
+
+            ab = table.get(a, {}).get(b)
+            ba = table.get(b, {}).get(a)
+
+            if ab is not None and ba is not None and ab != ba:
+                return ", ".join(sorted([a, b]))
+
+    return "commutative"
+
+
 def direct_question(question: str) -> bool:
     q = question.lower()
     rev = q[::-1]
@@ -620,20 +666,29 @@ class BasicAgent:
         )
         return {"context": context, "raw_answer": raw_answer}
 
-    def solve_spreadsheet(self, state: AgentState) -> dict[str, Any]:
-        question = state.get("question", "")
-        local_path = state.get("local_path")
+    def solve_spreadsheet(self, state: AgentState) -> dict:
+        path = state["local_path"]
+        question = state["question"]
 
-        if not local_path:
-            return {"raw_answer": "ERROR: spreadsheet route selected but no local file path found"}
+        xls = pd.ExcelFile(path)
+        parts = []
+
+        for sheet in xls.sheet_names:
+            df = pd.read_excel(path, sheet_name=sheet)
+            parts.append(f"Sheet: {sheet}")
+            parts.append(f"Columns: {list(df.columns)}")
+            parts.append(f"Shape: {df.shape}")
+            parts.append(df.head(20).to_csv(index=False))
+
+        context = "\n\n".join(parts)
 
-        context = read_spreadsheet_context(Path(local_path))
         raw_answer = self.answer_from_context(
             question=question,
-            context=context,
-            context_label="Spreadsheet contents",
+            context=context[:12000],
+            context_label="spreadsheet preview",
             llm=self.answer_llm,
         )
+
         return {"context": context, "raw_answer": raw_answer}
 
     def solve_code(self, state: AgentState) -> dict[str, Any]:
@@ -671,6 +726,10 @@ class BasicAgent:
         shortcut = solve_directly_with_python(question)
         if shortcut is not None:
             return {"context": "Solved by deterministic Python shortcut.", "raw_answer": shortcut}
+        
+        direct = solve_commutativity_table(question)
+        if direct is not None:
+            return {"raw_answer": direct}
 
         context = ""
         if local_path and file_kind in {"pdf", "text", "binary"}: