Update utils.py

utils.py

@@ -46,12 +46,16 @@ def txt2story(prompt: str, top_k: int, top_p: float, temperature: float) -> str:
     return story
 
 # Text-to-speech
-def txt2speech(text: str):
-    print("Initializing text-to-speech conversion...")
-
-    audio_model = pipeline("text-to-speech", model="myshell-ai/MeloTTS-English")
-    speech = captioning_model(text, max_new_tokens=200)[0]["generated_speech"]
-    return speech
+def txt2speech(text: str) -> None:
+    print("Initializing MeloTTS text-to-speech conversion...")
+    pipe = pipeline("text-to-speech", model="myshell-ai/MeloTTS-English")
+    
+    # Generate audio from the text
+    audio = pipe(text)
+    
+    # Save the audio output to a file
+    with open("audio_story.wav", "wb") as file:
+        file.write(audio["wav"])
 
 # Get user preferences for the story
 def get_user_preferences() -> Dict[str, str]: