api-key

consts.py

@@ -0,0 +1 @@
+API_KEY ='34dsadfF$$%#$TGREGEFGE%Q*)(*&%'

createlookalike.py

@@ -23,6 +23,7 @@ from collections import OrderedDict
 #
 # from generate_csv_file import generate_csv_files
 # from load_data import load_data, get_shops
+from consts import API_KEY
 from schemas import Shop
 
 
@@ -140,18 +141,15 @@ async def calculate_shop(shop: Shop, feat_extractor) -> None:
                         design_list.append(v)
 
                     design_dict = {"shop_id": shop.id, "design": im,
-                                   "results": design_list
+                                   "recommendations": design_list
                                    }
 
                     print(design_dict)
-                    try:
-                        response = requests.post(shop.webhook_url, json=design_dict)
-                        if response.status_code == 200:
-                            print(f"result for {im} is sent to {shop.webhook_url}")
-                        else:
-                            print(f"Error sending data {shop.webhook_url} to for result {im}: {response.status_code}")
-                    except Exception as e:
-                        print(f"Error sending data {shop.webhook_url} to for result {im}: {response.status_code}", e)
+                    if await push_home(design_dict, im, shop):
+                        pass
+                    else:
+                        raise
+
 
                     # idx_closest = sorted(range(len(similar_idx)), key=lambda k: similar_idx[k])
 
@@ -164,3 +162,18 @@ async def calculate_shop(shop: Shop, feat_extractor) -> None:
         end = datetime.today()
 
         # return {'shop_id': shop.id, 'start_time': start, 'end_time': end, 'designs': design_json}
+
+
+async def push_home(design_dict, im, shop):
+    headers: dict[str, str] = {
+        "Authorization": API_KEY,
+        "Content-type": "application/json",
+    }
+    try:
+
+        response = requests.post(shop.webhook_url, json=design_dict, headers=headers)
+        response.raise_for_status()
+        return True
+
+    except Exception as e:
+       return False