Update app.py

app.py

@@ -11,6 +11,7 @@ from zipfile import ZipFile
 import gradio as gr
 from docx import Document
 from io import BytesIO
+import pyheif
 
 # FUNCTIONS
 
@@ -78,6 +79,17 @@ def get_seo_tags(image_path, topical_map, attempts=0):
   def encode_image(image_path):
     # Check the file extension to determine if it is a HEIC file
     if image_path.lower().endswith('.heic'):
+        # Read the HEIC file
+        heif_file = pyheif.read(image_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,
+        )
         # Load the HEIC image
         image = Image.open(image_path)
         # Convert the image to JPEG format in memory