Update chatbot/core.py

chatbot/core.py

@@ -9,7 +9,15 @@ from langchain.schema import BaseRetriever, Document
 
 def translate_to_english(text: str) -> str:
     """Use Gemini LLM to translate text to English."""
-    prompt = f"Translate the following text to English:\n\n{text}"
+    prompt = f"""
+    You are an assistant for Wikipedia searches.
+    The query may be in any language.
+    Extract and return only the most relevant keyword (e.g. a person's name, city, or key term) in English/international form.
+    Return only the keyword—no explanations.
+    
+    Query:
+    {text}
+    """
     response = gemini_llm.invoke(prompt)  # Invoke Gemini for translation
     return response  # Assuming `gemini_llm.invoke()` returns plain text