Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import classla
 classla.download('hr')
 nlp = classla.Pipeline('hr')
+
 # def greet(name):
 #     doc = nlp(name)
 #     #print(doc.to_conll())
@@ -12,10 +13,13 @@ nlp = classla.Pipeline('hr')
 
 import gradio as gr
 from gtts import gTTS
-import pygame
+import wavio
+
 
-# Initialize the pygame mixer
-pygame.mixer.init()
+from io import BytesIO
+import base64
+
+    
 
 # Function to translate text
 def parse_text(text):
@@ -26,8 +30,16 @@ def parse_text(text):
 def play_original_audio(text_to_play):
     tts = gTTS(text_to_play, lang='hr')  # Assuming the text is in English
     tts.save('original_audio.mp3')
-    pygame.mixer.music.load('original_audio.mp3')
-    pygame.mixer.music.play()
+    
+    audio_bytes = BytesIO()
+    tts.write_to_fp(audio_bytes)
+    audio_bytes.seek(0)
+
+    audio = base64.b64encode(audio_bytes.read()).decode("utf-8")
+    audio_player = f'<audio src="data:audio/mpeg;base64,{audio}" controls autoplay></audio>'
+
+    return audio_player
+    
 
 # Create Gradio interface components
 text_input = gr.inputs.Textbox(lines=2, label="Enter text to parse")
@@ -56,6 +68,16 @@ def play_audio_button_click():
 translate_button.command = translate_button_click
 play_audio_button.command = play_audio_button_click
 
+def audio_to_html(audio):
+    audio_bytes = BytesIO()
+    wavio.write(audio_bytes, audio[1].astype(np.float32), audio[0], sampwidth=4)
+    audio_bytes.seek(0)
+
+    audio_base64 = base64.b64encode(audio_bytes.read()).decode("utf-8")
+    audio_player = f'<audio src="data:audio/mpeg;base64,{audio_base64}" controls autoplay></audio>'
+
+    return audio_player
+    
 # Launch the Gradio interface
 iface.launch()