Update app.py

app.py

@@ -46,6 +46,10 @@ def text2story(text):
     # Remove <BOS> and <superhero> tags from the generated story
     story = story.replace("<BOS>", "").replace("<superhero>", "").strip()
     
+    # Remove the input text (scenario) from the generated story
+    if text in story:
+        story = story.replace(text, "").strip()
+    
     return story
 
 # Function to convert text to audio using gTTS
@@ -90,7 +94,7 @@ if uploaded_file is not None:
         file.write(bytes_data)
 
     # Display the uploaded image
-    st.image(uploaded_file, caption="Your awesome picture!", use_container_width=True)
+    st.image(uploaded_file, caption="Your awesome picture!", use_column_width=True)
 
     # Stage 1: Image to Text
     st.text('✨ Turning your picture into words...')
@@ -111,10 +115,11 @@ if uploaded_file is not None:
     
     # Play button for the generated audio
     if st.button("🎵 **Play Audio**"):
-        st.audio(st.session_state.audio_file, format="audio/mp3")
+        if os.path.exists(st.session_state.audio_file):
+            st.audio(st.session_state.audio_file, format="audio/mp3")
+        else:
+            st.error("Audio file not found. Please try again.")
 
     # Clean up the generated audio file and uploaded image
-    if os.path.exists(st.session_state.audio_file):
-        os.remove(st.session_state.audio_file)
     if os.path.exists(uploaded_file.name):
         os.remove(uploaded_file.name)