Update app.py

app.py

@@ -24,7 +24,7 @@ def generate_story_from_caption(caption):
     # Generate story based on caption
     api_key = os.getenv("GOOGLE_API")
     prompt_template = """You are a story teller;
-    You can generate a short story based on a simple narrative, the story should between 30 to 50 words;
+    You can generate a short story based on a simple narrative, the story should between 30 to 80 words;
     CONTEXT: {scenario}
     Story: """
     PROMPT = PromptTemplate(template=prompt_template, input_variables=["scenario"])
@@ -35,10 +35,15 @@ def generate_story_from_caption(caption):
     return story
 
 def text_to_speech(text):
-    # Convert text to speech
-    tts = gTTS(text=text, lang='en')
-    tts.save("output.mp3")
-    return "output.mp3"
+    headers = {"Authorization": f"Bearer {os.getenv('HUGGING_FACE')}"}
+    payload = {"inputs": text}
+    API_URL = "https://api-inference.huggingface.co/models/facebook/mms-tts-eng"
+    response = requests.post(API_URL, headers=headers, json=payload)
+    
+    if response.status_code == 200:
+        with open("output.mp3", "wb") as f:
+            f.write(response.content)
+        return "output.mp3"
 
 def generate_story_from_image(image_input):
     input_image = Image.fromarray(image_input)