Update app.py

app.py

@@ -18,13 +18,21 @@ uploaded_images = st.file_uploader("Upload images", type=["jpg", "png"], accept_
 
 # Display thumbnail images alongside file names and sizes in the sidebar
 selected_images = []
+
 if uploaded_images:
+
+    # Add a "Select All" checkbox in the sidebar
+    select_all = st.sidebar.checkbox("Select All", False)
+    
     for idx, img in enumerate(uploaded_images):
         image = Image.open(img)
         checkbox_key = f"{img.name}_checkbox_{idx}"  # Unique key for each checkbox
         # Display thumbnail image and checkbox in sidebar
         st.sidebar.image(image, caption=f"{img.name} {img.size / 1024.0:.1f} KB", width=40)
-        selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=checkbox_key)
+        #selected = st.sidebar.checkbox(f"Select {img.name}", value=False, key=checkbox_key)
+        # If "Select All" is checked, all individual checkboxes are selected
+        selected = st.sidebar.checkbox(f"Select {img.name}", value=select_all, key=checkbox_key)        
+        
         if selected:
             selected_images.append(image)