Update app.py

app.py

@@ -35,14 +35,15 @@ def txt2story(img_text):
         {"role": "user", "content": f"Here is input set of words: {img_text}", "temperature": 1},
     ]
     
-    print("Generating story completion using the AI model...")
+    print("Story...")
     chat_completion = client.chat.completions.create(
         messages=messages,
         model="mistralai/Mixtral-8x7B-Instruct-v0.1",
         max_tokens=200)
-
-    return chat_completion.choices[0].message.content
+    
     print(chat_completion.choices[0].message.content)
+    return chat_completion.choices[0].message.content
+
 
 
 # Text-to-speech
@@ -52,16 +53,12 @@ def txt2speech(text):
     headers = {"Authorization": f"Bearer {os.environ['HUGGINGFACEHUB_API_TOKEN']}"}
     payloads = {'inputs': text}
 
-    print("Sending request for speech synthesis...")
     response = requests.post(API_URL, headers=headers, json=payloads)
     
-    print("Saving synthesized speech to audio file...")
     with open('audio_story.mp3', 'wb') as file:
         file.write(response.content)
-    
-    print("Text-to-speech conversion completed.")
-
 
+        
 # Streamlit web app main function
 def main():
     st.set_page_config(page_title="🎨 Image-to-Audio Story 🎧", page_icon="🖼️")