HEIC support

app.py

@@ -7,6 +7,8 @@ from transformers import AutoProcessor, AutoModelForDepthEstimation
 import tempfile
 from stl import mesh
 from sklearn.decomposition import PCA
+import pillow_heif
+pillow_heif.register_heif_opener()
 
 MAX_RESOLUTION = 1.5e6
 MODEL_ID = "depth-anything/Depth-Anything-V2-Large-hf"
@@ -16,7 +18,7 @@ model = AutoModelForDepthEstimation.from_pretrained(MODEL_ID).to(device)
 print("Model loaded successfully.")
 
 def create_3d_model(
-    input_image: np.ndarray,
+    input_filepath: str,
     texture_strength: float,
     max_z: float,
     min_z: float,
@@ -25,9 +27,15 @@ def create_3d_model(
     depth_map_smoothing: int,
     texture_smoothing: int
 ):
-    if input_image is None: raise gr.Error("Please upload an image.")
+    if input_filepath is None: raise gr.Error("Please upload an image.")
     if max_z <= min_z: raise gr.Error("Max Z-height must be greater than Min Z-height.")
 
+    try:
+        image_pil = Image.open(input_filepath)
+        input_image = np.array(image_pil.convert("RGB"))
+    except Exception as e:
+        raise gr.Error(f"Could not open image file. Error: {e}")
+
     # resize large images
     h, w, _ = input_image.shape
     if h * w > MAX_RESOLUTION:
@@ -122,7 +130,7 @@ with gr.Blocks(theme='base') as demo:
     gr.Markdown("# Image to 3D Relief Generator")
     with gr.Row():
         with gr.Column(scale=1):
-            input_image = gr.Image(type="numpy", label="Upload Image")
+            input_image = gr.Image(type="filepath", label="Upload Image")
             gr.Markdown("### Model Parameters")
             texture_strength = gr.Slider(minimum=0.0, maximum=1.0, value=0.1, step=0.001, label="Brightness Texture Strength")
             depth_map_smoothing = gr.Slider(minimum=0, maximum=51, value=0, step=1, label="Depth Map Smoothing", info="Smooths the base geometry. 0 = none.")
@@ -150,4 +158,4 @@ with gr.Blocks(theme='base') as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)