Debugging

app.py

@@ -1,3 +1,4 @@
+#app.py
 import gradio as gr
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
@@ -67,7 +68,7 @@ def get_translation_history(direction):
 # --- Translation ---
 def translate(text, direction):
     if not text.strip():
-        return "Enter some text.", gr.update(), gr.update()
+        return "Enter some text.", None
 
     if direction == "en_to_ks":
         src_lang, tgt_lang = "eng_Latn", "kas_Arab"
@@ -83,12 +84,12 @@ def translate(text, direction):
             output = model.generate(**tokens, max_length=256, num_beams=5)
         result = tokenizer.batch_decode(output, skip_special_tokens=True)
         final = ip.postprocess_batch(result, lang=tgt_lang)[0]
-        return final, gr.update(), gr.update()
+        return final, None
     except Exception as e:
         print("Translation error:", e)
-        return "⚠️ Translation failed.", gr.update(), gr.update()
+        return "⚠️ Translation failed.", None
 
-# --- TTS for English output only ---
+# --- TTS for English output ---
 def synthesize_tts(text):
     try:
         tts = gTTS(text=text, lang="en")
@@ -100,25 +101,31 @@ def synthesize_tts(text):
         return None
 
 # --- STT only for en_to_ks ---
-def handle_audio_translation(audio_path, direction):
+def generate_stt_for_input(audio_path, direction):
     if direction != "en_to_ks":
-        return "⚠️ Audio input is only supported for English to Kashmiri.", "", "", None
+        return "⚠️ Audio input is only supported for English to Kashmiri.", "", None
 
     try:
         transcription = asr(audio_path)["text"]
     except Exception as e:
         print("STT error:", e)
-        return "⚠️ Transcription failed.", "", "", None
+        return "⚠️ Transcription failed.", "", None
 
-    translated, _, _ = translate(transcription, direction)
-    return transcription, translated, transcription, None
+    translated, _ = translate(transcription, direction)
+    return transcription, translated, None
+
+# --- Generate TTS for English output ---
+def generate_tts_for_output(output_text, direction):
+    if direction == "ks_to_en" and output_text.strip():
+        return synthesize_tts(output_text)
+    return None
 
 # --- Switch UI direction ---
 def switch_direction(direction, input_text_val, output_text_val):
     new_direction = "ks_to_en" if direction == "en_to_ks" else "en_to_ks"
     input_label = "Kashmiri Text" if new_direction == "ks_to_en" else "English Text"
     output_label = "English Translation" if new_direction == "ks_to_en" else "Kashmiri Translation"
-    return new_direction, gr.update(value=output_text_val, label=input_label), gr.update(value=input_text_val, label=output_label)
+    return new_direction, gr.update(value=output_text_val, label=input_label), gr.update(value=input_text_val, label=output_label), None
 
 # === Gradio Interface ===
 with gr.Blocks() as interface:
@@ -126,8 +133,8 @@ with gr.Blocks() as interface:
     translation_direction = gr.State(value="en_to_ks")
 
     with gr.Row():
-        input_text = gr.Textbox(label="English Text", placeholder="Enter text here...")
-        output_text = gr.Textbox(label="Kashmiri Translation", placeholder="Translated text...")
+        input_text = gr.Textbox(label="English Text", placeholder="Enter text here...", lines=2)
+        output_text = gr.Textbox(label="Kashmiri Translation", placeholder="Translated text...", lines=2)
 
     with gr.Row():
         translate_button = gr.Button("Translate")
@@ -139,21 +146,45 @@ with gr.Blocks() as interface:
 
     with gr.Row():
         audio_input = gr.Audio(type="filepath", label="🎙️ Record English audio")
-        audio_output = gr.Audio(label="🔊 English TTS")
+        audio_output = gr.Audio(label="🔊 English TTS", interactive=False)
 
     stt_button = gr.Button("🎤 Transcribe & Translate (EN → KS Only)")
+    tts_button = gr.Button("🔊 Generate English Speech (KS → EN Only)")
 
     # Events
-    translate_button.click(fn=translate, inputs=[input_text, translation_direction], outputs=[output_text, input_text, output_text])
-
-    save_button.click(fn=save_to_supabase, inputs=[input_text, output_text, translation_direction], outputs=save_status)\
-               .then(fn=get_translation_history, inputs=translation_direction, outputs=history)
-
-    switch_button.click(fn=switch_direction, inputs=[translation_direction, input_text, output_text],
-                        outputs=[translation_direction, input_text, output_text])
-
-    stt_button.click(fn=handle_audio_translation, inputs=[audio_input, translation_direction],
-                     outputs=[input_text, output_text, input_text, audio_output])
+    translate_button.click(
+        fn=translate,
+        inputs=[input_text, translation_direction],
+        outputs=[output_text, audio_output]
+    )
+    
+    tts_button.click(
+        fn=generate_tts_for_output,
+        inputs=[output_text, translation_direction],
+        outputs=audio_output
+    )
+
+    save_button.click(
+        fn=save_to_supabase,
+        inputs=[input_text, output_text, translation_direction],
+        outputs=save_status
+    ).then(
+        fn=get_translation_history,
+        inputs=translation_direction,
+        outputs=history
+    )
+
+    switch_button.click(
+        fn=switch_direction,
+        inputs=[translation_direction, input_text, output_text],
+        outputs=[translation_direction, input_text, output_text, audio_output]
+    )
+
+    stt_button.click(
+        fn=generate_stt_for_input,
+        inputs=[audio_input, translation_direction],
+        outputs=[input_text, output_text, audio_output]
+    )
 
 if __name__ == "__main__":
-    interface.queue().launch()
+    interface.queue().launch(share=True)