Update gemini_agent.py

gemini_agent.py

@@ -183,11 +183,18 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         if is_explicit_retrieval_question(question):
             try:
                 wiki_context = self.wiki_tool.run(question)
+                print(f"Wiki context: {wiki_context[:200]}...")  # Debug
             except Exception as e:
                 print(f"Wikipedia tool failed: {e}")
-            if "duckduckgo" in question.lower() or "web search" in question.lower():
+            
+            # Use DuckDuckGo as fallback if Wikipedia context is poor or for web search requests
+            def is_poor_context(context):
+                return not context or any(x in context.lower() for x in ["not found", "no results", "does not contain", "disambiguation"])
+            
+            if "duckduckgo" in question.lower() or "web search" in question.lower() or is_poor_context(wiki_context):
                 try:
                     ddg_context = self.ddg_tool.run(question)
+                    print(f"DDG context: {ddg_context[:200]}...")  # Debug
                 except Exception as e:
                     print(f"DuckDuckGo tool failed: {e}")
         # Handle attached file questions with enhanced prompts
@@ -231,7 +238,10 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         def is_good_context(context):
             return context and not any(x in context.lower() for x in ["not found", "no results", "does not contain information"])
         if wiki_context and is_good_context(wiki_context):
-            prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\n{prompt}"
+            if "first name" in question.lower():
+                prompt = f"Based on this Wikipedia information, find the actor who played Ray/Roman and identify what character they played in Magda M., then give only their first name:\n{wiki_context}\n\n{question}\n\nAnswer:"
+            else:
+                prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\n{prompt}"
         elif ddg_context and is_good_context(ddg_context):
             prompt = f"Use the following web search context to answer the question:\n{ddg_context}\n\n{prompt}"