Debugging

app.py

@@ -1,4 +1,3 @@
-#app.py
 import gradio as gr
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, pipeline
@@ -7,7 +6,11 @@ import requests
 from datetime import datetime
 import tempfile
 from gtts import gTTS
+import os
+import logging
 
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -125,14 +128,14 @@ def transcribe_audio(audio_path):
 # --- Handle audio translation ---
 def handle_audio_translation(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
 
     transcription, error = transcribe_audio(audio_path)
     if error:
-        return error, "", None
+        return error, "", "", None
         
-    translated, _ = translate(transcription, direction)
-    return "", transcription, translated
+    translated, _ = translate(transcription, direction, generate_tts=False)
+    return "", transcription, translated, None
 
 # --- Switch UI direction ---
 def switch_direction(direction, input_text_val, output_text_val):
@@ -143,7 +146,13 @@ def switch_direction(direction, input_text_val, output_text_val):
 
 # === Gradio Interface ===
 with gr.Blocks() as interface:
-    gr.HTML("<h2 style='text-align:center;'>English ↔ Kashmiri Translator</h2>")
+    gr.HTML("""
+<div style="display: flex; justify-content: space-between; align-items: center; padding: 10px;">
+    <img src="https://raw.githubusercontent.com/BurhaanRasheedZargar/Images/211321a234613a9c3dd944fe9367cf13d1386239/assets/left_logo.png" style="height:150px; width:auto;">
+    <h2 style="margin: 0; text-align: center;">English ↔ Kashmiri Translator</h2>
+    <img src="https://raw.githubusercontent.com/BurhaanRasheedZargar/Images/77797f7f7cbee328fa0f9d31cf3e290441e04cd3/assets/right_logo.png">
+</div>
+""")
     translation_direction = gr.State(value="en_to_ks")
 
     with gr.Row():
@@ -198,7 +207,7 @@ with gr.Blocks() as interface:
     stt_button.click(
         fn=handle_audio_translation,
         inputs=[audio_input, translation_direction],
-        outputs=[save_status, input_text, output_text]
+        outputs=[save_status, input_text, output_text, audio_output]
     )
 
 if __name__ == "__main__":