Update app.py

app.py

@@ -10,21 +10,23 @@ if 'images' not in st.session_state:
 
 # Function to reset images
 def reset_images():
-    st.session_state.images = []
+    # Clear the images list
+    st.session_state.images.clear()
+    # Reset the file uploader widget
+    st.session_state.file_uploader = None
 
 # Main app function
 def main():
     st.title("Sock Image Uploader")
 
     # Image uploader
-    uploaded_file = st.file_uploader("Take a picture of your sock and upload it here", type=["jpg", "png"])
+    uploaded_file = st.file_uploader("Take a picture of your sock and upload it here", type=["jpg", "png"], key="file_uploader")
 
     # 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