Update agent.py

agent.py

@@ -87,7 +87,29 @@ 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:
-        # Parse table to extract non-commutative elements
+    # Dynamic parser for operation table
+    import re
+    import pandas as pd
+
+    try:
+        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}"
+# Parse table to extract non-commutative elements
     import re
     import pandas as pd