init project

.gitattributes

@@ -33,3 +33,11 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/garment/try-on-garment-2.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/garment/try-on-garment-4.jpg filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/garment/try-on-garment.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/models/try-on-model-2.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/models/try-on-model-3.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/models/try-on-model-4.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/models/try-on-model.png filter=lfs diff=lfs merge=lfs -text
+images[[:space:]]/results/after-try-on.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,157 @@
+import gradio as gr
+import base64
+from PIL import Image
+from io import BytesIO
+import os
+from tryon_api_client import TryOnAPIClient
+
+# Configuration
+API_URL = os.environ.get("API_BASE_URL", "http://localhost:5001")
+client = TryOnAPIClient(base_url=API_URL)
+
+def process_tryon(person_img, garment_img, garment_type):
+    # Validate inputs
+    if not person_img or not garment_img:
+        raise gr.Error("⚠️ Please upload both Person and Garment images.")
+    
+    # Validate file formats
+    try:
+        valid_extensions = ('.png', '.jpg', '.jpeg')
+        if not person_img.lower().endswith(valid_extensions):
+            raise gr.Error("⚠️ Person image must be PNG, JPEG, or JPG format.")
+        if not garment_img.lower().endswith(valid_extensions):
+            raise gr.Error("⚠️ Garment image must be PNG, JPEG, or JPG format.")
+    except AttributeError:
+        raise gr.Error("⚠️ Invalid image file. Please upload valid image files.")
+    
+    try:
+        # 1. Create Task
+        try:
+            task_data = client.create_task(person_img, garment_img, garment_type)
+        except ConnectionError:
+            raise gr.Error("🔌 Cannot connect to the Try-On service. Please check if the API server is running.")
+        except TimeoutError:
+            raise gr.Error("⏱️ Connection timeout. The server is taking too long to respond.")
+        except Exception as e:
+            raise gr.Error(f"❌ Failed to create try-on task: {str(e)}")
+        
+        task_id = task_data.get('task_id')
+        if not task_id:
+             raise gr.Error("❌ No task ID returned from API. Please try again.")
+             
+        # 2. Wait for Completion using Client logic
+        try:
+            result = client.wait_for_completion(task_id)
+        except TimeoutError:
+            raise gr.Error("⏱️ Try-on process is taking too long. Please try again with smaller images.")
+        except Exception as e:
+            raise gr.Error(f"❌ Failed while processing try-on: {str(e)}")
+        
+        # 3. Process result
+        image_base64 = result.get('image_base64')
+        if image_base64:
+            try:
+                return base64_to_image(image_base64)
+            except Exception as e:
+                raise gr.Error(f"❌ Failed to decode result image: {str(e)}")
+        else:
+            raise gr.Error("❌ Try-on completed but no image was generated. Please try again.")
+
+    except gr.Error:
+        # Re-raise Gradio errors as-is
+        raise
+    except Exception as e:
+        # Catch any unexpected errors
+        raise gr.Error(f"❌ Unexpected error occurred: {str(e)}")
+
+def base64_to_image(base64_string):
+    try:
+        image_data = base64.b64decode(base64_string)
+        image = Image.open(BytesIO(image_data))
+        return image
+    except base64.binascii.Error:
+        raise ValueError("Invalid base64 string format")
+    except Exception as e:
+        raise ValueError(f"Cannot decode image: {str(e)}")
+
+# UI Layout
+with gr.Blocks(title="Virtual Try-On") as app:
+    gr.Markdown("# 👕 Virtual Try-On Demo")
+    gr.Markdown("Upload a person image and a garment image to see the magic")
+    gr.HTML(
+    """
+    <center>
+        <a href="https://rapidapi.com/aiproviderlabs/api/virtual-outfit-try-on-api"
+           target="_blank"
+           rel="noopener noreferrer"
+           style="display:inline-flex;align-items:center;gap:8px;padding:8px 16px;border-radius:8px;border:1px solid #ccc;text-decoration:none;font-family:sans-serif;">
+            <img src="https://files.readme.io/9336831-small-rapid-logo-favicon.png"
+                 style="width:16px;height:16px;">
+            Check out the API @ RapidAPI.com
+        </a>
+    </center>
+    """
+)
+    
+    with gr.Row():
+        gr.Markdown("### Before - After Example")
+        
+    with gr.Row():
+        gr.Image(value="images /models/try-on-model.png", label="Model", interactive=False, height=300)
+        gr.Image(value="images /garment/try-on-garment.png", label="Garment - Full Set", interactive=False, height=300)
+        gr.Image(value="images /results/after-try-on.jpg", label="Result", interactive=False, height=300)
+
+    with gr.Row():
+        gr.Markdown("### Try-On Example")
+
+    with gr.Row():
+        with gr.Column():
+            person_input = gr.Image(
+                type="filepath", 
+                label="Person Image (PNG, JPEG, JPG only)", 
+                height=400
+            )
+            gr.Examples(
+                examples=[
+                    "images /models/try-on-model.png",
+                    "images /models/try-on-model-2.png",
+                    "images /models/try-on-model-3.png",
+                    "images /models/try-on-model-4.png",
+                ],
+                inputs=person_input
+            )
+            garment_input = gr.Image(
+                type="filepath", 
+                label="Garment Image (PNG, JPEG, JPG only)", 
+                height=400
+            )
+            gr.Examples(
+                examples=[
+                    "images /garment/try-on-garment.png",
+                    "images /garment/try-on-garment-2.png",
+                    "images /garment/try-on-garment-3.jpg",
+                    "images /garment/try-on-garment-4.jpg",
+                ],
+                inputs=garment_input
+            )
+            
+            garment_type_input = gr.Radio(
+                choices=["top", "bottom", "fullset"], 
+                value="top", 
+                label="Garment Type",
+                info="Select the type of clothing you are trying on."
+            )
+            
+            run_btn = gr.Button("🚀 Start Try-On", variant="primary")
+            
+        with gr.Column():
+            result_output = gr.Image(label="Try-On Result", height=400)
+    
+    run_btn.click(
+        fn=process_tryon,
+        inputs=[person_input, garment_input, garment_type_input],
+        outputs=result_output
+    )
+
+if __name__ == "__main__":
+    app.launch(server_name="0.0.0.0", server_port=7860)

