Update gemini_agent.py

gemini_agent.py

@@ -163,40 +163,29 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
     async def _handle_text_question(self, question: str) -> str:
         """Handle regular text-based questions"""
         prompt = ""
-        # Improved: Use retrieval for Wikipedia-style fact questions
+        # Only use retrieval for explicit web/Wikipedia questions
         def is_explicit_retrieval_question(question):
             q = question.lower()
-            # Add more patterns for Wikipedia-relevant fact questions
-            if (
+            return (
                 "according to wikipedia" in q or
                 "from wikipedia" in q or
                 "search the web" in q or
                 "duckduckgo" in q or
                 "web search" in q
-            ):
-                return True
-            # Factoid pattern: how many/who/what/when/where + entity + year/range
-            if re.search(r"(how many|who|what|when|where)", q) and re.search(r"\\b(19|20)\\d{2}\\b", q):
-                return True
-            # If question mentions Wikipedia or is about a person/entity and a year
-            if "wikipedia" in q:
-                return True
-            # Mercedes Sosa/artist/album/year pattern (custom for your use case)
-            if ("mercedes sosa" in q and ("album" in q or "discography" in q)):
-                return True
-            return False
+            )
         wiki_context = ""
         ddg_context = ""
         if is_explicit_retrieval_question(question):
-            try:
-                wiki_context = self.wiki_tool.run(question)
-            except Exception as e:
-                print(f"Wikipedia tool failed: {e}")
-            # Optionally, also try DuckDuckGo for broader coverage
-            try:
-                ddg_context = self.ddg_tool.run(question)
-            except Exception as e:
-                print(f"DuckDuckGo tool failed: {e}")
+            if "wikipedia" in question.lower():
+                try:
+                    wiki_context = self.wiki_tool.run(question)
+                except Exception as e:
+                    print(f"Wikipedia tool failed: {e}")
+            if "duckduckgo" in question.lower() or "web search" in question.lower():
+                try:
+                    ddg_context = self.ddg_tool.run(question)
+                except Exception as e:
+                    print(f"DuckDuckGo tool failed: {e}")
         # Handle attached file questions with enhanced prompts
         if 'attached' in question.lower():
             if 'python code' in question.lower():
@@ -237,11 +226,10 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         # Prepend context to the prompt if available and likely relevant
         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"])
-        # Improved: Prefer Wikipedia context if available
         if wiki_context and is_good_context(wiki_context):
-            prompt = f"Use the following Wikipedia context to answer the question:\n{wiki_context}\n\nProvide only the direct answer. If it's a number, give just the number. If it's a name, give just the name. No explanations.\n\n{question}\n\nAnswer:"
+            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\nProvide only the direct answer. If it's a number, give just the number. If it's a name, give just the name. No explanations.\n\n{question}\n\nAnswer:"
+            prompt = f"Use the following web search context to answer the question:\n{ddg_context}\n\n{prompt}"
         
         # Use the constructed prompt for all cases
         await self._rate_limit()
@@ -253,15 +241,6 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             )
         )
         answer = response.text.strip()
-        # Improved: If expecting a number, extract from Wikipedia context if not found in answer
-        if (re.search(r"how many", question.lower()) or re.search(r"number", question.lower())):
-            # Try to extract a number from the answer
-            num_match = re.search(r"\\b\\d+\\b", answer)
-            if not num_match and wiki_context:
-                # Try to extract from Wikipedia context
-                num_match = re.search(r"\\b\\d+\\b", wiki_context)
-            if num_match:
-                answer = num_match.group(0)
         
         # Extract the core answer
         if ':' in answer: