Update handler.py

handler.py

@@ -4,9 +4,7 @@ import torch
 import soundfile as sf
 from transformers import AutoTokenizer, AutoModelForTextToWaveform
 import cloudinary.uploader
-import tkinter as tk
-from tkinter import ttk
-import pygame
+import streamlit as st
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
 # Configure logging
@@ -36,7 +34,7 @@ class EndpointHandler():
         # Save the audio to a file
         sf.write("StoryAudio.wav", outputs["waveform"][0].numpy(), self.model.config.sampling_rate)
         uploadGraphFile("StoryAudio.wav")
-        playAudioFile()
+        st.audio("StoryAudio.wav")
         #return 'StoryAudio.wav'
         # Check if the request was successful
         
@@ -50,27 +48,3 @@ def uploadGraphFile(fileName):
     # Upload a file to Cloudinary
     result = cloudinary.uploader.upload(fileName, folder="poc-graph", resource_type="raw")
     return result
-
-def play_audio():
-    pygame.mixer.music.load("StoryAudio.wav")
-    pygame.mixer.music.play()
-
-def stop_audio():
-    pygame.mixer.music.stop()
-
-def playAudioFile():
-    root = tk.Tk()
-    root.title("Audio Player")
-
-    # Create a play button
-    play_button = tk.Button(root, text="Play", command=play_audio)
-    play_button.pack()
-
-    # Create a stop button
-    stop_button = tk.Button(root, text="Stop", command=stop_audio)
-    stop_button.pack()
-
-    # Create a progress bar
-    progress_bar = ttk.Progressbar(root, orient="horizontal", length=200, mode="determinate")
-    progress_bar.pack()
-    root.mainloop()