add code for scraper

amazon_oxy.py

@@ -0,0 +1,43 @@
+import requests
+import json
+from dotenv import load_dotenv
+import os
+
+load_dotenv() 
+
+username = os.environ.get('USERNAME')
+passwd = os.environ.get('PASSWORD')
+def scrape_amazon(search_query, n =10):
+    # Structure payload.
+    payload = {
+        'source': 'amazon_search',
+        'domain': 'com',
+        'query': f'{search_query}',
+        'start_page': 1,
+        'pages': 1,
+        'parse': True,
+        # 'context': [
+        #     {'key': 'category_id', 'value': 16391693031}
+        # ],
+    }
+
+    # Get response.
+    response = requests.request(
+        'POST',
+        'https://realtime.oxylabs.io/v1/queries',
+        auth=(username, passwd),
+        json=payload,
+    )
+    result = json.loads(response.content)
+    final_result = {}
+    i=0
+    # Parse result.
+    for item in result['results'][0]['content']['results']['organic']:
+        if i<n:
+            final_result[item['url_image']]= (item['price'],item['url'])
+            i+=1
+        else:
+            break
+        
+    return final_result
+print(scrape_amazon("yellow top"))

app.py

@@ -24,6 +24,17 @@ from preprocess.humanparsing.run_parsing import Parsing
 from preprocess.openpose.run_openpose import OpenPose
 from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
+import amazon_oxy
+
+def fetch_products(query):
+    result= amazon_oxy.scrape_amazon(query)
+    values = list(result.values())
+    prices = []
+    urls = []
+    for price, url in values:
+        prices.append(price)
+        urls.append(url)
+    return [list(result.keys()), prices, urls]
 
 
 def pil_to_binary_mask(pil_image, threshold=0):
@@ -280,11 +291,11 @@ with image_blocks as demo:
             garm_img = gr.Image(label="Garment", sources='upload', type="pil")
             with gr.Row(elem_id="prompt-container"):
                 with gr.Row():
-                    prompt = gr.Textbox(placeholder="Description of garment ex) Short Sleeve Round Neck T-shirts", show_label=False, elem_id="prompt")
-            example = gr.Examples(
-                inputs=garm_img,
-                examples_per_page=8,
-                examples=garm_list_path)
+                    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", input=garm_img).style(grid=[3], height="200px")
+            price_display = gr.Textbox(label="Price of Selected Image", interactive=False)
+            url_display = gr.Markdown()
         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)
@@ -303,6 +314,7 @@ 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])
 
     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')