Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 from transformers import pipeline
+from langdetect import detect  # 🔥 new for auto language detection
 
 # ---------------- CONFIG ----------------
 MODEL_REPO = "HuggingFaceH4/zephyr-7b-beta"
@@ -33,21 +34,25 @@ def is_translation_request(message: str) -> bool:
     return non_ascii_ratio > 0.4
 
 
+def translate_text(text: str) -> str:
+    """Detects language and translates to English."""
+    try:
+        src_lang = detect(text)  # auto-detect like "ru", "es", "fr"
+        result = translator(text, src_lang=src_lang, tgt_lang="en")
+        return result[0]["translation_text"]
+    except Exception as e:
+        return f"⚠️ Translation failed: {str(e)}"
+
+
 # ---------------- CHAT FUNCTION ----------------
 def stream_response(message, chat_history, system_message, max_tokens, temperature, top_p, response_style):
     # check if translation
     if is_translation_request(message):
-        try:
-            translated = translator(message, src_lang="auto", tgt_lang="en")[0]["translation_text"]
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": translated})
-            yield "", chat_history
-            return
-        except Exception as e:
-            chat_history.append({"role": "user", "content": message})
-            chat_history.append({"role": "assistant", "content": f"⚠️ Translation failed: {str(e)}"})
-            yield "", chat_history
-            return
+        translated = translate_text(message)
+        chat_history.append({"role": "user", "content": message})
+        chat_history.append({"role": "assistant", "content": translated})
+        yield "", chat_history
+        return
 
     # Otherwise → normal Zephyr response
     if response_style == "Concise":
@@ -80,18 +85,17 @@ def stream_response(message, chat_history, system_message, max_tokens, temperatu
     # finally clear input box once after streaming is done
     yield "", chat_history
 
+
 # ---------------- UI ----------------
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet", secondary_hue="pink")) as demo:
     gr.Markdown(
         """
         # 🤖 Zephyr-7B Chat + 🌍 Translator  
         Welcome! This app combines **Zephyr-7B** (for smart chat) with **M2M100** (for multilingual translation).  
-
         **How to use:**
         - 💬 Ask *anything* → Zephyr will reply.  
         - 🌐 Paste non-English text or say "translate" → auto translation into English.  
         - ⚙️ Adjust settings in the panel if you want different styles.  
-
         ---
         """
     )
@@ -138,4 +142,4 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet", secondary_hue="pink"))
     gr.Markdown("🔗 Built with ❤️ using [Zephyr-7B](https://huggingface.co/HuggingFaceH4/zephyr-7b-beta) & [M2M100](https://huggingface.co/facebook/m2m100_418M).")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()