add some logs

Rodin.py

@@ -20,12 +20,12 @@ def login(email, password):
     try:
         response_data = response.json()
     except json.JSONDecodeError as e:
-        print(response)
+        log("ERROR", f"Error in login: {response}")
         raise e
     
     if 'error' in response_data and response_data['error']:
         raise Exception(response_data['error'])
-    print("Login successful")
+    log("INFO", f"Logged successfully")
     user_uuid = response_data['user_uuid']
     token = response_data['token']
     
@@ -122,6 +122,9 @@ def load_image(img_path):
     image_bytes = byte_io.getvalue()
     return image_bytes
 
+def log(level, info_text):
+    print(f"[ {level} ] - {time.strftime("%Y%m%d_%H%M%S", time.localtime())} - {info_text}")
+
 class Generator:
     def __init__(self, user_id, password) -> None:
         _, self.token = login(user_id, password)
@@ -131,23 +134,23 @@ class Generator:
         
     def preprocess(self, prompt, image_path, cache_image_base64, task_uuid=""):
         if cache_image_base64 and prompt and task_uuid != "":
-            print("Using cached image and prompt...")
+            log("INFO", "Using cached image and prompt...")
             return prompt, cache_image_base64
-        print("Preprocessing image...")
+        log("INFO", "Preprocessing image...")
         success = False
         while not success:
-            print("Trying to preprocess image...")
+            log("INFO", "Loading image...")
             image_file = load_image(image_path)
             if prompt and task_uuid:
                 preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.png", token=self.token)
             else:
                 preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.png", token=self.token)
-            
+            log("INFO", f"Image preprocessed: {preprocess_response.get('statusCode')}")
             if 'error' in preprocess_response:
-                print("Error in image preprocessing:", preprocess_response['error'])
+                log("ERROR", f"Error in image preprocessing: {preprocess_response['error']}")
                 raise RuntimeError
             elif preprocess_response.get("statusCode") == 401:
-                print("Token expired, refreshing token...")
+                log("WARNING", "Token expired. Logging in again...")
                 _, self.token = login(self.user_id, self.password)
                 continue
             else:
@@ -157,13 +160,13 @@ class Generator:
                     processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
                     success = True
                 except Exception as e:
-                    print(preprocess_response)
+                    log("ERROR", f"Error in image preprocessing: {preprocess_response}")
                     raise e
 
         return prompt, processed_image
     
     def generate_mesh(self, prompt, processed_image, task_uuid=""):
-        print("Generating mesh...")
+        log("INFO", "Generating mesh...")
         if task_uuid == "":
             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
@@ -173,7 +176,7 @@ class Generator:
             try:
                 progress_checker.start()
             except Exception as e:
-                print(f"Error in generating mesh: {e}")
+                log("ERROR", f"Error in generating mesh: {e}")
                 time.sleep(5)
                             
             task_uuid = mesh_response['uuid'] # The task_uuid should be same during whole generation process
@@ -193,18 +196,17 @@ class Generator:
             try:
                 checker.start()
             except Exception as e:
-                print(f"Error in updating mesh: {e}")
+                log("ERROR", f"Error in updating mesh: {e}")
                 time.sleep(5)
         
         preview_image = next(reversed(rodin_history(task_uuid, self.token).items()))[1]["preview_image"]
-        # print(f"Preview image URL: {rodin_history(task_uuid, self.token)}")
         response = requests.get(preview_image, stream=True)
         if response.status_code == 200:
             # 创建一个PIL Image对象
             image = Image.open(response.raw)
             # 在这里对image对象进行处理，如显示、保存等
         else:
-            print(f"Can't get the preview image. Status code：{response.status_code}")
+            log("ERROR", f"Error in generating mesh: {response}")
             raise RuntimeError
         response.close()
         return image, task_uuid, crop_image(image, DEFAULT)