skkk commited on
Commit
8db01c3
·
1 Parent(s): f49fd2a
Files changed (1) hide show
  1. Rodin.py +12 -6
Rodin.py CHANGED
@@ -52,7 +52,7 @@ def rodin_preprocess_image(generate_prompt, image, name, token):
52
  'Authorization': f'Bearer {token}'
53
  }
54
  response = requests.post(f"{BASE_URL}/task/rodin_mesh_image_process", data=m, headers=headers)
55
- return response.json()
56
 
57
  def crop_image(image, type):
58
  if image == None:
@@ -154,11 +154,17 @@ class Generator:
154
  try_times += 1
155
  image_file = load_image(image_path)
156
  log("INFO", "Image loaded, processing...")
157
- if prompt and task_uuid:
158
- preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name=os.path.basename(image_path), token=self.token)
159
- else:
160
- preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name=os.path.basename(image_path), token=self.token)
161
- log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
 
 
 
 
 
 
162
  if 'error' in preprocess_response:
163
  log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
164
  raise RuntimeError
 
52
  'Authorization': f'Bearer {token}'
53
  }
54
  response = requests.post(f"{BASE_URL}/task/rodin_mesh_image_process", data=m, headers=headers)
55
+ return response, response.json()
56
 
57
  def crop_image(image, type):
58
  if image == None:
 
154
  try_times += 1
155
  image_file = load_image(image_path)
156
  log("INFO", "Image loaded, processing...")
157
+
158
+ try:
159
+ if prompt and task_uuid:
160
+ res, preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name=os.path.basename(image_path), token=self.token)
161
+ else:
162
+ res, preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name=os.path.basename(image_path), token=self.token)
163
+ log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
164
+ except Exception as e:
165
+ log("ERROR", f"Error in image preprocessing: {res}")
166
+ raise gr.Error("Error in image preprocessing, please try again.")
167
+
168
  if 'error' in preprocess_response:
169
  log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
170
  raise RuntimeError