Update app.py

app.py

@@ -230,6 +230,41 @@ def process_image(image_path):
 
     return image
 
+def convert_heic_to_jpeg(heic_path):
+    # Read the HEIC file
+    heif_file = pyheif.read(heic_path)
+    # Convert to a PIL image
+    image = Image.frombytes(
+        heif_file.mode, 
+        heif_file.size, 
+        heif_file.data,
+        "raw",
+        heif_file.mode,
+        heif_file.stride,
+    )
+    # Convert image to JPEG in memory
+    jpeg_buffer = BytesIO()
+    image.save(jpeg_buffer, format="JPEG")
+    jpeg_buffer.seek(0)
+    return jpeg_buffer
+
+def upload_image(image_path, upload_url):
+    # Check if the image is a HEIC file
+    if image_path.lower().endswith('.heic'):
+        # Convert HEIC to JPEG
+        data = convert_heic_to_jpeg(image_path)
+    else:
+        # Open other image types directly
+        data = open(image_path, 'rb')
+    
+    # Upload the image
+    response = requests.put(upload_url, data=data)
+    # Ensure you close the file stream if opened directly
+    if not image_path.lower().endswith('.heic'):
+        data.close()
+    
+    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):
     '''
@@ -263,10 +298,7 @@ def create_asset(client_name, collection_id, image_path, topical_map, tags=True,
     # container for the uploaded image to be used by the post request
     og_object_url = r.json()['object_url']
 
-    # uploads the image
-    with open(image_path, 'rb') as f:
-      response = requests.put(upload_url, data=f)
-
+    response = upload_image(image_path, upload_url)
 
     # binary upload of image_path
     r = requests.get('https://brandfolder.com/api/v4/upload_requests', params={}, headers=headers)