mylesai commited on
Commit
19428d4
·
verified ·
1 Parent(s): 02ec51f

Update bf_trigger.py

Browse files
Files changed (1) hide show
  1. bf_trigger.py +2 -5
bf_trigger.py CHANGED
@@ -633,10 +633,7 @@ def run_preprocess_ai(topical_map, client_name, section_type, query_engine=None,
633
  pre_process_ids = []
634
  run = True
635
  while run == True:
636
- if section_type == 'Pre-Processed Images':
637
- r = requests.get(f'https://brandfolder.com/api/v4/brandfolders/988cgqcg8xsrr5g9h7gtsqkg/assets?include=section,custom_fields&fields=created_at&page={page}&per=3000&sort_by=created_at&order=DESC', params={}, headers=headers)
638
- elif section_type == 'Original Project Assets':
639
- r = requests.get(f'https://brandfolder.com/api/v4/collections/{collection_id}/assets?include=section,custom_fields&fields=created_at&page={page}&per=3000&sort_by=created_at&order=DESC', params={}, headers=headers)
640
  page+=1
641
  asset_names = [item['id'] for item in r.json()['data'] if item['relationships']['section']['data']['id'] == section_id]
642
  if asset_names in pre_process_ids:
@@ -657,4 +654,4 @@ def run_preprocess_ai(topical_map, client_name, section_type, query_engine=None,
657
  except Exception as e:
658
  print(f'An unexpected error occured processing {asset_dict["asset_name"]}: {e}')
659
  gr.Info('Images have been processed!')
660
- return
 
633
  pre_process_ids = []
634
  run = True
635
  while run == True:
636
+ r = requests.get(f'https://brandfolder.com/api/v4/collections/{collection_id}/assets?include=section,custom_fields&fields=created_at&page={page}&per=3000&sort_by=created_at&order=DESC', params={}, headers=headers)
 
 
 
637
  page+=1
638
  asset_names = [item['id'] for item in r.json()['data'] if item['relationships']['section']['data']['id'] == section_id]
639
  if asset_names in pre_process_ids:
 
654
  except Exception as e:
655
  print(f'An unexpected error occured processing {asset_dict["asset_name"]}: {e}')
656
  gr.Info('Images have been processed!')
657
+ return 'Images Processed'