Update app.py

app.py

@@ -2,58 +2,57 @@ import os
 import requests
 import json
 import time
+import cv2
+import base64
+import random
+import numpy as np
 import gradio as gr
 
 MAX_SEED = 999999
 
-# Upload image to a temporary host (like Hugging Face or Imgur)
-def upload_image_to_host(image_data):
-    files = {'file': ('image.png', image_data)}
-    upload_response = requests.post("https://tmpfiles.org/api/v1/upload", files=files)
+# ✅ Hardcoded Pixelcut API key
+pixelcut_api_key = "sk_7c19e4f2ad434a3ebc70fdd85ade5309"
+
+
+# Convert images to PNG format
+def convert_to_png(image):
+    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+
+
+# Encode images to base64 PNG format
+def encode_image(image):
+    image = convert_to_png(image)
+    _, buffer = cv2.imencode('.png', image)
+    return base64.b64encode(buffer).decode('utf-8')
 
-    if upload_response.status_code == 200:
-        return upload_response.json().get("data", {}).get("url", None)
-    else:
-        print("Image upload failed:", upload_response.text)
-        return None
 
 # Main try-on function using Pixelcut API
 def tryon(person_img, garment_img, seed, randomize_seed):
     start_time = time.time()
 
-    # Check if images are provided
+    # 🛑 Check if images are provided
     if person_img is None or garment_img is None:
         return None, None, "Empty image"
 
-    # Handle seed randomization
+    # 🎲 Handle seed randomization
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # Upload images and get URLs
-    person_url = upload_image_to_host(person_img)
-    garment_url = upload_image_to_host(garment_img)
+    # 🔥 Convert images to base64 PNG
+    encoded_person_img = encode_image(person_img)
+    encoded_garment_img = encode_image(garment_img)
 
-    if not person_url or not garment_url:
-        return None, None, "Image upload failed"
-
-    # Load Pixelcut API key from environment
-    pixelcut_api_key = os.environ.get('pixelcut_api_key')
-    if not pixelcut_api_key:
-        return None, None, "API key missing — check environment settings"
-
-    # Setup API endpoint and headers
-    url = "https://api.developer.pixelcut.ai/v1/try-on"
+    # 🌟 Setup Pixelcut API endpoint and headers
+    url = "https://api.developer.pixelcut.ai/v1/remove-background"
     headers = {
         'Content-Type': 'application/json',
         'Accept': 'application/json',
         'X-API-KEY': pixelcut_api_key
     }
 
-    # Setup request payload with URLs
+    # 📩 Setup request payload (matches the curl command structure)
     data = {
-        "person_image_url": person_url,
-        "garment_image_url": garment_url,
-        "seed": seed,
+        "image_url": "https://cdn3.pixelcut.app/product.jpg",
         "format": "png"
     }
 
@@ -64,12 +63,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     try:
         session = requests.Session()
 
-        # Retry loop for "Too many users" error
+        # 🔁 Retry loop for "Too many users" error
         for attempt in range(max_retries):
             response = session.post(url, headers=headers, json=data, timeout=60)
             print("Response code:", response.status_code)
 
-            # If the response is successful
+            # ✅ Success — process the result
             if response.status_code == 200:
                 result_url = response.json().get('result_url', '')
                 print("✅ Success:", result_url)
@@ -77,13 +76,13 @@ def tryon(person_img, garment_img, seed, randomize_seed):
                 # If Pixelcut returns a result URL, fetch the final image
                 if result_url:
                     result_img_data = requests.get(result_url).content
-                    result_img = gr.Image()
-                    result_img.value = result_img_data
+                    result_np = np.frombuffer(result_img_data, np.uint8)
+                    result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED)
                     info = "Success"
                     break
 
             else:
-                # If API returns "Too many users" error, retry
+                # 🔥 If API returns "Too many users" error, retry with delay
                 print("Full response:", response.text)
                 if "Too many users" in response.text:
                     print(f"Attempt {attempt + 1}/{max_retries}: API busy. Retrying in {retry_delay} seconds...")
@@ -93,16 +92,16 @@ def tryon(person_img, garment_img, seed, randomize_seed):
                     break
 
         else:
-            # If all retries fail
+            # ❌ If all retries fail
             info = "API still overloaded — please try again later."
 
-    # Handle timeouts specifically
+    # 🛑 Handle timeouts specifically
     except requests.exceptions.ReadTimeout:
         print("Timeout!")
         info = "Too many users, please try again later"
         raise gr.Error("Too many users, please try again later")
 
-    # Handle any other unexpected errors
+    # ❗ Handle any other unexpected errors
     except Exception as err:
         print(f"❌ Other error: {err}")
         info = "Error, please contact the admin"
@@ -112,7 +111,6 @@ def tryon(person_img, garment_img, seed, randomize_seed):
 
     return result_img, seed, info
 
-
 example_path = os.path.join(os.path.dirname(__file__), 'assets')
 
 garm_list = os.listdir(os.path.join(example_path,"cloth"))