ff

gradio_seg2image.py

@@ -1,7 +1,22 @@
 # This file is adapted from https://github.com/lllyasviel/ControlNet/blob/f4748e3630d8141d7765e2bd9b1e348f47847707/gradio_seg2image.py
 # The original license file is LICENSE.ControlNet in this repo.
 import gradio as gr
+from PIL import Image
 
+def dummyfun(result_gallery):
+    print(f"type of gallery is ^^ - {type(result_gallery)}")
+    print(f"length of gallery is ^^ - {len(result_gallery)}")
+    print(f"first elem of gallery is ^^ - {result_gallery[0]}")
+    print(f"first elem of gallery is ^^ - {result_gallery[1]}")
+    # Load the image
+    image = result_gallery[1] £Image.open("example.jpg")
+    
+    # Get the filename
+    filename = image.filename
+    
+    # Print the filename
+    print(f"filename is ^^ - {filename}")
+    return filename
 
 def create_demo(process, max_images=12):
     with gr.Blocks() as demo:
@@ -59,6 +74,8 @@ def create_demo(process, max_images=12):
                                             show_label=False,
                                             elem_id='gallery').style(
                                                 grid=2, height='auto')
+                b1 = gr.Button('Get filenames')
+                filename = gr.Textbox(label="image file names")
         ips = [
             input_image, prompt, a_prompt, n_prompt, num_samples,
             image_resolution, detect_resolution, ddim_steps, scale, seed, eta
@@ -67,4 +84,6 @@ def create_demo(process, max_images=12):
                          inputs=ips,
                          outputs=[result_gallery],
                          api_name='seg')
+        b1.click(dummyfun, [result_gallery], [filename])
+        
     return demo