Update agent.py

agent.py

@@ -87,6 +87,27 @@ def answer_question(question: str, task_id: str = None) -> str:
             yt = get_youtube_transcript(question)
             file_context += f"\n[YouTube Transcript] {yt}"
         elif "* on the set" in question and file_context:
+    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"\n[Parsed Non-Commutative Set] {result}"
+    except Exception as e:
+        file_context += f"\n[Table Parse Error] {e}"
     # Dynamic parser for operation table
     import re
     import pandas as pd
@@ -106,9 +127,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}"
 # Parse table to extract non-commutative elements
     import re
     import pandas as pd