Update agent.py

agent.py

@@ -90,6 +90,25 @@ def answer_question(question: str, task_id: str = None) -> str:
     try:
         import re
         import pandas as pd
+        table_lines = [line.strip() for line in file_context.splitlines() if '|' in line and '*' not in line[:2]]
+        headers = re.split(r'\|+', table_lines[0])[1:-1]
+        data_rows = [re.split(r'\|+', row)[1:-1] for row in table_lines[1:]]
+        index = [row[0] for row in data_rows]
+        matrix = [row[1:] for row in data_rows]
+        df = pd.DataFrame(matrix, index=index, columns=headers)
+        non_comm = set()
+        for a in df.index:
+            for b in df.columns:
+                if df.at[a, b] != df.at[b, a]:
+                    non_comm.add(a)
+                    non_comm.add(b)
+        result = ", ".join(sorted(non_comm))
+        file_context += f" [Parsed Non-Commutative Set] {result}"
+    except Exception as e:
+        file_context += f" [Table Parse Error] {e}"
+    try:
+        import re
+        import pandas as pd
 
         table_lines = [line.strip() for line in file_context.splitlines() if '|' in line and '*' not in line[:2]]
         headers = re.split(r'\|+', table_lines[0])[1:-1]
@@ -105,9 +124,11 @@ def answer_question(question: str, task_id: str = None) -> str:
                     non_comm.add(a)
                     non_comm.add(b)
         result = ", ".join(sorted(non_comm))
-        file_context += f"\n[Parsed Non-Commutative Set] {result}"
+        file_context += f"
+[Parsed Non-Commutative Set] {result}"
     except Exception as e:
-        file_context += f"\n[Table Parse Error] {e}"
+        file_context += f"
+[Table Parse Error] {e}"
     # Dynamic parser for operation table
     import re
     import pandas as pd
@@ -127,9 +148,11 @@ def answer_question(question: str, task_id: str = None) -> str:
                     non_comm.add(a)
                     non_comm.add(b)
         result = ", ".join(sorted(non_comm))
-        file_context += f"[Parsed Non-Commutative Set] {result}"
+        file_context += f"
+[Parsed Non-Commutative Set] {result}"
     except Exception as e:
-        file_context += f"[Table Parse Error] {e}"
+        file_context += f"
+[Table Parse Error] {e}"
 # Parse table to extract non-commutative elements
     import re
     import pandas as pd
@@ -149,6 +172,8 @@ def answer_question(question: str, task_id: str = None) -> str:
                     non_comm.add(a)
                     non_comm.add(b)
         result = ", ".join(sorted(non_comm))
-        file_context += f"[Parsed Non-Commutative Set] {result}"
+        file_context += f"
+[Parsed Non-Commutative Set] {result}"
     except Exception as e:
-        file_context += f"[Table Parse Error] {e}"
+        file_context += f"
+[Table Parse Error] {e}"