Update app.py

app.py

@@ -20,18 +20,16 @@ def load_model():
 
 # Step 2: Upload image or video
 def upload_media():
-    st.write("Step 2: Upload an image or video")
     return st.file_uploader("Choose images or videos...", type=["jpg", "jpeg", "png", "mp4", "avi", "mov"], accept_multiple_files=True)
 
 # Step 3: Enter your question
 def get_user_question():
-    st.write("Step 3: Enter your question")
     return st.text_input("Ask a question about the images or videos:")
 
 # Process image
 def process_image(uploaded_file):
     image = Image.open(uploaded_file)
-    image = image.resize((256, 256))  # Reduce size to save memory
+    image = image.resize((512, 512))  # Reduce size to save memory
     st.image(image, caption='Uploaded Image.', use_column_width=True)
     return image
 
@@ -82,7 +80,8 @@ def generate_story(descriptions):
 
 # Main function to control the flow
 def main():
-    st.title("Media Description Generator")
+    st.title("Media Story Generator")
+    all_output_texts = []
 
     # Step 1: Load the model
     processor, model, device = load_model()
@@ -100,7 +99,6 @@ def main():
             generate_description_button = st.button("Generate Description")
 
             if generate_description_button:
-                all_output_texts = []
 
                 for uploaded_file in uploaded_files:
                     file_type = uploaded_file.type.split('/')[0]
@@ -125,15 +123,14 @@ def main():
                     torch.cuda.empty_cache()
                     torch.manual_seed(0)
 
-                if all_output_texts:
-                    # Step 5: Generate story
-                    st.write("Step 5: Generate story")
-                    generate_story_button = st.button("Generate Story")
+        if all_output_texts:
+            # Step 5: Generate story
+            generate_story_button = st.button("Generate Story")
 
-                    if generate_story_button:
-                        story = generate_story(all_output_texts)
-                        st.write("Generated Story:")
-                        st.write(story)
+            if generate_story_button:
+                story = generate_story(all_output_texts)
+                st.write("Generated Story:")
+                st.write(story)
 
 if __name__ == "__main__":
     main()