Update gemini_agent.py

gemini_agent.py

@@ -76,7 +76,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 video_prompt,
                 generation_config=genai.types.GenerationConfig(
                     max_output_tokens=50,
-                    temperature=0.2
+                    temperature=0.0
                 )
             )
             answer = response.text.strip()
@@ -144,7 +144,7 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                 excel_prompt,
                 generation_config=genai.types.GenerationConfig(
                     max_output_tokens=150,
-                    temperature=0.1
+                    temperature=0.0
                 )
             )
             answer = response.text.strip()
@@ -163,24 +163,28 @@ 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 = ""
-        # Only use retrieval for explicit web/Wikipedia questions
+        # Use retrieval for explicit web/Wikipedia questions and factual queries
         def is_explicit_retrieval_question(question):
             q = question.lower()
-            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
-            )
+            # Explicit retrieval requests
+            explicit_triggers = [
+                "according to wikipedia", "from wikipedia", "search the web", 
+                "duckduckgo", "web search"
+            ]
+            # Factual question patterns that benefit from Wikipedia
+            factual_triggers = [
+                "who played", "actor who", "actress who", "cast of", "starred in",
+                "tv show", "television", "movie", "film", "character in",
+                "version of", "adaptation of", "remake of"
+            ]
+            return any(trigger in q for trigger in explicit_triggers + factual_triggers)
         wiki_context = ""
         ddg_context = ""
         if is_explicit_retrieval_question(question):
-            if "wikipedia" in question.lower():
-                try:
-                    wiki_context = self.wiki_tool.run(question)
-                except Exception as e:
-                    print(f"Wikipedia tool failed: {e}")
+            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)
@@ -189,11 +193,11 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         # Handle attached file questions with enhanced prompts
         if 'attached' in question.lower():
             if 'python code' in question.lower():
-                prompt = f"""Analyze this Python code question carefully. If it involves calculations, show your work step by step and provide only the final numeric result:\n\n{question}\n\nFinal Answer:"""
+                prompt = f"""This question refers to attached Python code. Based on typical code execution patterns, provide the most likely numeric output:\n\n{question}\n\nAnswer:"""
             elif '.mp3' in question.lower():
-                prompt = f"""This question refers to an attached audio file. Based on common audio analysis patterns, provide the most precise answer:\n\n{question}\n\nAnswer:"""
+                prompt = f"""This question refers to an attached audio file. Provide the most likely answer based on the context:\n\n{question}\n\nAnswer:"""
             else:
-                prompt = f"""Analyze this attached file question. Provide the most accurate and specific answer possible:\n\n{question}\n\nAnswer:"""
+                prompt = f"""This question refers to an attached file. Provide the most likely answer:\n\n{question}\n\nAnswer:"""
         # Handle chess position question
         elif 'chess position' in question.lower() and 'image' in question.lower():
             prompt = f"""This is a chess question with an attached image. Provide the best chess move in algebraic notation:\n\n{question}\n\nAnswer:"""