Update app.py

app.py

@@ -1,12 +1,3 @@
-import gradio as gr
-import requests
-import json
-
-OXYLABS_ENDPOINT = 'https://realtime.oxylabs.io/v1/queries'
-OXYLABS_AUTH = ('user', 'pass1')
-OPENAI_ENDPOINT = 'https://api.openai.com/v1/engines/davinci/completions'
-OPENAI_AUTH = 'Bearer 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'
-
 def get_product_details(asin, domain):
     payload = {
         'source': 'amazon_product',
@@ -16,19 +7,24 @@ def get_product_details(asin, domain):
         'context': [{'key': 'autoselect_variant', 'value': True}]
     }
     response = requests.post(OXYLABS_ENDPOINT, auth=OXYLABS_AUTH, json=payload)
-    data = response.json()["results"][0]["content"]
+    data = response.json()
 
-    # Isolate information referring to the input ASIN
-    product_info = {
-        "asin": data.get("asin", "N/A"),
-        "title": data.get("title", "N/A"),
-        "brand": data.get("brand", "N/A"),
-        "price": data.get("price", "N/A"),
-        "rating": data.get("rating", "N/A"),
-        "description": data.get("description", "N/A")
-    }
-    
-    return json.dumps(product_info, indent=4)
+    # Check if 'results' and 'content' keys are present in the data
+    if 'results' in data and len(data['results']) > 0 and 'content' in data['results'][0]:
+        content = data['results'][0]['content']
+
+        product_info = {
+            "asin": content.get("asin", "N/A"),
+            "title": content.get("title", "N/A"),
+            "brand": content.get("brand", "N/A"),
+            "price": content.get("price", "N/A"),
+            "rating": content.get("rating", "N/A"),
+            "description": content.get("description", "N/A")
+        }
+
+        return json.dumps(product_info, indent=4)
+    else:
+        return "Failed to retrieve product details."
 
 def get_reviews(asin, domain):
     payload = {
@@ -38,38 +34,23 @@ def get_reviews(asin, domain):
         'parse': True
     }
     response = requests.post(OXYLABS_ENDPOINT, auth=OXYLABS_AUTH, json=payload)
-    reviews = response.json()["results"][0]["content"]["reviews"]
-    review_texts = "\n\n".join([review["content"] for review in reviews])
-    
-    # Summarize reviews using OpenAI
-    openai_payload = {
-        'prompt': f"Summarize the following reviews:\n\n{review_texts}",
-        'max_tokens': 150
-    }
-    headers = {
-        'Authorization': OPENAI_AUTH,
-        'Content-Type': 'application/json'
-    }
-    response = requests.post(OPENAI_ENDPOINT, headers=headers, json=openai_payload)
-    summary = response.json()["choices"][0]["text"].strip()
-    return summary
-
-def amazon_combined_interface(asin, domain):
-    product_details = get_product_details(asin, domain)
-    review_summary = get_reviews(asin, domain)
-    
-    return product_details, review_summary
+    data = response.json()
 
-interface = gr.Interface(
-    fn=amazon_combined_interface,
-    inputs=[
-        gr.Textbox(label="Enter the ASIN"),
-        gr.Dropdown(choices=["com", "uk", "de", "fr", "es", "it", "nl"], label="Select Marketplace")
-    ],
-    outputs=[
-        gr.Textbox(label="Product Details"),
-        gr.Textbox(label="Review Summary")
-    ]
-)
+    # Check if 'results', 'content', and 'reviews' keys are present in the data
+    if 'results' in data and len(data['results']) > 0 and 'content' in data['results'][0] and 'reviews' in data['results'][0]['content']:
+        reviews = data['results'][0]['content']['reviews']
+        review_texts = "\n\n".join([review["content"] for review in reviews])
 
-interface.launch()
+        openai_payload = {
+            'prompt': f"Summarize the following reviews:\n\n{review_texts}",
+            'max_tokens': 150
+        }
+        headers = {
+            'Authorization': OPENAI_AUTH,
+            'Content-Type': 'application/json'
+        }
+        response = requests.post(OPENAI_ENDPOINT, headers=headers, json=openai_payload)
+        summary = response.json()["choices"][0]["text"].strip()
+        return summary
+    else:
+        return "Failed to retrieve reviews."