Spaces:

webshop
/

amazon_shop

Runtime error

App Files Files Community

John Yang commited on Jul 10, 2022

Commit

7fb90e2

1 Parent(s): 730ca01

Update output format

Browse files

Files changed (1) hide show

app.py +45 -42

app.py CHANGED Viewed

@@ -91,6 +91,36 @@ def bert_predict(obs, info, softmax=True):
         idx = outputs.logits[0].argmax(0).item()
     return valid_acts[idx]
 def predict(obs, info):
     """
@@ -172,25 +202,7 @@ def run_episode(goal, env, verbose=True):
                 page_type = Page.SEARCH
             elif action == 'click[buy now]':
-                return_value = {}
-                if env == 'amazon':
-                    return_value['Product URL'] = f"https://www.amazon.com/dp/{asin}"
-                    if len(clicked_options) > 0:
-                        return_value['Selected Options'] = ', '.join(list(clicked_options))
-                if env == 'webshop':
-                    query_str = "+".join(search_terms.split())
-                    options_str = json.dumps(options)
-                    asin_url = (
-                        f'{WEBSHOP_URL}/item_page/{WEBSHOP_SESSION}/'
-                        f'{asin}/{query_str}/{page_num}/{options_str}'
-                    )
-                    return_value['Product URL'] = asin_url
-                if env == 'ebay':
-                    asin_url = f"https://www.ebay.com/itm/{asin}"
-                    return_value['Product URL'] = asin_url
-                    if len(clicked_options) > 0:
-                        return_value['Selected Options'] = ', '.join(list(clicked_options))
-                return return_value
             elif prev_page_type == Page.ITEM_PAGE:
                 found = False
@@ -214,7 +226,7 @@ def run_episode(goal, env, verbose=True):
             if search_terms in search_results_cache:
                 data = search_results_cache[search_terms]
                 if verbose:
-                    print(f"Loading cached results page for\"{search_terms}\"")
             else:
                 begin = time.time()
                 if env == 'amazon':
@@ -273,37 +285,28 @@ def run_episode(goal, env, verbose=True):
             print("Extracting available actions took", end-begin, "seconds")
         if i == 50:
-            return_value = {}
-            if env == 'amazon':
-                return_value['Product URL'] = f"https://www.amazon.com/dp/{asin}"
-                if len(clicked_options) > 0:
-                    return_value['Selected Options'] = ', '.join(list(clicked_options))
-            if env == 'webshop':
-                query_str = "+".join(search_terms.split())
-                options_str = json.dumps(options)
-                asin_url = (
-                    f'{WEBSHOP_URL}/item_page/{WEBSHOP_SESSION}/'
-                    f'{asin}/{query_str}/{page_num}/{options_str}'
-                )
-                return_value['Product URL'] = asin_url
-            if env == 'ebay':
-                asin_url = f"https://www.ebay.com/itm/{asin}"
-                return_value['Product URL'] = asin_url
-                if len(clicked_options) > 0:
-                    return_value['Selected Options'] = ', '.join(list(clicked_options))
-            return return_value
-gr.Interface(fn=run_episode,
     inputs=[
         gr.inputs.Textbox(lines=7, label="Input Text"),
         gr.inputs.Radio(['Amazon', 'eBay'], type="value", default="Amazon", label='Environment')
     ],
-    outputs="text",
     examples=[
         ["I want to find a gold floor lamp with a glass shade and a nickel finish that i can use for my living room, and price lower than 270.00 dollars", "Amazon"],
         ["I need some cute heart-shaped glittery cupcake picks as a gift to bring to a baby shower", "Amazon"],
         ["I'm trying to find white bluetooth speakers that are not only water resistant but also come with stereo sound", "eBay"],
-        ["find me the soy free 3.5 ounce 4-pack of dang thai rice chips, and make sure they are the aged cheddar flavor.  i also need the ones in the resealable bags", "eBay"]
     ],
     title="WebShop",
     article="<p style='padding-top:15px;text-align:center;'>To learn more about this project, check out the <a href='https://webshop-pnlp.github.io/' target='_blank'>project page</a>!</p>",

         idx = outputs.logits[0].argmax(0).item()
     return valid_acts[idx]
+def get_return_value(env, asin, options, search_terms, page_num, product):
+    asin_url = None
+    # Determine product URL + options based on environment
+    if env == 'webshop':
+        query_str = "+".join(search_terms.split())
+        options_str = json.dumps(options)
+        asin_url = (
+            f'{WEBSHOP_URL}/item_page/{WEBSHOP_SESSION}/'
+            f'{asin}/{query_str}/{page_num}/{options_str}'
+        )
+    else:
+        asin_url = f"https://www.ebay.com/itm/{asin}" if env == 'ebay' else \
+            f"https://www.amazon.com/dp/{asin}"
+    # Extract relevant fields for product
+    product_reduced = {k: v for k, v in product.items() if k in ["asin", "Title", "Description", "BulletPoints"]}
+    product_reduced["Description"] = product_reduced["Description"][:100] + "..."
+    product_reduced["Features"] = product_reduced.pop("BulletPoints")
+    product_reduced["Features"] = product_reduced["Features"][:100] + "..."
+    # Create HTML to show link to product
+    html = """<!DOCTYPE html><html><head><title>Chosen Product</title></head><body>"""
+    html += f"""Product Image:<img src="{product["MainImage"]}" height="50px" /><br>""" if len(product["MainImage"]) > 0 else ""
+    html += f"""Link to Product:
+        <a href="{asin_url}" style="color:blue;text-decoration:underline;" target="_blank">{asin_url}</a>
+        </body></html>"""
+    return product_reduced, options if len(options) > 0 else "None Selected", html
 def predict(obs, info):
     """
                 page_type = Page.SEARCH
             elif action == 'click[buy now]':
