app.py

app.py

@@ -253,10 +253,9 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
         return images[0], mask_gray
     # return images[0], mask_gray
 # Function to handle image selection from the gallery
-def select_image(image_url):
-    response = requests.get(image_url)
-    image = Image.open(BytesIO(response.content))
-    return image
+def select_image(images, evt: gr.SelectData):
+    print(images)
+    return images[evt.index]['name']
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
 garm_list_path = [os.path.join(example_path,"cloth",garm) for garm in garm_list]
@@ -295,15 +294,10 @@ with image_blocks as demo:
             )
 
         with gr.Column():
-            garm_img = gr.Image(label="Garment", elem_id="amazon-img",show_share_button=False)
-            with gr.Row(elem_id="prompt-container"):
-                with gr.Row():
-                    prompt = gr.Textbox(placeholder="Description of garment ex) Yellow Top", show_label=False, elem_id="prompt")
-                    fetch_button = gr.Button("Fetch Products")
-            example = gr.Examples(
-                inputs=garm_img,
-                examples_per_page=8,
-                examples=garm_list_path)
+            garm_img = gr.Image(label="Garment", elem_id="upload",show_share_button=False)
+            prompt = gr.Textbox(placeholder="Description of garment ex) Yellow Top", show_label=False, elem_id="prompt")
+            fetch_button = gr.Button("Fetch Products")
+            image_gallery = gr.Gallery(label="Fetched Images", columns=4)
             price_display = gr.Textbox(label="Price of Selected Image", interactive=False)
             url_display = gr.Textbox(label="URL of Selected Image", interactive=False)
         with gr.Column():
@@ -321,7 +315,8 @@ with image_blocks as demo:
                 seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=42)
 
 
-    fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[example, price_display, url_display])
+    fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
+    image_gallery.select(select_image, image_gallery, garm_img)
 
     try_button.click(fn=start_tryon, inputs=[imgs, garm_img, prompt, is_checked,is_checked_crop, denoise_steps, seed], outputs=[image_out,masked_img], api_name='tryon')