Update app.py

app.py

@@ -29,8 +29,61 @@ def verify_login_status(token: Optional[gr.OAuthToken]) -> bool:
         print(f"Could not verify user's login status: {e}")
         return False
 
+def upload_image_to_hosting(image_path: str) -> str:
+    """
+    Upload image to hosting service and return URL.
+    Using multiple fallback methods for reliability.
+    """
+    # Open the image
+    img = Image.open(image_path)
+    
+    # Method 1: Try imgbb.com (most reliable)
+    try:
+        buffered = BytesIO()
+        img.save(buffered, format="PNG")
+        buffered.seek(0)
+        img_base64 = base64.b64encode(buffered.getvalue()).decode()
+        
+        response = requests.post(
+            "https://api.imgbb.com/1/upload",
+            data={
+                'key': '6d207e02198a847aa98d0a2a901485a5',  # Free API key
+                'image': img_base64,
+            }
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            if data.get('success'):
+                return data['data']['url']
+    except Exception as e:
+        print(f"imgbb upload failed: {e}")
+    
+    # Method 2: Try 0x0.st (simple and reliable)
+    try:
+        buffered = BytesIO()
+        img.save(buffered, format="PNG")
+        buffered.seek(0)
+        
+        files = {'file': ('image.png', buffered, 'image/png')}
+        response = requests.post("https://0x0.st", files=files)
+        
+        if response.status_code == 200:
+            url = response.text.strip()
+            if url.startswith('http'):
+                return url
+    except Exception as e:
+        print(f"0x0.st upload failed: {e}")
+    
+    # Method 3: Fallback to data URI (last resort)
+    buffered = BytesIO()
+    img.save(buffered, format="PNG")
+    buffered.seek(0)
+    img_base64 = base64.b64encode(buffered.getvalue()).decode()
+    return f"data:image/png;base64,{img_base64}"
+
 def image_to_data_uri(image_path: str) -> str:
-    """Convert local image file to data URI format."""
+    """Convert local image file to data URI format (kept for backwards compatibility)."""
     with open(image_path, "rb") as img_file:
         img_data = img_file.read()
         img_base64 = base64.b64encode(img_data).decode('utf-8')
@@ -53,11 +106,18 @@ def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progre
             "prompt": prompt
         }
         
-        # If there's an input image, convert it to data URI
+        # If there's an input image, upload it to get a proper URL
         if 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.3, desc="📤 Uploading image...")
+            # Upload to hosting service for proper URL
+            image_url = upload_image_to_hosting(image_path)
+            
+            if image_url.startswith('http'):
+                print(f"Image uploaded successfully: {image_url[:50]}...")
+            else:
+                print("Using data URI fallback")
+            
+            input_data["image_input"] = [image_url]
         
         progress(0.5, desc="✨ Generating...")
         
@@ -71,43 +131,17 @@ def run_single_image_logic(prompt: str, image_path: Optional[str] = None, progre
         
         # Handle the output - output is already a URL string or FileObject
         if output:
-            # Check if output has a url attribute (FileObject)
-            if hasattr(output, 'url'):
-                # If url is a method, call it; if it's a property, just access it
-                image_url = output.url() if callable(output.url) else output.url
-            # If output is already a string URL
-            elif isinstance(output, str):
-                image_url = output
-            # If output is a list of URLs
-            elif isinstance(output, list) and len(output) > 0:
-                # Check first item in list
-                first_item = output[0]
-                if hasattr(first_item, 'url'):
-                    image_url = first_item.url() if callable(first_item.url) else first_item.url
-                else:
-                    image_url = first_item
-            else:
-                raise ValueError(f"Unexpected output format from Replicate: {type(output)}")
-            
-            # Download the image from URL
-            response = requests.get(image_url)
-            response.raise_for_status()
-            
-            # Save to temporary file
-            img = Image.open(BytesIO(response.content))
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmpfile:
-                img.save(tmpfile.name)
-                progress(1.0, desc="✅ Complete!")
-                return tmpfile.name
+            return process_output(output, progress)
         else:
             raise ValueError("No output received from Replicate API")
 
     except Exception as e:
         print(f"Error details: {e}")
         print(f"Error type: {type(e)}")
-        print(f"Output value: {output if 'output' in locals() else 'Not set'}")
-        print(f"Output type: {type(output) if 'output' in locals() else 'Not set'}")
-        raise gr.Error(f"Image generation failed: {e}")
+        if 'output' in locals():
+            print(f"Output value: {output}")
+            print(f"Output type: {type(output)}")
+        raise gr.Error(f"Image generation failed: {str(e)[:200]}")
 
 def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()) -> str:
     """
@@ -563,5 +597,4 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
 if __name__ == "__main__":
     demo.queue(max_size=None, default_concurrency_limit=None)
-    demo.launch()
-    
+    demo.launch()