Upload app.py with huggingface_hub

app.py

@@ -700,6 +700,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate"),
     with gr.Row():
         with gr.Column(scale=1, min_width=360):
             multiimage_prompt = gr.Gallery(label="Multi-View Images", format="png", type="pil", height=400, columns=3, interactive=True)
+            remove_img_btn = gr.Button("Remove Selected Image", size="sm", variant="secondary")
 
             resolution = gr.Radio(["512", "1024", "1536"], label="Resolution", value="1024")
             seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
@@ -754,11 +755,31 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate"),
     output_buf = gr.State()
     processed_buf = gr.State()
     render_buf = gr.State()
+    selected_img_idx = gr.State(value=None)
 
 
     # Handlers
     demo.load(start_session)
     demo.unload(end_session)
+
+    def on_gallery_select(evt: gr.SelectData):
+        return evt.index
+
+    def remove_selected_image(images, idx):
+        if images is None or idx is None or not images:
+            return images, None
+        images = list(images)
+        if idx < len(images):
+            images.pop(idx)
+        return images, None
+
+    multiimage_prompt.select(on_gallery_select, outputs=[selected_img_idx])
+    remove_img_btn.click(
+        remove_selected_image,
+        inputs=[multiimage_prompt, selected_img_idx],
+        outputs=[multiimage_prompt, selected_img_idx],
+    )
+
     multiimage_prompt.upload(
         preprocess_images,
         inputs=[multiimage_prompt],