Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 from groq import Groq
 
-# Setup Groq client
+# Initialize Groq client
 groq_api_key = "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"
 client = Groq(api_key=groq_api_key)
 
@@ -12,31 +12,41 @@ def translate_with_groq(text, src_lang, tgt_lang):
     if src_lang == tgt_lang:
         return text
 
-    prompt = f"Translate the following text from {src_lang} to {tgt_lang}:\n\n\"{text}\""
+    prompt = (
+        f"You are a translation assistant. Translate the following text from {src_lang} to {tgt_lang}:\n\n\"{text}\""
+    )
 
     try:
-        chat_completion = client.chat.completions.create(
-            messages=[{"role": "user", "content": prompt}],
-            model="gemma-7b-it",
-            temperature=0.4
+        resp = client.chat.completions.create(
+            model="mistral-saba-24b",
+            messages=[
+                {"role": "system", "content": "You are a multilingual translation assistant."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0.2,
+            max_completion_tokens=512
         )
-        translated = chat_completion.choices[0].message.content.strip()
-        return translated
+        return resp.choices[0].message.content.strip()
     except Exception as e:
-        return f"Error: {str(e)}"
-
-# Gradio UI
-iface = gr.Interface(
-    fn=translate_with_groq,
-    inputs=[
-        gr.Textbox(label="Enter text"),
-        gr.Dropdown(choices=lang_options, label="Source Language", value="English"),
-        gr.Dropdown(choices=lang_options, label="Target Language", value="Urdu")
-    ],
-    outputs=gr.Textbox(label="Translated Text"),
-    title="🌍 AI Translator with Groq",
-    description="Translate between multiple languages using Groq LLM (Gemma-7B-IT)."
-)
+        return f"❌ Error: {e}"
+
+# Gradio UI (Blocks with improved layout)
+with gr.Blocks(theme=gr.themes.Default(primary_hue="teal", font="poppins")) as demo:
+    gr.Markdown("# 🌍 AI Multilingual Translator")
+    gr.Markdown("Translate between multiple languages using **Groq's Mistral-Saba 24B** model.")
+
+    with gr.Row():
+        with gr.Column():
+            input_text = gr.Textbox(label="📝 Input Text", lines=5, placeholder="Type your text here...")
+            src_lang = gr.Dropdown(lang_options, value="English", label="🌐 From")
+            tgt_lang = gr.Dropdown(lang_options, value="Urdu", label="🌐 To")
+            translate_btn = gr.Button("🔁 Translate", variant="primary")
+        with gr.Column():
+            output_text = gr.Textbox(label="✅ Translated Text", lines=5)
+
+    gr.Markdown("---\nPowered by Groq & Gradio")
+
+    translate_btn.click(translate_with_groq, inputs=[input_text, src_lang, tgt_lang], outputs=output_text)
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()