Update app.py

app.py

@@ -1,4 +1,5 @@
 import requests
+import string
 import inspect
 import os
 import re
@@ -63,57 +64,55 @@ class BasicAgent:
             return context  # Fallback to context if QA fails
 
     def handle_logic_riddles(self, question: str) -> str | 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
+        # 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