Update app.py

app.py

@@ -81,7 +81,6 @@ def generate_story(descriptions):
 # Main function to control the flow
 def main():
     st.title("Media Story Generator")
-    all_output_texts = []
 
     # Step 1: Load the model
     processor, model, device = load_model()
@@ -96,9 +95,10 @@ def main():
         if user_question:
             # Step 4: Generate description
             st.write("Step 4: Generate description")
-            generate_description_button = st.button("Generate Description")
+            generate_description_button = st.button("Generate Description", key="generate_description")
 
             if generate_description_button:
+                all_output_texts = []
 
                 for uploaded_file in uploaded_files:
                     file_type = uploaded_file.type.split('/')[0]
@@ -122,10 +122,11 @@ def main():
                     del image
                     torch.cuda.empty_cache()
                     torch.manual_seed(0)
-
+                
                     if all_output_texts:
                         # Step 5: Generate story
-                        generate_story_button = st.button("Generate Story")
+                        st.write("Step 5: Generate story")
+                        generate_story_button = st.button("Generate Story", key="generate_story")
     
                         if generate_story_button:
                             story = generate_story(all_output_texts)