Update app.py

app.py

@@ -1,100 +1,64 @@
-import torch
 import gradio as gr
-import json
 from transformers import pipeline
-import pyttsx3
-import tempfile
+from gtts import gTTS
 import os
+import uuid
 
-# Load the NLLB model
-# model_path = "../text_translator/model/models--facebook--nllb-200-distilled-600M/snapshots/f8d333a098d19b4fd9a8b18f94170487ad3f821d"
-# translator = pipeline("translation", model=model_path)
-
+# Load translation model
 translator = pipeline("translation", model="facebook/nllb-200-distilled-600M")
-# Load language codes
-# def load_lang_codes(file_path='../text_translator/lang file/lang_code.json'):
-def load_lang_codes(file_path='lang_code.json'):
-    try:
-        with open(file_path, 'r', encoding='utf-8') as f:
-            data = json.load(f)
-            return {entry['Language']: entry['FLORES-200 code'] for entry in data}
-    except Exception as e:
-        print(f"Error loading language codes: {e}")
-        return {}
-
-lang_dict = load_lang_codes()
-language_list = list(lang_dict.keys())
 
-# Text-to-Speech function using pyttsx3
-def text_to_speech(text):
-    engine = pyttsx3.init()
-    engine.setProperty('rate', 160)
-    engine.setProperty('volume', 1.0)
+# Supported languages
+languages = {
+    "English": "eng_Latn",
+    "French": "fra_Latn",
+    "Spanish": "spa_Latn",
+    "German": "deu_Latn",
+    "Hindi": "hin_Deva",
+    "Marathi": "mar_Deva"
+}
 
-    # Save speech to a temporary file
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as fp:
-        filename = fp.name
-    engine.save_to_file(text, filename)
-    engine.runAndWait()
-    return filename
+def translate_text(text, source_lang, target_lang):
+    if not text.strip():
+        return "Please enter text to translate.", None
 
-# Translation + TTS function
-def translate_and_speak(text, src_lang, tgt_lang):
-    src_code = lang_dict.get(src_lang)
-    tgt_code = lang_dict.get(tgt_lang)
+    try:
+        translated = translator(
+            text,
+            src_lang=languages[source_lang],
+            tgt_lang=languages[target_lang]
+        )[0]['translation_text']
 
-    if not src_code or not tgt_code:
-        return "Invalid language selected.", None
+        # Create TTS for translated text
+        tts_file = f"{uuid.uuid4()}.mp3"
+        tts = gTTS(translated, lang=target_lang[:3].lower())  
+        tts.save(tts_file)
 
-    try:
-        result = translator(text, src_lang=src_code, tgt_lang=tgt_code)
-        translated_text = result[0]['translation_text']
-        audio_path = text_to_speech(translated_text)
-        return translated_text, audio_path
+        return translated, tts_file
     except Exception as e:
-        return f"Translation error: {e}", None
+        return f"Error: {str(e)}", None
 
-# Gradio UI
-with gr.Blocks(css="""
-    .gradio-container {
-        font-family: 'Segoe UI', sans-serif;
-        background-color: #f4f4f4;
-        padding: 40px;
-    }
-    h1 {
-        color: black !important;
-        font-size: 28px !important;
-        margin-bottom: 30px !important;
-    }
-    .gr-button {
-        font-size: 16px !important;
-        padding: 10px 20px !important;
-    }
-    .gr-box {
-        padding: 20px !important;
-    }
-""") as demo:
-    gr.Markdown("<h1 style='text-align: center;'>@GenAI MultiLanguage Translator Model</h1>")
+
+# 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>"
+    )
 
     with gr.Row():
-        with gr.Column(scale=1):
-            input_text = gr.Textbox(
-                label="🌐 Enter Text",
-                placeholder="Type something to translate...",
-                lines=4
-            )
-            gr.Markdown("<br>")
-            src_dropdown = gr.Dropdown(choices=language_list, label="📤 Source Language", value="English")
-            tgt_dropdown = gr.Dropdown(choices=language_list, label="📥 Target Language", value="French")
-            gr.Markdown("<br>")
-            translate_button = gr.Button("🔄 Translate", variant="primary")
-        with gr.Column(scale=1):
-            output_text = gr.Textbox(label="✅ Translated Text", lines=6, interactive=False)
-            audio_output = gr.Audio(label="🔊 Speak Translation", type="filepath", autoplay=True)
+        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")
+
+        with gr.Column():
+            translated_text = gr.Textbox(label="Translated Text")
+            audio_output = gr.Audio(label="Speak Translation", type="filepath")
 
-    translate_button.click(fn=translate_and_speak,
-                           inputs=[input_text, src_dropdown, tgt_dropdown],
-                           outputs=[output_text, audio_output])
+    translate_btn.click(
+        translate_text,
+        inputs=[src_text, src_lang, tgt_lang],
+        outputs=[translated_text, audio_output]
+    )
 
-# Launch
 demo.launch()