Update handler.py

handler.py

@@ -4,6 +4,9 @@ import torch
 import soundfile as sf
 from transformers import AutoTokenizer, AutoModelForTextToWaveform
 import cloudinary.uploader
+import tkinter as tk
+from tkinter import ttk
+import pygame
 
 # Configure logging
 logging.basicConfig(level=logging.DEBUG)
@@ -18,7 +21,7 @@ class EndpointHandler():
         
         self.tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-eng")
         self.model= AutoModelForTextToWaveform.from_pretrained("facebook/mms-tts-eng")
-        
+        pygame.init()
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         # Prepare the payload with input data
         logging.warning(f"------input_data-- {str(data)}")
@@ -34,6 +37,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")
+        playAudio()
         #return 'StoryAudio.wav'
         # Check if the request was successful
         
@@ -46,4 +50,28 @@ def uploadGraphFile(fileName):
     )
     # Upload a file to Cloudinary
     result = cloudinary.uploader.upload(fileName, folder="poc-graph", resource_type="raw")
-    return result
+    return result
+
+def play_audio():
+    pygame.mixer.music.load("StoryAudio.wav")
+    pygame.mixer.music.play()
+
+def stop_audio():
+    pygame.mixer.music.stop()
+
+def playAudio():
+    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()