tryon_api_client.py

@@ -0,0 +1,83 @@
+import requests
+import time
+import os
+
+class TryOnAPIClient:
+    def __init__(self, base_url=None):
+        self.base_url = base_url or os.environ.get("API_BASE_URL", "http://localhost:5001")
+        self.headers = {}
+        
+        rapid_key = os.environ.get("X_RAPIDAPI_KEY")
+        rapid_host = os.environ.get("X_RAPIDAPI_HOST")
+        
+        if rapid_key:
+            self.headers["x-rapidapi-key"] = rapid_key
+        if rapid_host:
+            self.headers["x-rapidapi-host"] = rapid_host
+
+    def create_task(self, person_image_path, garment_image_path, garment_type):
+        """
+        Uploads images to create a try-on task.
+        """
+        url = f"{self.base_url}/try-on/file"
+        
+        files = {
+            'person_image': open(person_image_path, 'rb'),
+            'garment_image': open(garment_image_path, 'rb')
+        }
+        
+        data = {
+            'garment_type': garment_type
+        }
+
+        try:
+            response = requests.post(url, files=files, data=data, headers=self.headers, timeout=60)
+            
+            if response.status_code != 200:
+                error_msg = response.json().get('error', 'Unknown Error')
+                raise Exception(f"Task Creation Failed: {error_msg}")
+                
+            return response.json()
+            
+        finally:
+            # Ensure files are closed
+            files['person_image'].close()
+            files['garment_image'].close()
+
+    def check_progress(self, task_id):
+        """
+        Checks the progress of a task.
+        """
+        url = f"{self.base_url}/try-on/tasks/{task_id}/progress"
+        response = requests.get(url, headers=self.headers, timeout=30)
+        
+        if response.status_code != 200:
+             raise Exception(f"Error checking progress: {response.text}")
+             
+        return response.json()
+
+    def wait_for_completion(self, task_id, timeout=600, poll_interval=2):
+        """
+        Polls for task completion.
+        Returns the result dictionary (with image_base64) or raises Exception.
+        """
+        start_time = time.time()
+        
+        while (time.time() - start_time) < timeout:
+            try:
+                data = self.check_progress(task_id)
+            except Exception as e:
+                print(f"Warning during poll: {e}")
+                time.sleep(poll_interval)
+                continue
+                
+            status = data.get('status')
+            
+            if status == 'FINISHED':
+                return data
+            elif status == 'FAILED':
+                raise Exception("Task failed on server.")
+                
+            time.sleep(poll_interval)
+            
+        raise Exception("Operation timed out.")