Update app.py

app.py

@@ -419,6 +419,7 @@ with gr.Blocks() as demo:
                         height=235,
                         allow_preview=False
                     )
+                    clear_selection_button = gr.Button("Clear Selection")
 
         with gr.Column(scale=2):
             style_images_gallery = gr.Gallery(
@@ -452,21 +453,22 @@ with gr.Blocks() as demo:
 
     selected_folder_images = gr.State(value=[])
 
-    def select_folder_image(evt: gr.SelectData, mode):
-        if mode == "Batch":
-            folder_image_names = [img[0] for img in loaded_images]
-            if evt.index < 0 or evt.index >= len(folder_image_names):
-                raise ValueError(f"Invalid index: {evt.index}")
-            selected_image_name = folder_image_names[evt.index]
-            selected_image = next(img for img in loaded_images if img[0] == selected_image_name)
-            current_images = selected_folder_images.value
-            if selected_image not in current_images:
-                current_images.append(selected_image)
-            return current_images
-        else:
-            return selected_folder_images.value  # No change if not in Batch mode
-
-    folder_images_gallery.select(fn=select_folder_image, inputs=[mode], outputs=selected_folder_images)
+    def select_folder_image(evt: gr.SelectData):
+        folder_image_names = [img[0] for img in loaded_images]
+        if evt.index < 0 or evt.index >= len(folder_image_names):
+            raise ValueError(f"Invalid index: {evt.index}")
+        selected_image_name = folder_image_names[evt.index]
+        selected_image = next(img for img in loaded_images if img[0] == selected_image_name)
+        current_images = selected_folder_images.value
+        if selected_image not in current_images:
+            current_images.append(selected_image)
+        return current_images
+
+    def clear_selected_folder_images():
+        return []
+
+    folder_images_gallery.select(fn=select_folder_image, inputs=[], outputs=selected_folder_images)
+    clear_selection_button.click(fn=clear_selected_folder_images, inputs=[], outputs=selected_folder_images)
 
     def generate_images_with_folder_images(prompt, negative_prompt, batch_count, use_controlnet, controlnet_type, mode, use_control_folder, selected_folder_images, batch_images_input, progress=gr.Progress(track_tqdm=True)):
         if mode == "Batch":