Update app.py

app.py

@@ -1,33 +1,26 @@
 import gradio as gr
-import io
+import IPython.display as ipd
 from gtts import gTTS
+import io
 from transformers import pipeline
 
-# Define the translation and text-to-speech function
-def play_sound(text, lang):
+def translate_and_play_sound(text, lang='hi'):
     translator = pipeline("translation", model="Amitesh007/finetuned-eng-hi-translation")
     translation = translator(text)
     tts = gTTS(text=translation[0]['translation_text'], lang=lang)
     fp = io.BytesIO()
     tts.write_to_fp(fp)
     fp.seek(0)
-    return fp.read(), translation[0]['translation_text']
+    audio = ipd.Audio(fp.read(), autoplay=True)
+    return translation[0]['translation_text'], audio
+
+iface = gr.Interface(
+    fn=translate_and_play_sound,
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Enter text in English"),
+    outputs=[gr.outputs.Textbox(label="Translated Text"), gr.outputs.Audio(label="Translation Audio")],
+    title="English to Hindi Translator",
+    description="Enter some text in English to translate and play the translation audio."
+)
 
-# Create the Gradio Blocks interface
-with gr.Blocks() as demo:
-    gr.Markdown("# English To Hindi Translator")
-    gr.Markdown("Enter some text in English to translate and play the audio. Play the audio a second time to listen clearly.")
-    
-    with gr.Row():
-        with gr.Column():
-            input_text = gr.Textbox(lines=5, label="Input Text here....", placeholder="Type a sentence to translate and play")
-            translate_button = gr.Button("Translate and play sound")
-        
-        with gr.Column():
-            output_audio = gr.Audio(label="Translated Speech")
-            output_text = gr.Textbox(label="Translated Text")
-    
-    translate_button.click(fn=play_sound, inputs=input_text, outputs=[output_audio, output_text])
+iface.launch()
 
-# Launch the interface
-demo.launch()