CORS

app.py

@@ -6,6 +6,7 @@ import cv2
 import numpy as np
 from PIL import Image
 from io import BytesIO
+import base64
 import os
 
 # ===============================
@@ -48,7 +49,7 @@ def depth_to_normal(depth):
 # CORE PROCESSING FUNCTION
 # ===============================
 def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
-    # Convert base to numpy
+    # (Your existing depth estimation + pattern blending logic unchanged)
     img_pil = base_image.convert("RGB")
     img_np = np.array(img_pil)
 
@@ -136,26 +137,25 @@ def _process_saree_core(base_image: Image.Image, pattern_image: Image.Image):
     return Image.fromarray(pattern_rgba, mode="RGBA")
 
 # ===============================
-# WRAPPER: ACCEPT BLOBS FROM DATA ARRAY
+# WRAPPER: ACCEPT BYTES OR BASE64
 # ===============================
 def process_saree(data):
     """
-    Accepts [base_blob, pattern_blob] as bytes (e.g., API POST with blobs)
+    Accepts [base_blob, pattern_blob] as bytes OR base64 strings
     """
     if not isinstance(data, (list, tuple)) or len(data) != 2:
         raise ValueError("Expected an array with two elements: [base_blob, pattern_blob]")
 
     base_blob, pattern_blob = data
 
-    if isinstance(base_blob, bytes):
-        base_image = Image.open(BytesIO(base_blob)).convert("RGBA")
-    else:
-        raise ValueError("Base image must be provided as bytes")
+    # Convert base64 to bytes if needed
+    if isinstance(base_blob, str):
+        base_blob = base64.b64decode(base_blob.split(",")[-1])
+    if isinstance(pattern_blob, str):
+        pattern_blob = base64.b64decode(pattern_blob.split(",")[-1])
 
-    if isinstance(pattern_blob, bytes):
-        pattern_image = Image.open(BytesIO(pattern_blob)).convert("RGBA")
-    else:
-        raise ValueError("Pattern image must be provided as bytes")
+    base_image = Image.open(BytesIO(base_blob)).convert("RGBA")
+    pattern_image = Image.open(BytesIO(pattern_blob)).convert("RGBA")
 
     return _process_saree_core(base_image, pattern_image)
 
@@ -167,8 +167,9 @@ iface = gr.Interface(
     inputs=gr.Dataframe(headers=["Base Blob", "Pattern Blob"], type="array"),
     outputs=gr.Image(type="pil", label="Final Saree Output"),
     title="Saree Depth + Pattern Draping (Blob API Compatible)",
-    description="Send image blobs as array [base, pattern] or use Gradio UI for testing."
+    description="Send image blobs (bytes or base64) as array [base, pattern] or use UI for testing."
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    # Enable CORS for Vite
+    iface.launch(server_name="0.0.0.0", share=True)