Update app.py

app.py

@@ -11,20 +11,79 @@ import base64
 # Set up Replicate API key from environment variable
 os.environ['REPLICATE_API_TOKEN'] = os.getenv('REPLICATE_API_TOKEN')
 
-def upload_to_imgur(image):
+def upload_image_to_hosting(image):
     """
-    Upload image to Imgur and return URL
+    Upload image to multiple hosting services with fallback
     """
+    # Method 1: Try imgbb.com (more reliable than Imgur)
+    try:
+        buffered = BytesIO()
+        image.save(buffered, format="PNG")
+        buffered.seek(0)
+        img_base64 = base64.b64encode(buffered.getvalue()).decode()
+        
+        # imgbb API (free tier)
+        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 freeimage.host
+    try:
+        buffered = BytesIO()
+        image.save(buffered, format="PNG")
+        buffered.seek(0)
+        
+        files = {'source': buffered}
+        response = requests.post(
+            "https://freeimage.host/api/1/upload",
+            files=files,
+            data={'key': '6d207e02198a847aa98d0a2a901485a5'}
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            if 'image' in data and 'url' in data['image']:
+                return data['image']['url']
+    except Exception as e:
+        print(f"freeimage.host upload failed: {e}")
+    
+    # Method 3: Try 0x0.st (simple and reliable)
+    try:
+        buffered = BytesIO()
+        image.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:
+            return response.text.strip()
+    except Exception as e:
+        print(f"0x0.st upload failed: {e}")
+    
+    # Method 4: Original Imgur as last resort
     try:
-        # Convert PIL image to base64
         buffered = BytesIO()
         image.save(buffered, format="PNG")
         buffered.seek(0)
         img_base64 = base64.b64encode(buffered.getvalue()).decode()
         
-        # Imgur API (anonymous upload)
         headers = {
-            'Authorization': 'Client-ID 0d90e8a3e7d8b4e'  # Public client ID
+            'Authorization': 'Client-ID 0d90e8a3e7d8b4e'
         }
         
         response = requests.post(
@@ -35,9 +94,11 @@ def upload_to_imgur(image):
         
         if response.status_code == 200:
             data = response.json()
-            return data['data']['link']
-    except:
-        pass
+            if data.get('success'):
+                return data['data']['link']
+    except Exception as e:
+        print(f"Imgur upload failed: {e}")
+    
     return None
 
 def save_temp_image(image):
@@ -63,45 +124,63 @@ def process_images(prompt, image1, image2=None):
         # Prepare image URLs list
         image_urls = []
         
-        # Try to upload images to Imgur to get public URLs
-        url1 = upload_to_imgur(image1)
+        # Upload first image
+        print("Uploading first image...")
+        url1 = upload_image_to_hosting(image1)
         if url1:
             image_urls.append(url1)
+            print(f"Image 1 uploaded: {url1}")
         else:
-            # If Imgur fails, you need to use another service or local server
-            # For now, we'll save locally (note: this won't work with Replicate)
-            return None, "⚠️ Image hosting failed. Please try again or use a different hosting service."
+            # If all hosting services fail, use base64 data URI as fallback
+            buffered = BytesIO()
+            image1.save(buffered, format="PNG")
+            buffered.seek(0)
+            img_base64 = base64.b64encode(buffered.getvalue()).decode()
+            data_uri = f"data:image/png;base64,{img_base64}"
+            image_urls.append(data_uri)
+            print("Using base64 data URI for image 1")
         
+        # Upload second image if provided
         if image2:
-            url2 = upload_to_imgur(image2)
+            print("Uploading second image...")
+            url2 = upload_image_to_hosting(image2)
             if url2:
                 image_urls.append(url2)
+                print(f"Image 2 uploaded: {url2}")
+            else:
+                buffered = BytesIO()
+                image2.save(buffered, format="PNG")
+                buffered.seek(0)
+                img_base64 = base64.b64encode(buffered.getvalue()).decode()
+                data_uri = f"data:image/png;base64,{img_base64}"
+                image_urls.append(data_uri)
+                print("Using base64 data URI for image 2")
         
-        # Prepare input exactly as shown in your example
+        # Prepare input exactly as shown in the reference
         input_data = {
             "prompt": prompt,
             "image_input": image_urls
         }
         
-        print(f"Sending to Replicate: {input_data}")
+        print(f"Sending to Replicate: prompt='{prompt}', images={len(image_urls)}")
         
-        # Run the model exactly as in your example
+        # Run the model
         output = replicate.run(
             "google/nano-banana",
             input=input_data
         )
         
-        # Process output - based on your example, output should have url() and read() methods
+        # Process output
         if output is None:
             return None, "❌ No output received from model"
         
         # Try different methods to get the image
         try:
-            # Method 1: Using read() method as in your example
+            # Method 1: Using read() method
             if hasattr(output, 'read'):
                 img_data = output.read()
                 img = Image.open(BytesIO(img_data))
-                return img, "✅ Image generated successfully using nano-banana!"
+                return img, "✅ Free Nano Banana generated your image successfully! 🍌"
         except:
             pass
         
@@ -112,7 +191,7 @@ def process_images(prompt, image1, image2=None):
                 response = requests.get(output_url, timeout=30)
                 if response.status_code == 200:
                     img = Image.open(BytesIO(response.content))
-                    return img, "✅ Image generated successfully using nano-banana!"
+                    return img, "✅ Free Nano Banana generated your image successfully! 🍌"
         except:
             pass
         
@@ -127,14 +206,14 @@ def process_images(prompt, image1, image2=None):
             response = requests.get(output_url, timeout=30)
             if response.status_code == 200:
                 img = Image.open(BytesIO(response.content))
-                return img, "✅ Image generated successfully using nano-banana!"
+                return img, "✅ Free Nano Banana generated your image successfully! 🍌"
         
         return None, f"❌ Could not process output. Output type: {type(output)}"
         
     except replicate.exceptions.ReplicateError as e:
         error_msg = str(e)
         if "502" in error_msg:
-            return None, "❌ Server error (502). The model might be unavailable. Please try again later."
+            return None, "❌ Server error (502). The nano-banana model might be temporarily unavailable."
         elif "404" in error_msg:
             return None, "❌ Model 'google/nano-banana' not found. Please verify the model exists."
         elif "401" in error_msg or "403" in error_msg:
@@ -190,7 +269,7 @@ css = """
 # Build the Gradio interface
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gr.HTML("""
-        <div class="header-text">🎨 AI Image Style Transfer Studio</div>
+        <div class="header-text">🍌 Free Nano Banana</div>
         <div class="description-text">
             Upload 1-2 images and describe how you want them styled.
             The AI will create a beautiful transformation using nano-banana model!
@@ -230,7 +309,8 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
             #### 💡 Tips:
             - Upload high-quality images for best results
             - Be specific in your style description
-            - Model: google/nano-banana
+            - Model: google/nano-banana 🍌
+            - Free image hosting included!
             """)
         
         with gr.Column(scale=1):