Update app.py

app.py

@@ -7,6 +7,7 @@ from langchain_huggingface import HuggingFaceEndpoint
 from langchain_core.prompts import PromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 from transformers import pipeline
+from langdetect import detect
 
 # Use environment variables for keys
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -97,6 +98,18 @@ def generate_follow_up(user_text):
         cleaned = ["Would you like to explore this topic further?"]
     return random.choice(cleaned)
 
+from langdetect import detect
+
+def ensure_english(text):
+    """Check if the model accidentally generated a non-English response."""
+    try:
+        detected_lang = detect(text)
+        if detected_lang != "en":
+            return "⚠️ Sorry, I only respond in English. Can you rephrase your question?"
+    except:
+        return "⚠️ Language detection failed. Please ask your question again."
+    return text
+
 def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     """
     Generates HAL's answer with depth and a follow-up question.
@@ -135,21 +148,25 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=512):
     
     # Instruct the model to generate a detailed, in-depth answer.
     prompt = PromptTemplate.from_template(
-        (
-            "[INST] {system_message}\n\nCurrent Conversation:\n{chat_history}\n\n"
-            "User: {user_text}.\n [/INST]\n"
-            "AI: Please provide a detailed explanation in depth. "
-            "Ensure your response covers the topic thoroughly and is written in a friendly, conversational style, "
-            "starting with a phrase like 'Certainly!', 'Of course!', or 'Great question!'." 
-            "Answer exclusively in English, and do not include extra commentary." + style_clause +
-            "\nHAL:"
-        )
+    (
+        "[INST] {system_message}\n\nCurrent Conversation:\n{chat_history}\n\n"
+        "User: {user_text}.\n [/INST]\n"
+        "AI: Please provide a detailed explanation in depth. "
+        "Ensure your response covers the topic thoroughly and is written in a friendly, conversational style, "
+        "starting with a phrase like 'Certainly!', 'Of course!', or 'Great question!'."
+        "🚨 IMPORTANT: Answer exclusively in **English only**. Do not generate responses in any other language."
+        "\nHAL:"
     )
+)
     
     chat = prompt | hf.bind(skip_prompt=True) | StrOutputParser(output_key='content')
     response = chat.invoke(input=dict(system_message=system_message, user_text=user_text, chat_history=filtered_history))
     response = response.split("HAL:")[-1].strip()
 
+    # 🚨 Ensure the response is in English
+    response = ensure_english(response)
+
+
     # Fallback in case the generated answer is empty
     if not response:
         response = "Certainly, here is an in-depth explanation: [Fallback explanation]."