Update handler.py

handler.py

@@ -11,7 +11,8 @@ logging.basicConfig(level=logging.ERROR)
 # Configure logging
 logging.basicConfig(level=logging.WARNING)
 
-
+import streamlit as st
+import pygame
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -32,8 +33,21 @@ class EndpointHandler():
 
         # Save the audio to a file
         sf.write("StoryAudio.wav", outputs["waveform"][0].numpy(), self.model.config.sampling_rate)
-        
-        return 'StoryAudio.wav'
+        st.audio("StoryAudio.wav")
+        # pygame.mixer.init()
+        # pygame.mixer.music.load("StoryAudio.wav")
+        # pygame.mixer.music.play()
+        #return 'StoryAudio.wav'
+        #return {"audio_file_path": 'StoryAudio.wav'}
+        # with open("StoryAudio.wav", "rb") as f:
+        #         audio_content = f.read()
+            
+        #     # Set the response content type
+        # response_headers = {"Content-Type": "audio/wav"}
+            
+            # Return the binary content of the audio file along with the response headers
+        #return send_file("StoryAudio.wav", mimetype="audio/wav", as_attachment=True)
+
         # Check if the request was successful