Update app.py

app.py

@@ -14,7 +14,7 @@ def text2story(text):
     generated = story_generator(prompt, max_new_tokens=150, do_sample=True)
     story_text = generated[0]['generated_text']
 
-    # Remove leading ". " if present
+    # Remove leading punctuation if present
     story_text = re.sub(r'^\s*[.,!?;:]+\s*', '', story_text)
     story_text = story_text.lstrip()
     
@@ -55,25 +55,31 @@ def text2audio(story_text):
     audio_output = audio_generator(story_text)
     return audio_output
 
-st.set_page_config(page_title="Once Upon A Time - Storytelling Application", page_icon="📖")
-st.header("Create a story of yours with an image!🧙")
-uploaded_file = st.file_uploader("Upload an image for creating your story!")
+def main():
+    st.set_page_config(page_title="Once Upon A Time - Storytelling Application", page_icon="📖")
+    st.header("Create a story of yours with 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_container_width=True)
+    uploaded_file = st.file_uploader("Upload an image for creating your story!")
 
-    st.text('Entering the scene...🏰')
-    scenario = img2text(uploaded_file.name)
-    st.write(scenario)
+    if uploaded_file is not None:
+        # Save the file
+        bytes_data = uploaded_file.getvalue()
+        with open(uploaded_file.name, "wb") as file:
+            file.write(bytes_data)
 
-    st.text('Your story is going to begin...🦄')
-    story = text2story(scenario)
-    st.write(story)
+        st.image(uploaded_file, caption="Uploaded Image", use_container_width=True)
 
-    st.text('Your story is going to be told...🎧')
-    audio_data = text2audio(story)
-    st.audio(audio_data['audio'], format="audio/wav", start_time=0, sample_rate = audio_data['sampling_rate'])
+        st.text('Entering the scene...🏰')
+        scenario = img2text(uploaded_file.name)
+        st.write(scenario)
+
+        st.text('Your story is going to begin...🦄')
+        story = text2story(scenario)
+        st.write(story)
+
+        st.text('Your story is going to be told...🎧')
+        audio_data = text2audio(story)
+        st.audio(audio_data['audio'], format="audio/wav", start_time=0, sample_rate=audio_data['sampling_rate'])
+
+if __name__ == "__main__":
+    main()