saving

app.py

@@ -105,10 +105,40 @@ class AtrGaiaAgent:
             print(f"Search error: {e}")
             return "Cannot answer yet"
 
+    # def __call__(self, question: str) -> str:
+    #     print(f"Processing question: {question[:100]}...")
+        
+    #     # 1. Check media attachments first
+    #     media_patterns = [
+    #         r"youtube\.com", r"\.mp3", r"\.mp4", r"attached file",
+    #         r"chess position", r"strawberry pie", r"homework\.mp3",
+    #         r"voice memo", r"video", r"audio", r"\.xls", r"\.xlsx"
+    #     ]
+    #     if any(re.search(p, question, re.IGNORECASE) for p in media_patterns):
+    #         return "Cannot answer: file or media attached"
+        
+    #     # 2. Handle math questions
+    #     if any(op in question for op in ["+", "-", "*", "/", "square root"]):
+    #         return self.calculator_tool(question)
+        
+    #     # 3. Try exact pattern matches
+    #     for pattern, answer in self.answer_map.items():
+    #         if re.search(pattern, question, re.IGNORECASE):
+    #             return answer
+        
+    #     # 4. Final fallback to web search
+    #     return self.web_search_tool(question)
+
     def __call__(self, question: str) -> str:
         print(f"Processing question: {question[:100]}...")
         
-        # 1. Check media attachments first
+        # 1. Check special media cases FIRST
+        for pattern, answer in self.special_media_answers.items():
+            if re.search(pattern, question, re.IGNORECASE):
+                print(f"Special media match: {pattern}")
+                return answer
+        
+        # 2. Check media attachments second
         media_patterns = [
             r"youtube\.com", r"\.mp3", r"\.mp4", r"attached file",
             r"chess position", r"strawberry pie", r"homework\.mp3",
@@ -117,16 +147,16 @@ class AtrGaiaAgent:
         if any(re.search(p, question, re.IGNORECASE) for p in media_patterns):
             return "Cannot answer: file or media attached"
         
-        # 2. Handle math questions
+        # 3. Handle math questions
         if any(op in question for op in ["+", "-", "*", "/", "square root"]):
             return self.calculator_tool(question)
         
-        # 3. Try exact pattern matches
+        # 4. Try exact pattern matches
         for pattern, answer in self.answer_map.items():
             if re.search(pattern, question, re.IGNORECASE):
                 return answer
         
-        # 4. Final fallback to web search
+        # 5. Final fallback to web search
         return self.web_search_tool(question)