Update app.py

app.py

@@ -114,6 +114,23 @@ class BasicAgent:
     
         return None
 
+    
+    def solve_riddle(self, riddle: str) -> str:
+        """Fallback riddle solver using QA pipeline with general logic context."""
+        riddle_context = (
+            "You are a riddle-solving assistant. Try to give a short and logical answer to riddles.\n"
+            "Examples:\n"
+            "Q: What has keys but can't open locks?\nA: A piano\n"
+            "Q: What runs but never walks?\nA: Water\n"
+            "Q: What comes once in a minute, twice in a moment, but never in a thousand years?\nA: The letter M\n"
+            f"Q: {riddle}\nA:"
+        )
+        try:
+            result = self.qa_pipeline(question=riddle, context=riddle_context)
+            return result["answer"]
+        except Exception as e:
+            return f"Could not solve riddle: {e}"
+
 
 
     def __call__(self, question: str, video_path: str = None) -> str:
@@ -123,6 +140,9 @@ class BasicAgent:
         logic_answer = self.handle_logic_riddles(question)
         if logic_answer is not None:
             return f"🧠 Logic Answer: {logic_answer}"
+        else:
+            riddle_guess = self.solve_riddle(question)
+            return f"🤖 Riddle Guess: {riddle_guess}"
 
         if video_path:
             transcription = self.call_whisper(video_path)