feat: added heic support

app.py

@@ -12,6 +12,15 @@ import logging
 import sys
 import tempfile
 
+# Import and register HEIF support
+try:
+    from pillow_heif import register_heif_opener
+    register_heif_opener()
+    HEIF_SUPPORTED = True
+except ImportError:
+    HEIF_SUPPORTED = False
+    logging.warning("pillow-heif not installed. HEIF/HEIC support disabled.")
+
 # Import Google Drive functionality
 from google_funcs import (
     get_drive_service, 
@@ -255,6 +264,24 @@ def extract_business_card_data(image, model_name="gemini-2.5-flash"):
         logger.warning("Single card extraction returned no results")
         return None
 
+def convert_image_for_processing(image, filename):
+    """Convert image to RGB JPEG format for better compatibility"""
+    try:
+        # Convert to RGB if necessary (HEIF images might be in different modes)
+        if image.mode != 'RGB':
+            image = image.convert('RGB')
+        
+        # Create a buffer for the converted image
+        buffer = io.BytesIO()
+        image.save(buffer, format='JPEG', quality=95)
+        buffer.seek(0)
+        
+        # Return the converted image
+        return Image.open(buffer)
+    except Exception as e:
+        logger.warning(f"Could not convert {filename}: {str(e)}. Using original image.")
+        return image
+
 def process_business_cards(images, model_name="gemini-2.5-flash", save_images=True):
     """Process multiple business card images and create both current run and cumulative Excel files"""
     
@@ -295,9 +322,12 @@ def process_business_cards(images, model_name="gemini-2.5-flash", save_images=Tr
                 image = image_path
                 filename = f"image_{idx+1}.png"
             
-            loaded_images.append(image)
+            # Convert image for better compatibility (especially for HEIF/HEIC)
+            converted_image = convert_image_for_processing(image, filename)
+            
+            loaded_images.append(converted_image)
             filenames.append(filename)
-            logger.debug(f"Successfully loaded image {idx+1}: {filename} (size: {image.size})")
+            logger.debug(f"Successfully loaded image {idx+1}: {filename} (size: {converted_image.size})")
             
         except Exception as e:
             error_msg = f"Error loading {image_path}: {str(e)}"
@@ -663,10 +693,15 @@ with gr.Blocks(title="Business Card Data Extractor") as demo:
     
     with gr.Row():
         with gr.Column():
+            # Define supported file types including phone formats
+            supported_types = [".jpg", ".jpeg", ".png", ".webp", ".bmp"]
+            if HEIF_SUPPORTED:
+                supported_types.extend([".heif", ".heic"])
+            
             image_input = gr.File(
                 label="Upload Business Cards",
                 file_count="multiple",
-                file_types=[".jpg", ".jpeg", ".png", ".webp", ".bmp"]
+                file_types=supported_types
             )
             
             model_selector = gr.Dropdown(

requirements.txt

@@ -4,6 +4,7 @@ google-generativeai==0.8.0
 pandas==2.1.4
 openpyxl==3.1.2
 Pillow==10.2.0
+pillow-heif==0.13.1
 google-auth==2.23.4
 google-auth-oauthlib==1.1.0
 google-api-python-client==2.108.0