Update app.py

app.py

@@ -96,6 +96,42 @@ def image_to_data_uri(image_path: str) -> str:
     data_uri = f"data:image/{img_format};base64,{img_base64}"
     return data_uri
 
+def process_output(output, progress=gr.Progress()) -> str:
+    """Process the output from Replicate API and return a local file path."""
+    try:
+        # 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
+            
+    except Exception as e:
+        print(f"Error processing output: {e}")
+        raise ValueError(f"Failed to process output: {str(e)}")
+
 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:
@@ -153,18 +189,20 @@ def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()
     try:
         progress(0.2, desc="🎨 Preparing images...")
         
-        # Convert all images to data URI format
-        data_uris = []
-        for image_path in images:
+        # Upload all images to get proper URLs
+        image_urls = []
+        for idx, image_path in enumerate(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)
+            
+            progress(0.2 + (0.2 * idx / len(images)), desc=f"📤 Uploading image {idx+1}/{len(images)}...")
+            image_url = upload_image_to_hosting(image_path)
+            image_urls.append(image_url)
         
         # Prepare input for Replicate API with multiple images
         input_data = {
             "prompt": prompt,
-            "image_input": data_uris
+            "image_input": image_urls
         }
 
         progress(0.5, desc="✨ Generating...")
@@ -177,36 +215,9 @@ def run_multi_image_logic(prompt: str, images: List[str], progress=gr.Progress()
         
         progress(0.8, desc="🖼️ Finalizing...")
         
-        # Handle the output - output is already a URL string or FileObject
+        # Handle the output using the process_output function
         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")
             
@@ -226,7 +237,6 @@ css = '''
     margin-bottom: 2rem;
     box-shadow: 0 10px 30px rgba(0,0,0,0.1);
 }
-
 .header-title {
     font-size: 2.5rem !important;
     font-weight: bold;
@@ -235,14 +245,12 @@ css = '''
     margin: 0 !important;
     text-shadow: 2px 2px 4px rgba(0,0,0,0.2);
 }
-
 .header-subtitle {
     color: rgba(255,255,255,0.9);
     text-align: center;
     margin-top: 0.5rem !important;
     font-size: 1.1rem;
 }
-
 /* Card Styling */
 .card {
     background: white;
@@ -251,28 +259,23 @@ css = '''
     box-shadow: 0 4px 6px rgba(0,0,0,0.1);
     border: 1px solid rgba(0,0,0,0.05);
 }
-
 .dark .card {
     background: #1f2937;
     border: 1px solid #374151;
 }
-
 /* Tab Styling */
 .tabs {
     border-radius: 0.5rem;
     overflow: hidden;
     margin-bottom: 1rem;
 }
-
 .tabitem {
     padding: 1rem !important;
 }
-
 button.selected {
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
     color: white !important;
 }
-
 /* Button Styling */
 .generate-btn {
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
@@ -287,12 +290,10 @@ button.selected {
     width: 100% !important;
     margin-top: 1rem !important;
 }
-
 .generate-btn:hover {
     transform: translateY(-2px) !important;
     box-shadow: 0 10px 20px rgba(102, 126, 234, 0.4) !important;
 }
-
 .use-btn {
     background: linear-gradient(135deg, #10b981 0%, #059669 100%) !important;
     border: none !important;
@@ -304,12 +305,10 @@ button.selected {
     transition: all 0.3s ease !important;
     width: 100% !important;
 }
-
 .use-btn:hover {
     transform: translateY(-1px) !important;
     box-shadow: 0 5px 15px rgba(16, 185, 129, 0.4) !important;
 }
-
 /* Input Styling */
 .prompt-input textarea {
     border-radius: 0.5rem !important;
@@ -318,29 +317,24 @@ button.selected {
     font-size: 1rem !important;
     transition: border-color 0.3s ease !important;
 }
-
 .prompt-input textarea:focus {
     border-color: #667eea !important;
     outline: none !important;
 }
-
 .dark .prompt-input textarea {
     border-color: #374151 !important;
     background: #1f2937 !important;
 }
-
 /* Image Output Styling */
 #output {
     border-radius: 0.5rem !important;
     overflow: hidden !important;
     box-shadow: 0 4px 6px rgba(0,0,0,0.1) !important;
 }
-
 /* Progress Bar Styling */
 .progress-bar {
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
 }
-
 /* Examples Styling */
 .examples {
     background: #f9fafb;
@@ -348,11 +342,9 @@ button.selected {
     padding: 1rem;
     margin-top: 1rem;
 }
-
 .dark .examples {
     background: #1f2937;
 }
-
 /* Login Message Styling */
 .login-message {
     background: linear-gradient(135deg, #fef3c7 0%, #fde68a 100%);
@@ -361,23 +353,19 @@ button.selected {
     text-align: center;
     border: 2px solid #f59e0b;
 }
-
 .dark .login-message {
     background: linear-gradient(135deg, #7c2d12 0%, #92400e 100%);
     border-color: #f59e0b;
 }
-
 /* Emoji Animations */
 @keyframes bounce {
     0%, 100% { transform: translateY(0); }
     50% { transform: translateY(-10px); }
 }
-
 .emoji-icon {
     display: inline-block;
     animation: bounce 2s infinite;
 }
-
 /* Responsive Design */
 @media (max-width: 768px) {
     .header-title {