Update handler.py

handler.py

@@ -11,7 +11,9 @@ logging.basicConfig(level=logging.ERROR)
 # Configure logging
 logging.basicConfig(level=logging.WARNING)
 
-import pygame
+from pydub import AudioSegment
+from pydub.playback import play
+
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -32,21 +34,11 @@ class EndpointHandler():
 
         # Save the audio to a file
         sf.write("StoryAudio.wav", outputs["waveform"][0].numpy(), self.model.config.sampling_rate)
-        #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)
+        audio = AudioSegment.from_wav("StoryAudio.wav")
 
+        # Play the audio file
+        play(audio)
+        return 'StoryAudio.wav'
         # Check if the request was successful