skkk commited on
Commit
ab026aa
1 Parent(s): 809deb1
Files changed (1) hide show
  1. Rodin.py +10 -3
Rodin.py CHANGED
@@ -17,7 +17,11 @@ def login(email, password):
17
  payload['email'] = email
18
 
19
  response = requests.post(f"{BASE_URL}/user/login", json=payload)
20
- response_data = response.json()
 
 
 
 
21
 
22
  if 'error' in response_data and response_data['error']:
23
  raise Exception(response_data['error'])
@@ -137,8 +141,11 @@ class Generator:
137
  print("Error in image preprocessing:", preprocess_response['error'])
138
  else:
139
  if not (prompt and task_uuid):
140
- prompt = preprocess_response.get('prompt', 'Default prompt if none returned')
141
- processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
 
 
 
142
 
143
  return prompt, processed_image
144
 
 
17
  payload['email'] = email
18
 
19
  response = requests.post(f"{BASE_URL}/user/login", json=payload)
20
+ try:
21
+ response_data = response.json()
22
+ except json.JSONDecodeError as e:
23
+ print(response)
24
+ raise e
25
 
26
  if 'error' in response_data and response_data['error']:
27
  raise Exception(response_data['error'])
 
141
  print("Error in image preprocessing:", preprocess_response['error'])
142
  else:
143
  if not (prompt and task_uuid):
144
+ prompt = preprocess_response.get('prompt', 'Default prompt if none returned')
145
+ try:
146
+ processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
147
+ except Exception as e:
148
+ print(preprocess_response)
149
 
150
  return prompt, processed_image
151