Update app.py

app.py

@@ -22,31 +22,41 @@ def text2audio(story_text):
      
 
 #main part
-
-st.set_page_config(page_title="Your Image to Audio Story",
-                   page_icon="🦜")
-st.header("Turn Your Image to Audio Story")
-uploaded_file = st.file_uploader("Select an Image...")
-
-if uploaded_file is not None:
-    print(uploaded_file)
-    bytes_data = uploaded_file.getvalue()
-    with open(uploaded_file.name, "wb") as file:
-        file.write(bytes_data)
-
-    st.image(uploaded_file, caption="Uploaded Image",
-             use_column_width=True)
-
-    #Stage 1: Image to Text
-    st.text('Processing img2text...')
-    scenario = img2text(uploaded_file.name)
-    st.write(scenario)
-
-    #Stage 2: Text to Story
-    st.text('Generating a story...')
-    story = text2story(scenario)
-    st.write(story)
-
-    #Stage 3: Story to Audio data
-    st.text('Generating audio data...')
-    audio_data =text2audio(story)
+def main():
+    st.set_page_config(page_title="Your Image to Audio Story",
+                       page_icon="🦜")
+    st.header("Turn Your Image to Audio Story")
+    uploaded_file = st.file_uploader("Select an Image...")
+    
+    if uploaded_file is not None:
+        print(uploaded_file)
+        bytes_data = uploaded_file.getvalue()
+        with open(uploaded_file.name, "wb") as file:
+            file.write(bytes_data)
+    
+        st.image(uploaded_file, caption="Uploaded Image",
+                 use_column_width=True)
+    
+        #Stage 1: Image to Text
+        st.text('Processing img2text...')
+        scenario = img2text(uploaded_file.name)
+        st.write(scenario)
+    
+        #Stage 2: Text to Story
+        st.text('Generating a story...')
+        story = text2story(scenario)
+        st.write(story)
+    
+        #Stage 3: Story to Audio data
+        st.text('Generating audio data...')
+        audio_data =text2audio(story)
+
+        # Play button
+        if st.button("Play Audio"):
+            st.audio(audio_data['audio'], 
+                     format="audio/wav", 
+                     start_time=0, 
+                     sample_rate = audio_data['sampling_rate'])
+
+if __name__ == "__main__":
+    main()