Update app.py

app.py

@@ -65,15 +65,52 @@ class BasicAgent:
     def handle_logic_riddles(self, question: str) -> str | None:
         q = question.lower().strip()
 
-        if re.search(r"opposite of the word ['\"]?left['\"]?", q):
-            return "right"
-
-        # Add more patterns here
-        if re.match(r".*first letter of the alphabet.*", q):
-            return "a"
+        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):
+                return item["answer"]
 
         return None
 
+
     def __call__(self, question: str, video_path: str = None) -> str:
         print(f"Agent received question: {question[:60]}...")