Update app.py

app.py

@@ -43,6 +43,9 @@ def text2story(text):
     prompt = f"<BOS> <superhero> {text}"
     story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
     
+    # Remove <BOS> and <superhero> tags from the generated story
+    story = story.replace("<BOS>", "").replace("<superhero>", "").strip()
+    
     return story
 
 # Function to convert text to audio using gTTS
@@ -101,12 +104,17 @@ if uploaded_file is not None:
 
     # Stage 3: Story to Audio
     st.text('🎧 Turning your story into audio...')
-    audio_file = text2audio(story)
-
+    
+    # Use session state to avoid regenerating audio on button click
+    if 'audio_file' not in st.session_state:
+        st.session_state.audio_file = text2audio(story)
+    
     # Play button for the generated audio
     if st.button("🎵 **Play Audio**"):
-        st.audio(audio_file, format="audio/mp3")
+        st.audio(st.session_state.audio_file, format="audio/mp3")
 
     # Clean up the generated audio file and uploaded image
-    os.remove(audio_file)
-    os.remove(uploaded_file.name)
+    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)