Update app.py

app.py

@@ -29,7 +29,7 @@ def get_user_question():
 # Process image
 def process_image(uploaded_file):
     image = Image.open(uploaded_file)
-    image = image.resize((512, 512))  # Reduce size to save memory
+    image = image.resize((256,256))  # Reduce size to save memory
     st.image(image, caption='Uploaded Image.', use_column_width=True)
     return image
 
@@ -117,17 +117,19 @@ def main():
                         st.write(f"Description for file {idx + 1}:")
                         st.write(description)
                         all_output_texts.append(description)
-
-                st.write("Description of all images:")
-                st.write(all_output_texts)
-                if all_output_texts:
-                    st.write("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)
+                
+                # Store descriptions in session state
+                st.session_state["all_output_texts"] = all_output_texts
+
+    # Check if descriptions are available in session state
+    if "all_output_texts" in st.session_state and st.session_state["all_output_texts"]:
+        st.write("Generate story")
+        generate_story_button = st.button("Generate Story", key="generate_story")
+
+        if generate_story_button:
+            story = generate_story(st.session_state["all_output_texts"])
+            st.write("Generated Story:")
+            st.write(story)
 
 if __name__ == "__main__":
     main()