Update app.py

app.py

@@ -453,13 +453,13 @@ with gr.Blocks() as demo:
 
     selected_folder_images = gr.State(value=[])
 
-    def select_folder_image(evt: gr.SelectData):
+    def select_folder_image(evt: gr.SelectData, selected_folder_images):
         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
+        current_images = selected_folder_images or []
         if selected_image not in current_images:
             current_images.append(selected_image)
         return current_images
@@ -467,7 +467,7 @@ with gr.Blocks() as demo:
     def clear_selected_folder_images():
         return []
 
-    folder_images_gallery.select(fn=select_folder_image, inputs=[], outputs=selected_folder_images)
+    folder_images_gallery.select(fn=select_folder_image, inputs=[selected_folder_images], 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)):
@@ -503,4 +503,4 @@ with gr.Blocks() as demo:
 if __name__ == "__main__":
     # Your Gradio interface setup here
     demo.launch(auth=("roland", "roland"), debug=True)
-    clear_memory()
+    clear_memory()