Update bf_trigger.py

bf_trigger.py

@@ -306,7 +306,10 @@ def create_ai_asset(asset_dict, topical_map, collection_name, new_imgs, tags=Tru
     topical_map = get_topical_map_text(topical_map)
     client_name = asset_dict['client_name']
     access_key = asset_dict['access_key']
-    client_key = client_key_dict[collection_name]
+    try:
+        client_key = client_key_dict[collection_name]
+    except:
+        client_key = 'no key'
     if client_name != collection_name and access_key != client_key:
       print(f'{collection_name} != {client_name}')
       print(f'{access_key} != {client_key}')
@@ -578,11 +581,14 @@ def run_preprocess_ai(topical_map, client_name, section_type, progress=gr.Progre
   
   new_imgs = []
   for asset_id in progress.tqdm(asset_names, desc="Uploading..."):
-    time.sleep(2)
-    asset_dict = get_asset_info(asset_id)
-    new_img = create_ai_asset(asset_dict, topical_map, client_name, new_imgs)
-    new_imgs.append(new_img)
-    if new_img:
-      delete_og_asset(asset_id)
+    try:
+        time.sleep(2)
+        asset_dict = get_asset_info(asset_id)
+        new_img = create_ai_asset(asset_dict, topical_map, client_name, new_imgs)
+        new_imgs.append(new_img)
+        if new_img:
+          delete_og_asset(asset_id)
+    except Exception as e:
+        print(f'An unexpected error occured processing {asset_dict['asset_name']}: {e}')
   gr.Info('Images have been processed!')
   return