fix a bug

Rodin.py

@@ -88,7 +88,7 @@ def rodin_mesh(prompt, group_uuid, settings, images, name, token):
         'Authorization': f'Bearer {token}'
     }
     response = requests.post(f"{BASE_URL}/task/rodin_mesh", data=m, headers=headers)
-    return response.json()
+    return response
 
 # Convert base64 to binary since the result from `rodin_preprocess_image` is encoded with base64
 def convert_base64_to_binary(base64_string):
@@ -105,7 +105,7 @@ def rodin_update(prompt, task_uuid, token, settings):
         'Authorization': f'Bearer {token}'
     }
     response = requests.post(f"{BASE_URL}/task/rodin_update", data={"uuid": task_uuid, "prompt": prompt, "settings": settings}, headers=headers)
-    return response.json()
+    return response
 
 def load_image(img_path):
     try:
@@ -116,9 +116,9 @@ def load_image(img_path):
     # 按比例缩小图像到长度为1024
     width, height = image.size
     if width > height:
-        scale = 1024 / width
+        scale = 512 / width
     else:
-        scale = 1024 / height
+        scale = 512 / height
     new_width = int(width * scale)
     new_height = int(height * scale)
     resized_image = image.resize((new_width, new_height))
@@ -200,13 +200,14 @@ class Generator:
             settings = {'view_weights': [1]}  # Define weights as per your requirements, for multiple images, use multiple values, e,g [0.5, 0.5]
             images = [processed_image]  # List of images, all the images should be processed first
             
-            mesh_response = rodin_mesh(prompt=prompt, group_uuid=None, settings=settings, images=images, name="images.jpeg", token=self.token)
-            progress_checker = JobStatusChecker(BASE_URL, mesh_response['job']['subscription_key'])
+            res = rodin_mesh(prompt=prompt, group_uuid=None, settings=settings, images=images, name="images.jpeg", token=self.token)
             try:
+                mesh_response = res.json()
+                progress_checker = JobStatusChecker(BASE_URL, mesh_response['job']['subscription_key'])
                 progress_checker.start()
             except Exception as e:
-                log("ERROR", f"Error in generating mesh: {e}")
-                time.sleep(5)
+                log("ERROR", f"Error in generating mesh: {e} and response: {res}")
+                raise gr.Error("Error in generating mesh, please try again later.")
                             
             task_uuid = mesh_response['uuid'] # The task_uuid should be same during whole generation process
         else:
@@ -216,17 +217,15 @@ class Generator:
                 "seed": random.randint(0, 10000), # Customize your seed here
                 "escore": 5.5, # Temprature
             }
-
-            update_response = rodin_update(new_prompt, task_uuid, self.token, settings)
-
-            # Check progress
-            subscription_key = update_response['job']['subscription_key']
-            checker = JobStatusChecker(BASE_URL, subscription_key)
+            res = rodin_update(new_prompt, task_uuid, self.token, settings)
             try:
+                update_response = res.json()
+                subscription_key = update_response['job']['subscription_key']
+                checker = JobStatusChecker(BASE_URL, subscription_key)
                 checker.start()
             except Exception as e:
                 log("ERROR", f"Error in updating mesh: {e}")
-                time.sleep(5)
+                raise gr.Error("Error in generating mesh, please try again later.")
 
         try:
             history = rodin_history(task_uuid, self.token)