Update app.py

app.py

@@ -23,9 +23,14 @@ def main():
             image = Image.open(uploaded_file)
             st.session_state.images.append(image)
 
-            # Display images
-            for idx, img in enumerate(st.session_state.images, start=1):
-                st.image(img, caption=f"Image {idx}", use_column_width=True)
+    # Display images in a grid
+    cols_per_row = 4
+    for i in range(0, len(st.session_state.images), cols_per_row):
+        cols = st.columns(cols_per_row)
+        for j in range(cols_per_row):
+            if i + j < len(st.session_state.images):
+                with cols[j]:
+                    st.image(st.session_state.images[i + j], caption=f"Image {i + j + 1}", use_column_width=True)
 
     # Reset button
     if st.button('Reset and Start Over'):