Update app.py

app.py

@@ -15,6 +15,13 @@ garm_list_path = [os.path.join(example_path, "cloth", garm) for garm in garm_lis
 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]
 
+# API details
+base_url = "https://huggingface.co/spaces/zaghamrasool/Z-Virtual-Try-On"
+upload_image_url = f"{base_url}/upload_image"
+create_save_task_url = f"{base_url}/create_save_task"
+execute_task_url = f"{base_url}/execute_task"
+query_task_url = f"{base_url}/query_task"
+
 def tryon(person_img, garment_img, seed, randomize_seed):
     post_start_time = time.time()
     if person_img is None or garment_img is None:
@@ -23,12 +30,13 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     
+    # Encode images
     encoded_person_img = cv2.imencode('.jpg', cv2.cvtColor(person_img, cv2.COLOR_RGB2BGR))[1].tobytes()
     encoded_person_img = base64.b64encode(encoded_person_img).decode('utf-8')
     encoded_garment_img = cv2.imencode('.jpg', cv2.cvtColor(garment_img, cv2.COLOR_RGB2BGR))[1].tobytes()
     encoded_garment_img = base64.b64encode(encoded_garment_img).decode('utf-8')
 
-    url = "https://huggingface.co/spaces/zaghamrasool/Z-Virtual-Try-On"
+    # Prepare data
     data = {
         "clothImage": encoded_garment_img,
         "humanImage": encoded_person_img,
@@ -37,11 +45,16 @@ def tryon(person_img, garment_img, seed, randomize_seed):
 
     uuid = None
     try:
-        response = requests.post(url, data=json.dumps(data), timeout=50)
+        # First API call to create task
+        response = requests.post(create_save_task_url, data=json.dumps(data), timeout=50)
         if response.status_code == 200:
             result = response.json().get('result', {})
             if result.get('status') == "success":
-                uuid = result.get('result')
+                uuid = result.get('taskId')  # Use taskId for querying
+            else:
+                raise Exception("Failed to create task, no task ID received.")
+        else:
+            raise Exception(f"Failed to create task. Status Code: {response.status_code}")
     except Exception as err:
         print(f"Post Exception Error: {err}")
         raise gr.Error("Too many users, please try again later")
@@ -49,7 +62,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     post_end_time = time.time()
     print(f"post time used: {post_end_time - post_start_time}")
 
-    # Retry loop
+    # Retry loop to query task status
     get_start_time = time.time()
     time.sleep(5)
     Max_Retry = 20
@@ -63,7 +76,7 @@ def tryon(person_img, garment_img, seed, randomize_seed):
     else:
         for i in range(Max_Retry):
             try:
-                url = f"https://huggingface.co/spaces/zaghamrasool/Z-Virtual-Try-On/SubmitQuery?taskId={uuid}"
+                url = f"{query_task_url}?taskId={uuid}"
                 response = requests.get(url, timeout=20)
                 if response.status_code == 200:
                     result = response.json()['result']
@@ -148,11 +161,9 @@ with gr.Blocks(css=css) as Tryon:
     with gr.Column(elem_id="col-showcase"):
         gr.HTML("<div style='text-align: center; font-size: 20px;'>Virtual try-on examples in pairs of person and garment images</div>")
         gr.Examples(
-            examples=[
-                ["assets/examples/model2.png", "assets/examples/garment2.png", "assets/examples/result2.png"],
-                ["assets/examples/model3.png", "assets/examples/garment3.png", "assets/examples/result3.png"],
-                ["assets/examples/model1.png", "assets/examples/garment1.png", "assets/examples/result1.png"]
-            ],
+            examples=[["assets/examples/model2.png", "assets/examples/garment2.png", "assets/examples/result2.png"],
+                      ["assets/examples/model3.png", "assets/examples/garment3.png", "assets/examples/result3.png"],
+                      ["assets/examples/model1.png", "assets/examples/garment1.png", "assets/examples/result1.png"]],
             inputs=[imgs, garm_img, image_out]
         )