Update app.py

app.py

@@ -10,48 +10,15 @@ import time
 
 
 def tryon(person_img, garment_img, seed, randomize_seed):
-    post_start_time = time.time()
-    if person_img is None or garment_img is None:
-        gr.Warning("Empty image")
-        return None, None, "Empty image"
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-    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')
+    import time
+    import numpy as np
 
-    url = "http://" + os.environ['tryon_url'] + "Submit"
-    token = os.environ['token']
-    cookie = os.environ['Cookie']
-    referer = os.environ['referer']
-    headers = {'Content-Type': 'application/json', 'token': token, 'Cookie': cookie, 'referer': referer}
-    data = {
-        "clothImage": encoded_garment_img,
-        "humanImage": encoded_person_img,
-        "seed": seed
-    }
-    try:
-        response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
-        # print("post response code", response.status_code)
-        if response.status_code == 200:
-            result = response.json()['result']
-            status = result['status']
-            if status == "success":
-                uuid = result['result']
-                # print(uuid)
-    except Exception as err:
-        print(f"Post Exception Error: {err}")
-        raise gr.Error("Too many users, please try again later")
-    post_end_time = time.time()
-    print(f"post time used: {post_end_time-post_start_time}")
+    if person_img is None or garment_img is None:
+        return None, seed, "Empty image"
 
-    get_start_time =time.time()
-    time.sleep(9)
-    Max_Retry = 12
-    result_img = None
-    info = ""
-    err_log = ""
+    time.sleep(2)  # จำลอง delay 2 วินาที
+    mock_result = np.zeros_like(person_img)  # mock image สีดำเท่ากับภาพต้นฉบับ
+    return mock_result, seed, "Mock success"
     for i in range(Max_Retry):
         try:
             url = "http://" + os.environ['tryon_url'] + "Query?taskId=" + uuid