Update gemini_agent.py

gemini_agent.py

@@ -228,6 +228,13 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             sentences = answer.split('. ')
             answer = sentences[0] + '.'
         
+        # 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)
+            match = re.search(r'^[A-Za-z0-9 ,+-]+', answer)
+            if match:
+                answer = match.group(0).strip()
+        
         return answer
     
     async def _generate_video_answer_from_question(self, question: str, video_id: str) -> str: