Update app.py

app.py

@@ -1,64 +1,54 @@
+import json
+import torch
 import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 import os
-import uuid
 
-# Load translation model
-translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
-
-# Supported languages
-languages = {
-    "English": "eng_Latn",
-    "French": "fra_Latn",
-    "Spanish": "spa_Latn",
-    "German": "deu_Latn",
-    "Hindi": "hin_Deva",
-    "Marathi": "mar_Deva"
-}
+# Load FLORES-200 language codes from lang_code.json
+with open("lang_code.json", "r", encoding="utf-8") as f:
+    LANG_CODES = json.load(f)
 
-def translate_text(text, source_lang, target_lang):
-    if not text.strip():
-        return "Please enter text to translate.", None
+# Translation pipeline
+translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
 
+# Translation + TTS function
+def translate_and_speak(text, src_lang, tgt_lang):
     try:
-        translated = translator(
-            text,
-            src_lang=languages[source_lang],
-            tgt_lang=languages[target_lang]
-        )[0]['translation_text']
+        src_code = LANG_CODES[src_lang]
+        tgt_code = LANG_CODES[tgt_lang]
 
-        # Create TTS for translated text
-        tts_file = f"{uuid.uuid4()}.mp3"
-        tts = gTTS(translated, lang=target_lang[:3].lower())  
-        tts.save(tts_file)
+        # Translate
+        result = translator(text, src_lang=src_code, tgt_lang=tgt_code)
+        translated_text = result[0]['translation_text']
 
-        return translated, tts_file
-    except Exception as e:
-        return f"Error: {str(e)}", None
+        # Generate speech for translated text
+        tts = gTTS(translated_text, lang=tgt_lang.split("_")[0])
+        audio_path = "translated_audio.mp3"
+        tts.save(audio_path)
 
+        return translated_text, audio_path
+    except Exception as e:
+        return f"Error: {e}", None
 
-# UI
-with gr.Blocks(css=".gradio-container {max-width: 750px; margin: auto;}") as demo:
-    gr.Markdown(
-        "<h1 style='text-align: center; color: black;'>@GenAi MultiLanguage Translator Model</h1>"
-    )
+# Gradio UI
+with gr.Blocks(theme="soft") as demo:
+    gr.HTML("<h1 style='color:black; text-align:center;'>@GenAi MultiLanguage Translator Model</h1>")
 
     with gr.Row():
         with gr.Column():
-            src_text = gr.Textbox(label="Enter your text", placeholder="Type something here...")
-            src_lang = gr.Dropdown(list(languages.keys()), value="English", label="Source Language")
-            tgt_lang = gr.Dropdown(list(languages.keys()), value="French", label="Target Language")
-            translate_btn = gr.Button("Translate", variant="primary")
+            src_text = gr.Textbox(label="Enter Text", placeholder="Type something here...")
+            src_lang = gr.Dropdown(choices=list(LANG_CODES.keys()), value="English (Latin script)", label="Source Language")
+            tgt_lang = gr.Dropdown(choices=list(LANG_CODES.keys()), value="French", label="Target Language")
+            translate_btn = gr.Button("Translate")
 
         with gr.Column():
-            translated_text = gr.Textbox(label="Translated Text")
-            audio_output = gr.Audio(label="Speak Translation", type="filepath")
+            output_text = gr.Textbox(label="Translated Text")
+            output_audio = gr.Audio(label="Speech Output", type="filepath")
 
-    translate_btn.click(
-        translate_text,
-        inputs=[src_text, src_lang, tgt_lang],
-        outputs=[translated_text, audio_output]
-    )
+    translate_btn.click(fn=translate_and_speak,
+                        inputs=[src_text, src_lang, tgt_lang],
+                        outputs=[output_text, output_audio])
 
+# Run app
 demo.launch()