Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from transformers import pipeline, MarianTokenizer, AutoModelForSeq2SeqLM
 import torch
-import unicodedata
 import re
 import whisper
 import tempfile
@@ -118,23 +117,19 @@ def extract_text_from_file(uploaded_file):
         else:
             raise ValueError("Unsupported file type")
 
-# --- Main Gradio Function ---
-def process(input_mode, target_lang, text_input, audio_input, file_input):
+# --- Main Function ---
+def process(target_lang, text_input, audio_input, file_input):
     input_text = ""
 
-    if input_mode == "Text" and text_input:
+    if text_input and text_input.strip():
         input_text = text_input
-    elif input_mode == "Audio" and audio_input:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
-            tmp.write(audio_input.read())
-            tmp_path = tmp.name
+    elif audio_input:
         input_text = transcribe_audio(audio_input)
-        os.remove(tmp_path)
-    elif input_mode == "File" and file_input:
+    elif file_input:
         input_text = extract_text_from_file(file_input)
 
     if not input_text.strip():
-        return "", "No input text provided."
+        return "", "No valid input provided."
 
     translated_text = translate(input_text, target_lang)
     return input_text, translated_text
@@ -143,21 +138,22 @@ def process(input_mode, target_lang, text_input, audio_input, file_input):
 with gr.Blocks() as demo:
     gr.Markdown("## 🌐 LocaleNLP Translator — English ↔ Darija / Hausa / Wolof")
 
-    with gr.Row():
-        input_mode = gr.Dropdown(["Text", "Audio", "File"], label="Select input mode")
-        target_lang = gr.Dropdown(["Darija (Morocco)", "Hausa (Nigeria)", "Wolof (Senegal)"], label="Select target language")
+    target_lang = gr.Dropdown(
+        ["Darija (Morocco)", "Hausa (Nigeria)", "Wolof (Senegal)"], 
+        label="Select target language"
+    )
 
     with gr.Row():
-        text_input = gr.Textbox(label="Enter English text", lines=10)
-        audio_input = gr.Audio(type="filepath", label="Upload Audio")
-        file_input = gr.File(label="Upload Document")
+        text_input = gr.Textbox(label="✏️ Enter English text", lines=10)
+        audio_input = gr.Audio(type="filepath", label="🔊 Upload Audio")
+        file_input = gr.File(label="📄 Upload Document")
 
     with gr.Row():
         extracted_text = gr.Textbox(label="Extracted / Transcribed Text", lines=10)
         translated_output = gr.Textbox(label="Translated Text", lines=10)
 
     run_btn = gr.Button("Translate")
-    run_btn.click(process, inputs=[input_mode, target_lang, text_input, audio_input, file_input], outputs=[extracted_text, translated_output])
+    run_btn.click(process, inputs=[target_lang, text_input, audio_input, file_input], outputs=[extracted_text, translated_output])
 
 if __name__ == "__main__":
     demo.launch()