Update gemini_agent.py

gemini_agent.py

@@ -6,8 +6,8 @@ import re
 import time
 import asyncio
 # Add LangChain tools for Wikipedia and DuckDuckGo
-from langchain.tools import DuckDuckGoSearchRun, WikipediaQueryRun
-from langchain.utilities import WikipediaAPIWrapper
+from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
 
 load_dotenv()
 
@@ -19,7 +19,7 @@ class GeminiAgent:
         api_key = os.getenv('GOOGLE_API_KEY')
         genai.configure(api_key=api_key)
         
-        self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
+        self.model = genai.GenerativeModel('gemini-1.5-pro-latest')
         self.last_request_time = 0
         self.min_request_interval = 6.0  # 6 seconds between requests (10 per minute limit)
         
@@ -162,7 +162,6 @@ 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
         def is_explicit_retrieval_question(question):
             q = question.lower()
@@ -186,43 +185,10 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
                     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():
-                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. Provide the most likely answer based on the context:\n\n{question}\n\nAnswer:"""
-            else:
-                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:"""
-        # Handle list extraction and formatting
-        elif (
-            'alphabetize' in question.lower() or 
-            'comma separated' in question.lower() or 
-            'list' in question.lower() or 
-            'ingredients' in question.lower() or 
-            'page numbers' in question.lower() or 
-            'vegetables' in question.lower()
-        ):
-            # Add domain definition for botanical vegetables
-            if 'vegetable' in question.lower() and ('botany' in question.lower() or 'botanical' in question.lower()):
-                definition = ("In botany, a vegetable is any edible part of a plant that is not a fruit or seed. "
-                              "Fruits contain seeds and develop from the ovary of a flower. Use this definition.")
-                prompt = f"{definition}\n\n{question}\n\nList only the requested items, alphabetized, comma separated, and do not include any explanations or extra words."
-            else:
-                prompt = f"{question}\n\nList only the requested items, alphabetized, comma separated, and do not include any explanations or extra words."
-        # Create enhanced prompt based on question type
-        elif 'how many' in question.lower() or 'what is the' in question.lower():
-            prompt = f"""Provide only the exact answer to this question. No explanations, just the specific number, name, or fact requested:\n\n{question}\n\nAnswer:"""
-        elif 'who' in question.lower():
-            prompt = f"""Provide only the name requested. No explanations or additional context:\n\n{question}\n\nAnswer:"""
-        elif 'where' in question.lower():
-            prompt = f"""Provide only the location requested. No explanations:\n\n{question}\n\nAnswer:"""
-        else:
-            prompt = f"""Answer this question with only the essential information requested:\n\n{question}\n\nAnswer:"""
         
+        # Simplified prompt construction
+        prompt = f"Answer the following question:\n\n{question}"
+
         # 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"])