Update app.py

app.py

@@ -1,65 +1,61 @@
 import requests
-import gradio as gr
+import base64
 
-# Set your Pixelcut API key directly
+# --- Configuration ---
 API_KEY = "sk_7c19e4f2ad434a3ebc70fdd85ade5309"
+PERSON_IMAGE_PATH = "path_to_person_image.jpg"
+GARMENT_IMAGE_PATH = "path_to_garment_image.jpg"
+
+# --- Helper Function: Read Image and Convert to Base64 ---
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode("utf-8")
+
+# --- Step 1: Remove Background from Person Image ---
+def remove_background(image_path):
+    with open(image_path, "rb") as image_file:
+        response = requests.post(
+            "https://api.developer.pixelcut.ai/v1/remove-background",
+            headers={
+                "Accept": "application/json",
+                "X-API-KEY": API_KEY,
+            },
+            files={"image": image_file},
+            data={"format": "png"},
+        )
 
-# Function to remove background from a local image file
-def remove_bg(image_path):
-    try:
-        with open(image_path, "rb") as image_file:
-            response = requests.post(
-                "https://api.developer.pixelcut.ai/v1/remove-background",
-                headers={
-                    "Accept": "application/json",
-                    "X-API-KEY": API_KEY,
-                },
-                files={"image": image_file},
-                data={"format": "png"},
-            )
-
-        # Handle API response
-        if response.status_code == 200:
-            result = response.json()
-            return result["output_url"]  # Return the output image URL from API
-        else:
-            return f"Error: {response.status_code} - {response.json()['error']}"
-
-    except Exception as e:
-        return f"Exception: {e}"
-
-
-# Define the Tryonn function without CSS
-def tryonn(person_image, garment_image):
-    # Process the images using the Pixelcut API
-    person_result = remove_bg(person_image)
-    garment_result = remove_bg(garment_image)
+    if response.status_code == 200:
+        return response.json().get("url")
+    else:
+        print("Error:", response.json())
+        return None
+
+# --- Step 2: Try-On Function ---
+def tryonn(person_image_path, garment_image_path):
+    # Get background-removed URLs
+    person_url = remove_background(person_image_path)
+    garment_url = remove_background(garment_image_path)
+
+    if not person_url or not garment_url:
+        print("Failed to process images!")
+        return
+
+    # Make the try-on API call
+    tryon_response = requests.post(
+        "https://api.example.com/v1/tryon",
+        headers={"Accept": "application/json", "X-API-KEY": API_KEY},
+        json={"person_image_url": person_url, "garment_image_url": garment_url},
+    )
 
-    # If both images are processed successfully, combine them (mock-up display)
-    if "Error" not in person_result and "Error" not in garment_result:
-        return person_result, garment_result, "✅ Success!"
+    if tryon_response.status_code == 200:
+        tryon_result_url = tryon_response.json().get("result_url")
+        print("Try-On Success! Result:", tryon_result_url)
     else:
-        return person_result, garment_result, "❌ Failed, check errors."
+        print("Try-On Failed:", tryon_response.json())
 
-# Gradio interface setup (CSS removed)
-with gr.Blocks() as Tryon:
-    gr.Markdown("## Upload a Person Image and Garment Image")
-    with gr.Row():
-        person_input = gr.Image(type="filepath", label="Person Image")
-        garment_input = gr.Image(type="filepath", label="Garment Image")
-    output_person = gr.Image(label="Processed Person Image")
-    output_garment = gr.Image(label="Processed Garment Image")
-    status = gr.Textbox(label="Status")
-
-    submit_button = gr.Button("Run Try-On")
-
-    submit_button.click(
-        fn=tryonn,
-        inputs=[person_input, garment_input],
-        outputs=[output_person, output_garment, status],
-    )
+# --- Execute the try-on function ---
+tryonn(PERSON_IMAGE_PATH, GARMENT_IMAGE_PATH)
 
-Tryon.launch()
 
 
 example_path = os.path.join(os.path.dirname(__file__), 'assets')