Update generate_image.py

generate_image.py

@@ -1,23 +1,28 @@
+# generate_image.py
+
 import openai
-import os
-from io import BytesIO
 from PIL import Image
+from io import BytesIO
+import os
+import base64
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def generate_ghibli_avatar(prompt, reference_image):
-    # 将 PIL Image 转成字节流文件
-    img_byte_arr = BytesIO()
-    reference_image.save(img_byte_arr, format='PNG')
-    img_byte_arr.seek(0)
+    # 将 PIL Image 转换为文件-like 对象
+    buffered = BytesIO()
+    reference_image.save(buffered, format="PNG")
+    buffered.seek(0)
 
+    # 使用 openai v0.27.8 的 images.create_edit 接口
     response = openai.Image.create_edit(
-        image=img_byte_arr,
-        mask=None,  # 如果你没有 mask，设为 None
+        image=buffered,
+        mask=None,
         prompt=prompt,
         n=1,
         size="512x512",
-        response_format="b64_json"
+        response_format="b64_json",
     )
-    image_data = response['data'][0]['b64_json']
+
+    image_data = response["data"][0]["b64_json"]
     return Image.open(BytesIO(base64.b64decode(image_data)))