lixiang46
commited on
Commit
•
a079d6c
1
Parent(s):
2240654
add log info
Browse files
app.py
CHANGED
@@ -32,15 +32,15 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
32 |
}
|
33 |
try:
|
34 |
response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
|
35 |
-
print("post response code", response.status_code)
|
36 |
if response.status_code == 200:
|
37 |
result = response.json()['result']
|
38 |
status = result['status']
|
39 |
if status == "success":
|
40 |
uuid = result['result']
|
41 |
-
print(uuid)
|
42 |
except Exception as err:
|
43 |
-
print(f"Error: {err}")
|
44 |
raise gr.Error("Too many users, please try again later")
|
45 |
post_end_time = time.time()
|
46 |
print(f"post time used: {post_end_time-post_start_time}")
|
@@ -49,11 +49,12 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
49 |
time.sleep(9)
|
50 |
Max_Retry = 10
|
51 |
result_img = None
|
|
|
52 |
for i in range(Max_Retry):
|
53 |
try:
|
54 |
url = "http://" + os.environ['tryon_url'] + "Query?taskId=" + uuid
|
55 |
response = requests.get(url, headers=headers, timeout=15)
|
56 |
-
print("get response code", response.status_code)
|
57 |
if response.status_code == 200:
|
58 |
result = response.json()['result']
|
59 |
status = result['status']
|
@@ -65,18 +66,23 @@ def tryon(person_img, garment_img, seed, randomize_seed):
|
|
65 |
info = "Success"
|
66 |
break
|
67 |
elif status == "error":
|
|
|
68 |
raise gr.Error("Too many users, please try again later")
|
69 |
else:
|
70 |
print(response.text)
|
71 |
info = "URL error, pleace contact the admin"
|
72 |
except requests.exceptions.ReadTimeout:
|
73 |
-
print("
|
74 |
info = "Too many users, please try again later"
|
75 |
except Exception as err:
|
76 |
-
print(f"Error: {err}")
|
77 |
time.sleep(1)
|
78 |
get_end_time = time.time()
|
79 |
print(f"get time used: {get_end_time-get_start_time}")
|
|
|
|
|
|
|
|
|
80 |
|
81 |
return result_img, seed, info
|
82 |
|
|
|
32 |
}
|
33 |
try:
|
34 |
response = requests.post(url, headers=headers, data=json.dumps(data), timeout=50)
|
35 |
+
# print("post response code", response.status_code)
|
36 |
if response.status_code == 200:
|
37 |
result = response.json()['result']
|
38 |
status = result['status']
|
39 |
if status == "success":
|
40 |
uuid = result['result']
|
41 |
+
# print(uuid)
|
42 |
except Exception as err:
|
43 |
+
print(f"Post Exception Error: {err}")
|
44 |
raise gr.Error("Too many users, please try again later")
|
45 |
post_end_time = time.time()
|
46 |
print(f"post time used: {post_end_time-post_start_time}")
|
|
|
49 |
time.sleep(9)
|
50 |
Max_Retry = 10
|
51 |
result_img = None
|
52 |
+
info = ""
|
53 |
for i in range(Max_Retry):
|
54 |
try:
|
55 |
url = "http://" + os.environ['tryon_url'] + "Query?taskId=" + uuid
|
56 |
response = requests.get(url, headers=headers, timeout=15)
|
57 |
+
# print("get response code", response.status_code)
|
58 |
if response.status_code == 200:
|
59 |
result = response.json()['result']
|
60 |
status = result['status']
|
|
|
66 |
info = "Success"
|
67 |
break
|
68 |
elif status == "error":
|
69 |
+
print(f"Status is Error")
|
70 |
raise gr.Error("Too many users, please try again later")
|
71 |
else:
|
72 |
print(response.text)
|
73 |
info = "URL error, pleace contact the admin"
|
74 |
except requests.exceptions.ReadTimeout:
|
75 |
+
print("Http Timeout")
|
76 |
info = "Too many users, please try again later"
|
77 |
except Exception as err:
|
78 |
+
print(f"Get Exception Error: {err}")
|
79 |
time.sleep(1)
|
80 |
get_end_time = time.time()
|
81 |
print(f"get time used: {get_end_time-get_start_time}")
|
82 |
+
print(f"all time used: {get_end_time-get_start_time+post_end_time-post_start_time}")
|
83 |
+
if info == "":
|
84 |
+
print(f"No image after {Max_Retry} retries")
|
85 |
+
info = "Too many users, please try again later"
|
86 |
|
87 |
return result_img, seed, info
|
88 |
|