Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 from PIL import Image
-import io
 
 # Initialize a session state to keep track of images
 if 'images' not in st.session_state:
@@ -15,7 +14,7 @@ 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
     if uploaded_file is not None:
@@ -23,9 +22,12 @@ 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)
+        # Clear the file uploader for the next upload
+        st.session_state.file_uploader = None
+
+    # Display images
+    for idx, img in enumerate(st.session_state.images, start=1):
+        st.image(img, caption=f"Image {idx}", use_column_width=True)
 
     # Reset button
     if st.button('Reset and Start Over'):