Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 from PIL import Image
+from gtts import gTTS
+from io import BytesIO
 import io
 from openai import OpenAI
 #from dotenv import load_dotenv
@@ -21,7 +23,7 @@ uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "pn
 if 'history' not in st.session_state:
     st.session_state['history'] = []
 
-personality = st.text_input("Enter a personality to give to the narrator")
+personality = st.text_input("Enter a personality")
 image_narration = "No narration given"
 
 # Check if an image has been uploaded
@@ -61,5 +63,11 @@ def update_and_get_narration(personality, user_input):
 if st.button('Narrate'):
     narration = update_and_get_narration(personality, image_narration)
     st.write(narration)
+    tts = gTTS(text=narration, lang='en')
+    audio_buffer = BytesIO()
+    tts.write_to_fp(audio_buffer)
+    audio_buffer.seek(0)
+
+    st.audio(audio_buffer, format='audio/mp3', start_time=0)
 else:
     st.write(st.session_state['history'][-1]['content'] if st.session_state['history'] else "Narration will appear here.")