Update bf_trigger.py

bf_trigger.py

@@ -302,8 +302,32 @@ def get_seo_tags(image_url, topical_map, new_imgs, attempts=0, max_attempts=5):
     print("Exception during API call:", str(e))
     return get_seo_tags(image_url, topical_map, new_imgs, attempts=attempts + 1)
 
+def personalize_answer(answer, query_engine):
+    if query_engine:
+        prompt = f'''
+        % You are an expert construction contracter describing this image
+    
+        % Goal: Add relevant company information to the Answer based on the context provided. Only output the Enhancement. Remove the triple quotes from the output
+    
+        % Answer:
+        ```{answer}```
+    
+        % Instructions:
+            Step 1 - Identify what relevant company information from the context is relevant to the Answer
+            Step 2 - Enhance the Answer with the relevant company information. This will be known as the Enhancement
+            Step 3 - Make the Enhancement the same character length as the Answer. Use Python to check that they are the same character length.
+            Step 4 - Only output the Enhancement. Remove the triple quotes from the output
+            
+        '''
+    
+        response = query_engine.query(prompt)
+    
+        return response.response.replace("`", "")
+    else:
+        return answer
+
 # creates the asset in the client's brand folder
-def create_ai_asset(asset_dict, topical_map, collection_name, new_imgs, tags=True):
+def create_ai_asset(asset_dict, topical_map, collection_name, new_imgs, query_engine, tags=True):
     '''
     Creates asset from image path. Also creates seo tags, topic, and alt tag for
     image
@@ -346,6 +370,10 @@ def create_ai_asset(asset_dict, topical_map, collection_name, new_imgs, tags=Tru
     alt_tag = json_dict['alt_tag']
     image_name = json_dict['filename']
 
+    description = personalize_answer(description, query_engine)
+    caption = personalize_answer(caption, query_engine)
+    alt_tag = personalize_answer(alt_tag, query_engine)
+    
     headers = {
       'Content-Type': 'application/json',
       'Authorization': brandfolder_api
@@ -593,7 +621,7 @@ def delete_og_asset(asset_id):
 
   return
 
-def run_preprocess_ai(topical_map, client_name, section_type, progress=gr.Progress()):
+def run_preprocess_ai(topical_map, client_name, section_type, query_engine=query_engine, progress=gr.Progress()):
   section_id = section_key_dict[section_type]
   headers = {
     'Content-Type': 'application/json',
@@ -622,7 +650,7 @@ def run_preprocess_ai(topical_map, client_name, section_type, progress=gr.Progre
     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_img = create_ai_asset(asset_dict, topical_map, client_name, new_imgs, query_engine=query_engine[-1])
         new_imgs.append(new_img)
         if new_img:
           delete_og_asset(asset_id)