Update app.py

app.py

@@ -5,8 +5,12 @@ import json
 import time
 import os
 import re
+import base64
+from io import BytesIO
+from PIL import Image
 
 _COOKIES = os.environ.get("COOKIES", "")
+_COOKIES = "session_token=955b54566c702e02d011fd41b6d913c3b136dc6c3d67ba6d6ef89b6fac4086eb; cf_clearance=e0TWZDlzBpV28hqeENTPBMjX9rgxAtZBsrK3_vpisDA-1740845308-1.2.1.1-R4D99qqFa0Y2um46_2sWjttiyVEPDPM2MrpoL2xwzdwEsDFVCXsynY3AyOrO4bmtCIyebHsS_05JXsv_YqAdFpSdneBNdgnW0zkgGsP1HZiBv.Gdb_vE8LsGpNz19FynpCdOfcOXB0J3lnx4epF.BWaykyT140O38VPYKGUhr0H9BnOVg7KAI727twEqhCC8DMMgn81qV8Vp1YaDmDX_EBzA76eKU5CwjxxVDZrxzaGZDbkCyho.OsUeqIBFWELI.5G0K42MNrMVEy0S0pEFfWIzbkVXAFj_mvFUgyITnC9aaMna9HI0VnBT1dJ4G__gLrNS.CpGneS5QicBAvn58su7raHd9i4_uOuBWgkldZpCD2Dp_b2v0EYtOKFH1Lji1oNL6IgyLzE_Y4ghOgcELWFcLERQKgutGfCA6PaYze8"
 
 API_KEY = os.getenv("API_KEY", "linux.do")
 
@@ -191,6 +195,23 @@ def chat_completions():
                             # 检查响应状态码是否为200
                             if _data[0]["status"] == "completed":
                                 print("图片生成完成:", job_id)
+                                # 分离出 header 和 base64 编码的数据
+                                _header, _encoded = _data[0]["result"].split(',', 1)
+                                # 解码 base64 得到二进制数据
+                                webp_data = base64.b64decode(_encoded)
+
+                                # 使用 BytesIO 将二进制数据转换为 PIL Image 对象
+                                input_buffer = BytesIO(webp_data)
+                                _image = Image.open(input_buffer)
+
+                                # 创建一个 BytesIO 缓存用于存储 PNG 数据
+                                output_buffer = BytesIO()
+                                _image.save(output_buffer, format="PNG")
+                                png_data = output_buffer.getvalue()
+
+                                # 将 PNG 二进制数据转换为 base64 编码
+                                encoded_png = base64.b64encode(png_data).decode('utf-8')
+                                data_png = "data:image/png;base64," + encoded_png
                                 return Response(
                                     json.dumps({
                                         "object": "chat.completion",
@@ -201,7 +222,7 @@ def chat_completions():
                                                 "index": 0,
                                                 "message": {
                                                     "role": "user",
-                                                    "content": _data[0]["result"]
+                                                    "content": data_png
                                                 },
                                                 "finish_reason": "stop"
                                             }