Update app.py

app.py

@@ -1,74 +1,67 @@
-# Text → Translate → Speech | CPU-only Hugging-Face Space
+# Text → Translate → Speech  |  CPU-only Hugging-Face Space
 
 import tempfile
 from pathlib import Path
 
-import torch
-import TTS.utils.radam as tts_radam
 from transformers import pipeline
 from TTS.api import TTS
 import gradio as gr
 
-# ─────────────────────────────
-# 1. Allow-list Coqui-TTS custom class  (PyTorch ≥ 2.6)
-# ─────────────────────────────
-torch.serialization.add_safe_globals({tts_radam.RAdam})  # ← must pass the class object
-
-# ─────────────────────────────
-# 2. Translation pipelines
-# ─────────────────────────────
+# ──────────────────────────
+# Translation pipelines
+# ──────────────────────────
 PIPE_EN_ES = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es", device=-1)
 PIPE_ES_EN = pipeline("translation", model="Helsinki-NLP/opus-mt-es-en", device=-1)
 
-# ─────────────────────────────
-# 3. TTS models
-# ─────────────────────────────
+# ──────────────────────────
+# TTS models
+# ──────────────────────────
 TTS_EN = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False)
 TTS_ES = TTS(model_name="tts_models/es/mai/tacotron2-DDC", progress_bar=False)
 
-# ─────────────────────────────
-# 4. Helper: synthesize WAV
-# ─────────────────────────────
-def synthesize(tts_model: TTS, text: str) -> str:
+# ──────────────────────────
+# Helper – save speech to WAV
+# ──────────────────────────
+def wav_file(tts_obj: TTS, text: str) -> str:
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as tmp:
-        wav_path = Path(tmp.name)
-    tts_model.tts_to_file(text=text, file_path=wav_path)
-    return str(wav_path)
+        path = Path(tmp.name)
+    tts_obj.tts_to_file(text=text, file_path=path)
+    return str(path)
 
-# ─────────────────────────────
-# 5. Core translate-and-speak
-# ─────────────────────────────
-def translate_and_speak(txt: str, target_lang: str):
-    if not txt.strip():
+# ──────────────────────────
+# Core translate-and-speak
+# ──────────────────────────
+def translate_and_speak(text: str, target: str):
+    if not text.strip():
         return "", None
 
-    if target_lang == "Spanish":
-        translated = PIPE_EN_ES(txt)[0]["translation_text"]
-        audio_file = synthesize(TTS_ES, translated)
+    if target == "Spanish":
+        translated = PIPE_EN_ES(text)[0]["translation_text"]
+        audio      = wav_file(TTS_ES, translated)
     else:
-        translated = PIPE_ES_EN(txt)[0]["translation_text"]
-        audio_file = synthesize(TTS_EN, translated)
+        translated = PIPE_ES_EN(text)[0]["translation_text"]
+        audio      = wav_file(TTS_EN, translated)
 
-    return translated, audio_file
+    return translated, audio
 
-# ─────────────────────────────
-# 6. Gradio UI
-# ─────────────────────────────
+# ──────────────────────────
+# Gradio UI
+# ──────────────────────────
 with gr.Blocks(title="Translator & TTS") as demo:
     gr.Markdown(
         "# 🌐💬 Text → Translate → Speech\n"
-        "Type a sentence, choose the target language, and hear it spoken."
+        "Type a sentence, choose the target language, hear it spoken."
     )
 
     sentence = gr.Textbox(label="Sentence (English or Spanish)", lines=2)
-    lang     = gr.Radio(["Spanish", "English"], value="Spanish", label="Translate to")
+    target   = gr.Radio(["Spanish", "English"], value="Spanish", label="Translate to")
 
-    run_btn  = gr.Button("Translate & Speak", variant="primary")
+    run  = gr.Button("Translate & Speak", variant="primary")
 
-    translated_txt = gr.Textbox(label="Translated text", interactive=False)
-    audio_out      = gr.Audio(label="Speech output", type="filepath")
+    out_text  = gr.Textbox(label="Translated text", interactive=False)
+    out_audio = gr.Audio(label="Speech output", type="filepath")
 
-    run_btn.click(translate_and_speak, [sentence, lang], [translated_txt, audio_out])
+    run.click(translate_and_speak, [sentence, target], [out_text, out_audio])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0")