Update app.py

app.py

@@ -134,6 +134,7 @@ def get_seo_tags(image_path, topical_map, attempts=0):
                     Using the description, create a 160 character caption. Make sure the caption is less than 160 characters.
                     Using the description, create 3 topic-relevant SEO tags for this image that will drive traffic to our website. The SEO tags must be two words or less. You must give 3 SEO tags.
                     Using the description, provide a topic-relevant SEO alt tag for the image that will enhance how the website is ranked on search engines.
+                    Using the description, provide a new file name for the image as well. Use hyphens for the filename. Do not include extension.
                     Ignore all personal information within the image.
                     Be as specific as possible when identifying tools in the image.
                     
@@ -144,6 +145,7 @@ def get_seo_tags(image_path, topical_map, attempts=0):
                     "caption": caption,
                     "seo": [seo],
                     "alt_tag": [alt tag],
+                    "filename": filename
                     }
                     """
           },
@@ -172,7 +174,7 @@ def get_seo_tags(image_path, topical_map, attempts=0):
       # generates dictionary based on response
       json_dict = ast.literal_eval(response.json()['choices'][0]['message']['content'])
 
-      keys = ['topic', 'description', 'caption', 'seo', 'alt_tag']
+      keys = ['topic', 'description', 'caption', 'seo', 'alt_tag', 'filename']
       json_keys = []
       json_keys = list(json_dict.keys())
 
@@ -266,7 +268,7 @@ def upload_image(image_path, upload_url):
     return response
 
 # creates the asset in the client's brand folder
-def create_asset(client_name, collection_id, image_path, topical_map, tags=True, project_bool=False):
+def create_asset(client_name, collection_id, image_path, topical_map, new_imgs, tags=True, project_bool=False):
     '''
     Creates asset from image path. Also creates seo tags, topic, and alt tag for
     image
@@ -283,7 +285,13 @@ def create_asset(client_name, collection_id, image_path, topical_map, tags=True,
     caption = json_dict['caption']
     seo_tags = json_dict['seo']
     alt_tag = json_dict['alt_tag']
-    image_name = str(image_path).split('/')[-1].split('.')[0]
+    image_name = json_dict['filename']
+    
+    counter = 1
+    while image_name in new_imgs:
+        image_name = f'{image_name}_{counter}'
+        counter += 1
+        
 
     headers = {
       'Accept': 'application/json',
@@ -314,7 +322,7 @@ def create_asset(client_name, collection_id, image_path, topical_map, tags=True,
 
     with NamedTemporaryFile(delete=True, suffix='.jpg') as temp_image:
       # fp = TemporaryFile()
-      cv2.imwrite(temp_image.name, image)
+      cv2.imwrite(temp_image.name, image, [int(cv2.IMWRITE_JPEG_QUALITY), 95])
       # fp.seek(0)
       response = requests.put(upload_url, data=temp_image)
     # fp.close()
@@ -588,6 +596,7 @@ def import_client_data(client_name, zipfile, topical_map, password, project_bool
   # gets all image files from the google drive folder
   zip = ZipFile(zipfile.name)
   img_list = get_imgs_from_folder([], zipfile)
+  new_imgs = []
   print(img_list)
 
   # iterates all images and puts them into brandfolder with AI elements
@@ -596,7 +605,8 @@ def import_client_data(client_name, zipfile, topical_map, password, project_bool
     time.sleep(5)
     img = zip.extract(img)
     print(client_name)
-    create_asset(client_name, collection_id, img, topical_map, project_bool=project_bool)
+    new_img = create_asset(client_name, collection_id, img, topical_map, new_imgs, project_bool=project_bool)
+    new_imgs.append(new_img)
   gr.Info('Images have been uploaded!')
   return "Images Uploaded"