add change

app.py

@@ -44,12 +44,14 @@ def pil_to_binary_mask(pil_image, threshold=0):
 def fetch_products(query):
     result= amazon_oxy.scrape_amazon(query)
     values = list(result.values())
-    prices = []
+    pic_and_prices = []
     urls = []
+    i = 0
     for price, url in values:
-        prices.append(price)
+        pic_and_prices.append((values[i], "$"+str(price)))
+        i+=1
         urls.append(url)
-    return [list(result.keys()), prices, urls]
+    return [pic_and_prices, urls]
 
 base_path = 'yisol/IDM-VTON'
 example_path = os.path.join(os.path.dirname(__file__), 'example')
@@ -318,9 +320,14 @@ 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=[image_gallery, price_display, url_display])
+    fetch_button.click(fn=fetch_products, inputs=prompt, outputs=[image_gallery, url_display])
     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=None,inputs=buy_link,js=f"(buy_link) => {{ window.open(buy_link.substring(buy_link.indexOf('amazon.com')), '_blank');console.log(buy_link) }}")
-    
+    #buy_button.click(fn=None,inputs=buy_link,js=f"(buy_link) => {{ window.open(buy_link.substring(buy_link.indexOf('amazon.com')), '_blank');console.log(buy_link) }}")
+    buy_button.click(fn=None,inputs=buy_link, js=f"(buy_link) => {{ 
+        const clean_link = buy_link.includes('http') ? buy_link : 'https://' + buy_link;
+        window.open(clean_link.substring(clean_link.indexOf('amazon.com')), '_blank');
+        console.log(clean_link);
+    }}")
+
 image_blocks.launch()