changes for link

app.py

@@ -253,9 +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(images, evt: gr.SelectData):
-    print(images)
-    return images[evt.index][0]
+def select_image(images, urls, evt: gr.SelectData):
+    print(urls)
+    return images[evt.index][0], urls[evt.index][0]
 
 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]
@@ -277,7 +277,7 @@ for ex_human in human_list_path:
 
 image_blocks = gr.Blocks(theme=gr.themes.Soft()).queue()
 with image_blocks as demo:
-    gr.HTML("<center><h1>Shop with Window 🛍️👗👗</h1></center>")
+    gr.HTML("<center><h1>Shop with Window 🛍️👗👕🛒</h1></center>")
     gr.HTML("<center><p>Upload an image of yourself or select from examples then describe your garment in the text box and wait for the magic. ✨</p></center>")
     
     with gr.Row():
@@ -299,30 +299,28 @@ with image_blocks as demo:
             fetch_button = gr.Button("Fetch Products")
             garm_img = gr.Image(label="Garment", sources='upload', type="pil")
 
-            image_gallery = gr.Gallery(label="Fetched Images", columns=4,allow_preview=False)
+            image_gallery = gr.Gallery(label="Generated images", show_label=True, elem_id="gallery"
+    , columns=[3], rows=[1], object_fit="contain", height="auto", allow_preview= False)
             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():
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
             masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)
-        with gr.Column():
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
             image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False)
 
     with gr.Column():
         try_button = gr.Button(value="Try-on")
+        buy_button = gr.Button(value="Like it? Click to buy")
         with gr.Accordion(label="Advanced Settings", open=False):
             with gr.Row():
                 denoise_steps = gr.Number(label="Denoising Steps", minimum=20, maximum=40, value=30, step=1)
                 seed = gr.Number(label="Seed", minimum=-1, maximum=2147483647, step=1, value=42)
-
+        
 
     fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
-    image_gallery.select(select_image, image_gallery, garm_img)
-
+    image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_button])
     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')
-
+    
             
-
-
 image_blocks.launch()