Update gemini_agent.py

gemini_agent.py

@@ -62,7 +62,14 @@ class GeminiAgent:
 
 {question}
 
-Provide only the direct answer. If it's a quote, give just the quoted text. If it's a number, give just the number. If it's about bird species count, analyze carefully and give the exact count. If it's about dialogue, provide the exact words spoken."""
+Provide only the direct answer requested. Be extremely precise:
+- For quotes: give exact words without quotation marks
+- For numbers: give only the number
+- For names: give only the name requested
+- For bird species counts: count all visible species simultaneously on screen
+- For dialogue responses: give the exact spoken words
+
+Answer:"""
         
         try:
             await self._rate_limit()
@@ -160,11 +167,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"""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:"""
+                prompt = f"""This question asks for the final numeric output from attached Python code. Trace through the code execution step by step and provide only the final number that would be printed or returned:\n\n{question}\n\nFinal numeric output:"""
+            elif '.mp3' in question.lower() and ('ingredients' in question.lower() or 'page numbers' in question.lower()):
+                prompt = f"""This question refers to an attached audio file. Based on the context provided, extract the requested information and format it exactly as specified:\n\n{question}\n\nAnswer (formatted as requested):"""
             else:
-                prompt = f"""This question refers to an attached file. Provide the most likely answer:\n\n{question}\n\nAnswer:"""
+                prompt = f"""This question refers to an attached file. Provide the most accurate answer based on the context:\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:"""
@@ -179,11 +186,12 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         ):
             # 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."
+                definition = ("IMPORTANT: Use strict botanical definitions. A vegetable is any edible plant part that is NOT a botanical fruit. "
+                              "Botanical fruits develop from flowers and contain seeds (tomatoes, peppers, beans, corn, peas, zucchini, etc. are fruits). "
+                              "True vegetables include roots, stems, leaves, and other plant parts.")
+                prompt = f"{definition}\n\n{question}\n\nProvide ONLY the botanical vegetables from the list, alphabetized, comma separated. No fruits, no explanations."
             else:
-                prompt = f"{question}\n\nList only the requested items, alphabetized, comma separated, and do not include any explanations or extra words."
+                prompt = f"{question}\n\nProvide exactly what is requested: alphabetized, comma separated list with no extra text or explanations."
         # 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:"""
@@ -224,10 +232,10 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
         
         # If the question expects a single value, extract it
         if any(kw in question.lower() for kw in ["how many", "what is the", "who", "where", "give only", "provide only"]):
-            # Extract the first number, word, or phrase (tweak regex as needed)
+            # Extract the first meaningful part of the answer
             match = re.search(r'^[A-Za-z0-9 ,+-]+', answer)
             if match:
-                answer = match.group(0).strip()
+                answer = match.group().strip()
         
         return answer