Update app.py

app.py

@@ -1,9 +1,9 @@
-import streamlit as st
-import IPython.display as ipd
-from gtts import gTTS
+import gradio as gr
 import io
+from gtts import gTTS
 from transformers import pipeline
 
+# Define the translation and text-to-speech function
 def play_sound(text, lang):
     translator = pipeline("translation", model="Amitesh007/finetuned-eng-hi-translation")
     translation = translator(text)
@@ -11,12 +11,23 @@ def play_sound(text, lang):
     fp = io.BytesIO()
     tts.write_to_fp(fp)
     fp.seek(0)
-    return ipd.Audio(fp.read(), autoplay=True)
-
-st.title("English To Hindi Translator")
+    return fp.read(), translation[0]['translation_text']
 
-text = st.text_input("Enter some text in English to translate.\nPlay the audio 2nd time to listen clearly")
+# 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])
 
-if st.button("Translate and play sound"):
-    audio = play_sound(text, lang='hi')
-    st.write(audio)
+# Launch the interface
+demo.launch()