fix a bug

Rodin.py

@@ -133,24 +133,30 @@ class Generator:
         if cache_image_base64 and prompt and task_uuid != "":
             return prompt, cache_image_base64
         print("Preprocessing image...")
-        _, self.token = login(self.user_id, self.password)
-        
-        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)
-        
-        if 'error' in preprocess_response:
-            print("Error in image preprocessing:", preprocess_response['error'])
-        else:
-            if not (prompt and task_uuid):
-                prompt = preprocess_response.get('prompt', 'Default prompt if none returned')   
-            try:             
-                processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
-            except Exception as e:
-                print(preprocess_response)
-                raise e
+        success = False
+        while not success:
+            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)
+            
+            if 'error' in preprocess_response:
+                print("Error in image preprocessing:", preprocess_response['error'])
+                raise RuntimeError
+            elif preprocess_response.get("statusCode") == "401":
+                print("Token expired, refreshing token...")
+                _, self.token = login(self.user_id, self.password)
+                continue
+            else:
+                if not (prompt and task_uuid):
+                    prompt = preprocess_response.get('prompt', 'Default prompt if none returned')   
+                try:             
+                    processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
+                    success = True
+                except Exception as e:
+                    print(preprocess_response)
+                    raise e
 
         return prompt, processed_image