Update app.py

app.py

@@ -10,6 +10,7 @@ import time
 from zipfile import ZipFile
 import gradio as gr
 from docx import Document
+from io import BytesIO
 
 # FUNCTIONS
 
@@ -74,10 +75,22 @@ def get_seo_tags(image_path, topical_map, attempts=0):
   topic_list = [topic.strip() for topic in topic_list]
   topic_list.insert(0, "irrelevant")
   
-  # base64 upload of the image to OpenAI
   def encode_image(image_path):
-    with open(image_path, "rb") as image_file:
-      return base64.b64encode(image_file.read()).decode('utf-8')
+    # Check the file extension to determine if it is a HEIC file
+    if image_path.lower().endswith('.heic'):
+        # Load the HEIC image
+        image = Image.open(image_path)
+        # Convert the image to JPEG format in memory
+        with BytesIO() as img_buffer:
+            image.save(img_buffer, format='JPEG')
+            # Seek to the beginning of the stream
+            img_buffer.seek(0)
+            # Read the JPEG image data and encode it in base64
+            return base64.b64encode(img_buffer.read()).decode('utf-8')
+    else:
+        # Handle other image types directly
+        with open(image_path, "rb") as image_file:
+            return base64.b64encode(image_file.read()).decode('utf-8')
 
   print(image_path)
   base64_image = encode_image(image_path)