Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,7 +18,7 @@ if not HF_TOKEN:
     st.stop()
 
 client = InferenceClient(
-    model="google-t5/t5-base",
+    model="google/gemma-7b-it",
     token=HF_TOKEN
 )
 
@@ -68,42 +68,42 @@ def speech_to_text(audio_file):
     return text
 
 # ==============================
-# LANGUAGE MAP (IMPORTANT)
-# ==============================
-lang_map = {
-    "English": "en",
-    "Tamil": "ta",
-    "Hindi": "hi",
-    "French": "fr",
-    "Arabic": "ar",
-    "Spanish": "es",
-    "German": "de"
-}
-
-# ==============================
-# TRANSLATION FUNCTION (FIXED)
+# TRANSLATION FUNCTION
 # ==============================
 def translate(text, target_lang):
 
-    # Detect source language
+    # Detect language
     try:
         src_lang = detect(text)
     except:
-        src_lang = "en"
-
-    tgt_lang = lang_map[target_lang]
+        src_lang = "unknown"
 
     memory = retrieve_memory(text)
 
+    prompt = f"""
+You are a professional multilingual translator.
+
+Translate from {src_lang} to {target_lang}.
+
+Text:
+{text}
+
+Rules:
+- Preserve meaning accurately
+- Use natural language
+- Do NOT explain
+- Return ONLY translated text
+"""
+
     try:
-        # ✅ Correct API for T5
-        response = client.translation(
-            text=text,
-            src_lang=src_lang,
-            tgt_lang=tgt_lang
+        response = client.text_generation(
+            prompt,
+            max_new_tokens=150,   # ⚡ optimized for speed
+            temperature=0.2,
+            top_p=0.9
         )
 
-        translated = response["translation_text"]
+        translated = response.strip()
 
     except Exception as e:
         translated = f"❌ Translation failed: {str(e)}"
@@ -117,7 +117,7 @@ def translate(text, target_lang):
 # ==============================
 st.set_page_config(page_title="AI Translator", layout="wide")
 
-st.title("🌍 AI Translator with Voice (T5)")
+st.title("🌍 AI Translator with Voice (Gemma 7B)")
 
 tab1, tab2 = st.tabs(["📝 Text Input", "🎤 Voice Input"])
 
@@ -127,7 +127,7 @@ input_text = ""
 # TEXT INPUT
 # ------------------------------
 with tab1:
-    input_text = st.text_area("Enter text")
+    input_text = st.text_area("Enter text", height=150)
 
 # ------------------------------
 # VOICE INPUT
@@ -149,11 +149,11 @@ with tab2:
 # ------------------------------
 target_lang = st.selectbox(
     "Target Language",
-    list(lang_map.keys())
+    ["English", "Tamil", "Hindi", "French", "Arabic", "Spanish", "German"]
 )
 
 # ------------------------------
-# TRANSLATE BUTTON
+# TRANSLATE
 # ------------------------------
 if st.button("Translate"):
     if not input_text.strip():