lixiang46 commited on
Commit
a079d6c
1 Parent(s): 2240654

add log info

Browse files
Files changed (1) hide show
  1. app.py +12 -6
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("timeout")
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