change app.py

app.py

@@ -254,7 +254,7 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
     # return images[0], mask_gray
 # Function to handle image selection from the gallery
 def select_image(images, urls, evt: gr.SelectData):
-    print(urls)
+    print(urls[evt.index])
     return images[evt.index][0], urls[evt.index]
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))
@@ -271,7 +271,9 @@ for ex_human in human_list_path:
     ex_dict['composite'] = None
     human_ex_list.append(ex_dict)
 
-
+def open_link(link):
+    print(f"link here is {link}")
+    return f'<script>window.open("{link}", "_blank");</script>'
 ##default human
 
 
@@ -301,17 +303,18 @@ with image_blocks as demo:
 
             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, visible=False)
-            url_display = gr.Textbox(label="URL of Selected Image", interactive=False, visible=False)
+            price_display = gr.Textbox(label="Prices of  Images", interactive=False, visible=False)
+            url_display = gr.Textbox(label="URLs of  Images", interactive=False, visible=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)
             # 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")
+            try_button = gr.Button(value="Try-on")
+             buy_link = gr.Textbox(label="URL of Selected Image", interactive=False, visible=False)
+            buy_button = gr.Button(value="Like it? Click to buy")
+            output = gr.HTML()
+    with gr.Column():    
         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)
@@ -319,8 +322,8 @@ with image_blocks as demo:
         
 
     fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
-    image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_button])
+    image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_link])
     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')
-    
+    buy_button.click(fn=open_link, input=buy_link, outputs= output)
             
 image_blocks.launch()