cmahima commited on
Commit
725736a
1 Parent(s): ac3944a

add url tab logic

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -13,6 +13,7 @@ from transformers import (
13
  from diffusers import DDPMScheduler,AutoencoderKL
14
  from typing import List
15
  import ast
 
16
 
17
  import torch
18
  import os
@@ -252,10 +253,7 @@ def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_ste
252
  # return images[0], mask_gray
253
  # Function to handle image selection from the gallery
254
  def select_image(images, urls, evt: gr.SelectData):
255
-
256
- print(type(urls))
257
  urls = ast.literal_eval(urls)
258
- print(f"URL of image is {urls[evt.index]}")
259
  return images[evt.index][0], urls[evt.index]
260
 
261
  garm_list = os.listdir(os.path.join(example_path,"cloth"))
@@ -274,7 +272,7 @@ for ex_human in human_list_path:
274
 
275
  def open_link(link):
276
  print(f"link here is {link}")
277
- return f'<script>window.open("{link}", "_blank");</script>'
278
  ##default human
279
 
280
 
@@ -323,6 +321,6 @@ with image_blocks as demo:
323
  fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
324
  image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_link])
325
  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')
326
- buy_button.click(fn=open_link, inputs=buy_link, outputs= output)
327
 
328
  image_blocks.launch()
 
13
  from diffusers import DDPMScheduler,AutoencoderKL
14
  from typing import List
15
  import ast
16
+ import webbrowser
17
 
18
  import torch
19
  import os
 
253
  # return images[0], mask_gray
254
  # Function to handle image selection from the gallery
255
  def select_image(images, urls, evt: gr.SelectData):
 
 
256
  urls = ast.literal_eval(urls)
 
257
  return images[evt.index][0], urls[evt.index]
258
 
259
  garm_list = os.listdir(os.path.join(example_path,"cloth"))
 
272
 
273
  def open_link(link):
274
  print(f"link here is {link}")
275
+ webbrowser.open_new_tab(link)
276
  ##default human
277
 
278
 
 
321
  fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, price_display, url_display])
322
  image_gallery.select(select_image, [image_gallery, url_display], [garm_img, buy_link])
323
  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')
324
+ buy_button.click(fn=open_link, inputs=buy_link)
325
 
326
  image_blocks.launch()