update UI

app.py

@@ -119,40 +119,39 @@ with gr.Blocks(title="Nepali-English Translator") as demo:
     gr.Markdown("# Nepali-English Translation Service")
     gr.Markdown("Translate between Nepali and English, transcribe audio, and convert text to speech.")
     
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("### Text Translation")
-            text_input = gr.Textbox(label="Input Text", lines=5)
-            model_choice = gr.Radio(
-                choices=["XTransformer", "M2M100"], 
-                value="XTransformer", 
-                label="Translation Model"
-            )
-            source_lang = gr.Dropdown(
-                choices=["Auto-detect", "en", "ne"], 
-                value="Auto-detect", 
-                label="Source Language"
-            )
-            target_lang = gr.Dropdown(
-                choices=["Auto-select", "en", "ne"], 
-                value="Auto-select", 
-                label="Target Language"
-            )
-            translate_button = gr.Button("Translate")
-            translation_output = gr.Textbox(label="Translation Output", lines=5)
-            tts_button = gr.Button("Convert to Speech")
-            audio_output = gr.Audio(label="Audio Output")
+    with gr.Column():
+        gr.Markdown("### Speech to Text")
+        audio_input = gr.Audio(label="Upload or Record Audio", type="filepath")
+        asr_language = gr.Radio(
+            choices=["eng", "npi"], 
+            value="npi", 
+            label="Speech Language"
+        )
+        transcribe_button = gr.Button("Transcribe")
+        transcription_output = gr.Textbox(label="Transcription Output", lines=3)
         
-        with gr.Column():
-            gr.Markdown("### Speech to Text")
-            audio_input = gr.Audio(label="Upload or Record Audio", type="filepath")
-            asr_language = gr.Radio(
-                choices=["eng", "npi"], 
-                value="npi", 
-                label="Speech Language"
-            )
-            transcribe_button = gr.Button("Transcribe")
-            transcription_output = gr.Textbox(label="Transcription Output", lines=3)
+        gr.Markdown("### Text Translation")
+        model_choice = gr.Dropdown(
+            choices=["XTransformer", "M2M100"], 
+            value="M2M100", 
+            label="Translation Model"
+        )
+        source_lang = gr.Dropdown(
+            choices=["Auto-detect", "en", "ne"], 
+            value="Auto-detect", 
+            label="Source Language"
+        )
+        target_lang = gr.Dropdown(
+            choices=["Auto-select", "en", "ne"], 
+            value="Auto-select", 
+            label="Target Language"
+        )
+        translate_button = gr.Button("Translate")
+        translation_output = gr.Textbox(label="Translation Output", lines=5)
+        
+        gr.Markdown("### Text to Speech")
+        tts_button = gr.Button("Convert to Speech")
+        audio_output = gr.Audio(label="Audio Output")
     
     # Define event handlers
     def process_translation(text, model, src_lang, tgt_lang):
@@ -171,9 +170,15 @@ with gr.Blocks(title="Nepali-English Translator") as demo:
         return transcribe_audio(audio_path, language)
     
     # Connect the components
+    transcribe_button.click(
+        process_transcription,
+        inputs=[audio_input, asr_language],
+        outputs=transcription_output
+    )
+    
     translate_button.click(
         process_translation, 
-        inputs=[text_input, model_choice, source_lang, target_lang], 
+        inputs=[transcription_output, model_choice, source_lang, target_lang], 
         outputs=translation_output
     )
     
@@ -182,12 +187,6 @@ with gr.Blocks(title="Nepali-English Translator") as demo:
         inputs=translation_output,
         outputs=audio_output
     )
-    
-    transcribe_button.click(
-        process_transcription,
-        inputs=[audio_input, asr_language],
-        outputs=transcription_output
-    )
 
 # Launch the app
 if __name__ == "__main__":