Update app.py

app.py

@@ -44,22 +44,54 @@ class AssistantAgent:
             return f"Error calling xAI API: {e}"
 
     def check_commutative(self, table: str) -> str:
-        rows = table.strip().split('\n')[2:]
-        elements = ['a', 'b', 'c', 'd', 'e']
-        operation = {}
-        for i, row in enumerate(rows):
-            cols = row.split('|')[1:-1]
-            for j, val in enumerate(cols[1:]):
-                operation[(elements[i], elements[j])] = val
-
-        non_commutative = set()
-        for a in elements:
-            for b in elements:
-                if operation.get((a, b)) != operation.get((b, a)):
-                    non_commutative.add(a)
-                    non_commutative.add(b)
-
-        return ", ".join(sorted(non_commutative)) if non_commutative else "No counter-examples found"
+        try:
+            # Tách các dòng của bảng
+            rows = table.strip().split('\n')
+            if len(rows) < 3:  # Cần ít nhất 3 dòng (header, phân cách, dữ liệu)
+                return "Error: Table format is invalid (not enough rows)."
+
+            # Lấy danh sách phần tử từ dòng đầu tiên (header)
+            header = rows[0].split('|')
+            if len(header) < 3:  # Cần ít nhất 3 cột (|, phần tử, |)
+                return "Error: Table format is invalid (header too short)."
+
+            elements = header[1:-1][1:]  # Bỏ cột đầu tiên (*) và cột cuối, bỏ ký hiệu |
+            num_elements = len(elements)
+            if num_elements < 1:
+                return "Error: No elements found in table header."
+
+            # Kiểm tra số dòng dữ liệu (phải khớp với số phần tử + 2 dòng header)
+            if len(rows) != num_elements + 2:
+                return f"Error: Table row count ({len(rows)}) does not match expected ({num_elements + 2})."
+
+            # Kiểm tra định dạng dòng phân cách
+            if not rows[1].startswith('|---'):
+                return "Error: Table format is invalid (missing separator row)."
+
+            # Xây dựng bảng phép toán
+            operation = {}
+            for i in range(2, len(rows)):
+                cols = rows[i].split('|')
+                if len(cols) != num_elements + 2:  # Số cột phải khớp với số phần tử + 2
+                    return f"Error: Row {i} has incorrect number of columns (expected {num_elements + 2}, got {len(cols)})."
+
+                for j in range(1, len(cols) - 1):
+                    val = cols[j + 1].strip()
+                    if val not in elements:
+                        return f"Error: Invalid value '{val}' in table (not in {elements})."
+                    operation[(elements[i - 2], elements[j - 1])] = val
+
+            # Tìm các cặp không giao hoán: a*b != b*a
+            non_commutative = set()
+            for a in elements:
+                for b in elements:
+                    if operation.get((a, b)) != operation.get((b, a)):
+                        non_commutative.add(a)
+                        non_commutative.add(b)
+
+            return ", ".join(sorted(non_commutative)) if non_commutative else "No counter-examples found"
+        except Exception as e:
+            return f"Error processing table: {e}"
 
     def classify_vegetables(self, items: str) -> str:
         all_items = [item.strip() for item in items.split(",")]
@@ -103,7 +135,7 @@ class AssistantAgent:
                     answers.append(answer)
             if i + batch_size < len(questions):
                 print("Waiting 1 second before next batch to avoid rate limit...")
-                time.sleep(1)  # Độ trễ nhỏ để tránh gọi API quá nhanh
+                time.sleep(1)
         return answers
 
     def process_file(self, question: str, file_path: str) -> str: