Update app.py

app.py

@@ -16,11 +16,9 @@ pixelcut_api_key = "sk_299d9c6e36d240cbb3dd65fcbac947a4"
 # ✅ ImgBB API Key (for uploading images to get valid URLs)
 imgbb_api_key = "03a2ddf1ffa5df33a3999cf20c2fb20f"
 
-
-# 🎯 Convert image to RGB format (fixes blue tint!)
-def convert_to_rgb(image):
-    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
-
+# 🎯 Convert image to PNG format  
+def convert_to_png(image):
+    return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)  # This causes the blue tint
 
 # 🔥 Resize large images to prevent upload failures (ImgBB limit: 32MB)
 def resize_image(image, max_size=1024):
@@ -30,7 +28,6 @@ def resize_image(image, max_size=1024):
         return cv2.resize(image, (int(width * scale), int(height * scale)))
     return image
 
-
 # 🛠️ Upload images to ImgBB (fixed payload!)
 def upload_image_to_imgbb(image_data):
     url = f"https://api.imgbb.com/1/upload?key={imgbb_api_key}"
@@ -43,32 +40,37 @@ def upload_image_to_imgbb(image_data):
         print("❌ ImgBB upload failed:", response.text)
         return None
 
-
 # 🚀 Main try-on function using Pixelcut API
 def tryon(person_img, cloth_img, seed, random_seed):
     import cv2
     import numpy as np
 
-    # 🎲 Handle seed
+    # Handle seed
     if random_seed:
         import random
         seed = random.randint(0, 1000000)
 
-    # 🔥 Convert images to RGB and normalize
-    person_img = convert_to_rgb(person_img)
-    cloth_img = convert_to_rgb(cloth_img)
+    # Convert images to **BGR** (causes blue tint)
+    person_img = cv2.cvtColor(person_img, cv2.COLOR_BGR2RGB)
+    cloth_img = cv2.cvtColor(cloth_img, cv2.COLOR_BGR2RGB)
 
-    # ✂️ Resize to 256x256
+    # Resize to 256x256
     person_img = cv2.resize(person_img, (256, 256))
     cloth_img = cv2.resize(cloth_img, (256, 256))
 
-    # 🎯 Simulate "Success" output (model processing should go here)
+    # (Simulated output — blue tint included)
     output_img = person_img.copy()
 
-    # 🔵 **Blue Tint Fix: Ensure correct color space in final output**
-    output_img = cv2.cvtColor(output_img, cv2.COLOR_BGR2RGB)
+    return output_img, seed, "Success"
+
+# 🔧 Paths for examples
+example_path = os.path.join(os.path.dirname(__file__), "assets")
+
+garm_list = os.listdir(os.path.join(example_path, "cloth"))
+garm_list_path = [os.path.join(example_path, "cloth", garm) for garm in garm_list]
 
-    return output_img, seed, "✅ Success"
+human_list = os.listdir(os.path.join(example_path, "human"))
+human_list_path = [os.path.join(example_path, "human", human) for human in human_list]
 
 css="""
 #col-left {