+                return get_return_value(env, asin, options, search_terms, page_num, product_map[asin])
             elif prev_page_type == Page.ITEM_PAGE:
                 found = False
             if search_terms in search_results_cache:
                 data = search_results_cache[search_terms]
                 if verbose:
+                    print(f"Loading cached results page for \"{search_terms}\"")
             else:
                 begin = time.time()
                 if env == 'amazon':
             print("Extracting available actions took", end-begin, "seconds")
         if i == 50:
+            return get_return_value(env, asin, options, search_terms, page_num, product_map[asin])
+gr.Interface(
+    fn=run_episode,
     inputs=[
         gr.inputs.Textbox(lines=7, label="Input Text"),
         gr.inputs.Radio(['Amazon', 'eBay'], type="value", default="Amazon", label='Environment')
     ],
+    outputs=[
+        gr.outputs.JSON(label="Selected Product"),
+        gr.outputs.JSON(label="Selected Options"),
+        gr.outputs.HTML()
+    ],
     examples=[
         ["I want to find a gold floor lamp with a glass shade and a nickel finish that i can use for my living room, and price lower than 270.00 dollars", "Amazon"],
         ["I need some cute heart-shaped glittery cupcake picks as a gift to bring to a baby shower", "Amazon"],
+        ["I want to buy ballet shoes which have rubber sole in grey suede color and a size of 6", "Amazon"],
+        ["I would like a 7 piece king comforter set decorated with flowers and is machine washable", "Amazon"],
         ["I'm trying to find white bluetooth speakers that are not only water resistant but also come with stereo sound", "eBay"],
+        ["find me the soy free 3.5 ounce 4-pack of dang thai rice chips, and make sure they are the aged cheddar flavor.  i also need the ones in the resealable bags", "eBay"],
+        ["I am looking for a milk chocolate of 1 pound size in a single pack for valentine day", "eBay"],
+        ["I'm looking for a mini pc intel core desktop computer which supports with windows 11", "eBay"]
     ],
     title="WebShop",
     article="<p style='padding-top:15px;text-align:center;'>To learn more about this project, check out the <a href='https://webshop-pnlp.github.io/' target='_blank'>project page</a>!</p>",