Update app.py

app.py

@@ -452,18 +452,21 @@ with gr.Blocks() as demo:
 
     selected_folder_images = gr.State(value=[])
 
-    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 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=[], outputs=selected_folder_images)
+    folder_images_gallery.select(fn=select_folder_image, inputs=[mode], 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":
@@ -499,4 +502,3 @@ if __name__ == "__main__":
     # Your Gradio interface setup here
     demo.launch(auth=("roland", "roland"), debug=True)
     clear_memory()
-