Update app.py

app.py

@@ -95,12 +95,12 @@ def main():
         if user_question:
             # Step 4: Generate description
             st.write("Step 4: Generate description")
-            generate_description_button = st.button("Generate Description", key="generate_description")
+            generate_description_button = st.button("Generate Descriptions", key="generate_descriptions")
 
             if generate_description_button:
                 all_output_texts = []
 
-                for uploaded_file in uploaded_files:
+                for idx, uploaded_file in enumerate(uploaded_files):
                     file_type = uploaded_file.type.split('/')[0]
                     image = None
 
@@ -114,7 +114,7 @@ def main():
 
                     if image:
                         description = generate_description(processor, model, device, image, user_question)
-                        st.write("Description:")
+                        st.write(f"Description for file {idx + 1}:")
                         st.write(description)
                         all_output_texts.append(description)
 
@@ -123,15 +123,15 @@ 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", key="generate_story")
-    
-                        if generate_story_button:
-                            story = generate_story(all_output_texts)
-                            st.write("Generated Story:")
-                            st.write(story)
+                if all_output_texts:
+                    # Step 5: 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)
+                        st.write("Generated Story:")
+                        st.write(story)
 
 if __name__ == "__main__":
     main()