Update app.py

app.py

@@ -63,52 +63,58 @@ class BasicAgent:
             return context  # Fallback to context if QA fails
 
     def handle_logic_riddles(self, question: str) -> str | None:
-        q = question.lower().strip()
-
-        logic_patterns = [
-            {
-                "pattern": r"opposite of the word ['\"]?left['\"]?",
-                "answer": "right"
-            },
-            {
-                "pattern": r"what comes after ['\"]?a['\"]?",
-                "answer": "b"
-            },
-            {
-                "pattern": r"first letter of the alphabet",
-                "answer": "a"
-            },
-            {
-                "pattern": r"what is the color of the clear sky",
-                "answer": "blue"
-            },
-            {
-                "pattern": r"how many sides does a triangle have",
-                "answer": "3"
-            },
-            {
-                "pattern": r"how many legs does a spider have",
-                "answer": "8"
-            },
-            {
-                "pattern": r"what is 2 \+ 2",
-                "answer": "4"
-            },
-            {
-                "pattern": r"what is the opposite of ['\"]?up['\"]?",
-                "answer": "down"
-            },
-            {
-                "pattern": r"if you understand this sentence.*opposite.*left",
-                "answer": "right"
-            }
-        ]
-
-        for item in logic_patterns:
-            if re.search(item["pattern"], q, re.IGNORECASE):
-                return item["answer"]
-
-        return None
+    import string
+
+    # Normalize the input
+    q = question.lower().strip()
+    q = q.translate(str.maketrans("", "", string.punctuation))  # remove punctuation
+    q = re.sub(r"\s+", " ", q)  # normalize multiple spaces
+
+    logic_patterns = [
+        {
+            "pattern": r"opposite of the word left",
+            "answer": "right"
+        },
+        {
+            "pattern": r"what comes after a",
+            "answer": "b"
+        },
+        {
+            "pattern": r"first letter of the alphabet",
+            "answer": "a"
+        },
+        {
+            "pattern": r"what is the color of the clear sky",
+            "answer": "blue"
+        },
+        {
+            "pattern": r"how many sides does a triangle have",
+            "answer": "3"
+        },
+        {
+            "pattern": r"how many legs does a spider have",
+            "answer": "8"
+        },
+        {
+            "pattern": r"what is 2 \+ 2",
+            "answer": "4"
+        },
+        {
+            "pattern": r"what is the opposite of up",
+            "answer": "down"
+        },
+        {
+            "pattern": r"if you understand this sentence.*opposite.*left",
+            "answer": "right"
+        }
+    ]
+
+    for item in logic_patterns:
+        if re.search(item["pattern"], q, re.IGNORECASE):
+            return item["answer"]
+
+    return None
+
 
 
     def __call__(self, question: str, video_path: str = None) -> str: