Update app.py

app.py

@@ -1,26 +1,21 @@
 import os
-import cv2
-import base64
-import random
 import requests
 import json
 import time
-import numpy as np
 import gradio as gr
 
 MAX_SEED = 999999
 
-# Convert images to PNG format (ensure supported format)
-def convert_to_png(image):
-    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-
-# Encode images to base64 (forces PNG format)
-def encode_image(image):
-    # Ensure image is PNG
-    image = convert_to_png(image)
-    _, buffer = cv2.imencode('.png', image)
-    return base64.b64encode(buffer).decode('utf-8')
+# 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)
 
+    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):
@@ -34,9 +29,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
-    # Convert images to base64 PNG
-    encoded_person_img = encode_image(person_img)
-    encoded_garment_img = encode_image(garment_img)
+    # Upload images and get URLs
+    person_url = upload_image_to_host(person_img)
+    garment_url = upload_image_to_host(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')
@@ -51,12 +49,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
         'X-API-KEY': pixelcut_api_key
     }
 
-    # Setup request payload with PNG format
+    # Setup request payload with URLs
     data = {
-        "person_image": encoded_person_img,
-        "garment_image": encoded_garment_img,
+        "person_image_url": person_url,
+        "garment_image_url": garment_url,
         "seed": seed,
-        "format": "png"  # Ensure API gets PNG format
+        "format": "png"
     }
 
     result_img = None
@@ -79,8 +77,8 @@ 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_np = np.frombuffer(result_img_data, np.uint8)
-                    result_img = cv2.imdecode(result_np, cv2.IMREAD_UNCHANGED)
+                    result_img = gr.Image()
+                    result_img.value = result_img_data
                     info = "Success"
                     break
 
@@ -115,7 +113,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"))