Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 import requests
 from io import BytesIO
 import tempfile
+import base64
 
 # --- Replicate API Configuration ---
 REPLICATE_API_TOKEN = os.getenv("REPLICATE_API_TOKEN")
@@ -28,6 +29,20 @@ def verify_login_status(token: Optional[gr.OAuthToken]) -> bool:
         print(f"Could not verify user's login status: {e}")
         return False
 
+def image_to_data_uri(image_path: str) -> str:
+    """Convert local image file to data URI format."""
+    with open(image_path, "rb") as img_file:
+        img_data = img_file.read()
+        img_base64 = base64.b64encode(img_data).decode('utf-8')
+        
+    # Get the image format
+    img = Image.open(image_path)
+    img_format = img.format.lower() if img.format else 'png'
+    
+    # Create data URI
+    data_uri = f"data:image/{img_format};base64,{img_base64}"
+    return data_uri
+
 def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progress=gr.Progress()) -> str:
     """Handles text-to-image or single image-to-image using Replicate's Nano Banana."""
     try:
@@ -38,12 +53,11 @@ def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progre
             "prompt": prompt
         }
         
-        # If there's an input image, add it to the input
+        # If there's an input image, convert it to data URI
         if image_path:
-            # Upload the local image file and get a URL
-            # For Replicate, we need to provide URLs, not local paths
-            # We'll read the file and create a temporary URL or use file directly
-            input_data["image_input"] = [image_path]
+            # Convert local image to data URI format
+            data_uri = image_to_data_uri(image_path)
+            input_data["image_input"] = [data_uri]
         
         progress(0.5, desc="✨ 생성 중...")
         
@@ -95,10 +109,18 @@ def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()
     try:
         progress(0.2, desc="🎨 이미지 준비 중...")
         
+        # Convert all images to data URI format
+        data_uris = []
+        for image_path in images:
+            if isinstance(image_path, (list, tuple)):
+                image_path = image_path[0]
+            data_uri = image_to_data_uri(image_path)
+            data_uris.append(data_uri)
+        
         # Prepare input for Replicate API with multiple images
         input_data = {
             "prompt": prompt,
-            "image_input": images  # Pass the list of image paths directly
+            "image_input": data_uris
         }
 
         progress(0.5, desc="✨ 생성 중...")