Update app.py

app.py

@@ -1,6 +1,9 @@
 import streamlit as st
 from PIL import Image
 
+# Set the page config to make the app use the full width of the browser
+st.set_page_config(layout="wide")
+
 # Initialize a session state to keep track of images
 if 'images' not in st.session_state:
     st.session_state.images = []
@@ -16,12 +19,12 @@ def main():
     # Image uploader
     uploaded_file = st.file_uploader("Take a picture of your sock and upload it here", type=["jpg", "png"])
 
-    # Button to confirm the upload and process the image
-    if st.button("Upload Next Image"):
-        if uploaded_file is not None:
-            # Convert the file to an image
-            image = Image.open(uploaded_file)
-            st.session_state.images.append(image)
+    # If a file is uploaded, process and display it immediately
+    if uploaded_file is not None:
+        # Convert the file to an image
+        image = Image.open(uploaded_file)
+        st.session_state.images.append(image)
+        uploaded_file = None  # Reset the uploader for the next file
 
     # Display images in a grid
     cols_per_row = 4