small prompt change

gemini_agent.py

@@ -229,6 +229,26 @@ Provide only the direct answer. If it's a quote, give just the quoted text. If i
             if match:
                 answer = match.group(0).strip()
         
+        # Post-processing for chess move extraction
+        if 'chess position' in question.lower() and 'image' in question.lower():
+            move_match = re.search(r'([KQRBN]?[a-h]?[1-8]?x?[a-h][1-8](=[QRBN])?[+#]?)', answer)
+            if move_match:
+                answer = move_match.group(1)
+
+        # Post-processing for strict list extraction
+        if any(kw in question.lower() for kw in ["alphabetize", "comma separated", "list", "ingredients", "page numbers", "vegetables"]):
+            # Extract only a comma-separated list of words (allowing spaces)
+            list_match = re.findall(r'[A-Za-z][A-Za-z ]*', answer)
+            if list_match:
+                answer = ', '.join([item.strip() for item in list_match if item.strip()])
+
+        # Wikipedia tool integration (simple version)
+        if 'wikipedia' in question.lower() or 'according to wikipedia' in question.lower():
+            # Add a Wikipedia search instruction to the prompt if not already present
+            if 'wikipedia' not in prompt.lower():
+                prompt += "\nIf you do not know the answer, search the latest English Wikipedia and use only information from there."
+            # Optionally, you could call a real Wikipedia API here for retrieval-augmented generation
+
         return answer
     
     async def _generate_video_answer_from_question(self, question: str, video_id: str) -> str: