Fix a bug of preprocess

Rodin.py

@@ -6,6 +6,7 @@ import random
 import base64
 import io
 from PIL import Image
+from io import BytesIO
 from requests_toolbelt.multipart.encoder import MultipartEncoder
 from constant import *
 
@@ -98,6 +99,24 @@ def rodin_update(prompt, task_uuid, token, settings):
     response = requests.post(f"{BASE_URL}/task/rodin_update", data={"uuid": task_uuid, "prompt": prompt, "settings": settings}, headers=headers)
     return response.json()
 
+def load_image(img_path):
+    image = Image.open(img_path)
+
+    # 按比例缩小图像到长度为1024
+    width, height = image.size
+    if width > height:
+        scale = 1024 / width
+    else:
+        scale = 1024 / height
+    new_width = int(width * scale)
+    new_height = int(height * scale)
+    resized_image = image.resize((new_width, new_height))
+
+    # 将 PIL.Image 对象转换为字节流
+    byte_io = BytesIO()
+    resized_image.save(byte_io, format='PNG')
+    image_bytes = byte_io.getvalue()
+    return image_bytes
 
 class Generator:
     def __init__(self, user_id, password) -> None:
@@ -109,22 +128,17 @@ class Generator:
             return prompt, cache_image_base64
         print("Preprocessing image...")
         
-        image_file = open(image_path, 'rb')
-        if image_file == None:
-            print("Invalid image file.")
-        try:
-            if prompt and task_uuid:
-                preprocess_response = rodin_preprocess_image(generate_prompt=False, image=image_file, name="images.jpeg", token=self.token)
-            else:
-                preprocess_response = rodin_preprocess_image(generate_prompt=True, image=image_file, name="images.jpeg", 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')                
-                processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
-        finally:
-            image_file.close()
+        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')                
+            processed_image = "data:image/png;base64," + preprocess_response.get('processed_image', None)
 
         return prompt, processed_image