File size: 3,448 Bytes
68f0220 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import torch
import gradio as gr
import json
from transformers import pipeline
import pyttsx3
import tempfile
import os
# Load the NLLB model
# model_path = "../text_translator/model/models--facebook--nllb-200-distilled-600M/snapshots/f8d333a098d19b4fd9a8b18f94170487ad3f821d"
# translator = pipeline("translation", model=model_path)
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)
# 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
# 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)
if not src_code or not tgt_code:
return "Invalid language selected.", None
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
except Exception as e:
return f"Translation error: {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>")
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)
translate_button.click(fn=translate_and_speak,
inputs=[input_text, src_dropdown, tgt_dropdown],
outputs=[output_text, audio_output])
# Launch
demo.launch()
|