Update app.py

app.py

@@ -1003,6 +1003,10 @@ def handsome_embeddings():
     except requests.exceptions.RequestException as e:
         return jsonify({"error": str(e)}), 500
 
+import base64
+import io
+from PIL import Image
+
 @app.route('/handsome/v1/images/generations', methods=['POST'])
 def handsome_images_generations():
     if not check_authorization(request):
@@ -1037,7 +1041,9 @@ def handsome_images_generations():
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
     }
-
+    
+    response_data = {}
+    
     if "stable-diffusion" in model_name:
         # Map OpenAI-style parameters to SiliconFlow's parameters
         siliconflow_data = {
@@ -1090,16 +1096,37 @@ def handsome_images_generations():
             
             try:
                 images = response_json.get("images", [])
-                if isinstance(images, list):
-                    openai_images = [{"url": image} for image in images]
-                else:
-                    openai_images = []
+                openai_images = []
+                for image_url in images:
+                    if data.get("response_format") == "b64_json":
+                         try:
+                            image_data = requests.get(image_url, stream=True).raw
+                            image = Image.open(image_data)
+                            buffered = io.BytesIO()
+                            image.save(buffered, format="PNG")
+                            img_str = base64.b64encode(buffered.getvalue()).decode()
+                            openai_images.append({"b64_json": img_str})
+                         except Exception as e:
+                             logging.error(f"图片转base64失败: {e}")
+                             openai_images.append({"url": image_url})
+                    else:
+                        openai_images.append({"url": image_url})
+
+                response_data = {
+                    "created": int(time.time()),
+                    "data": openai_images
+                }
+
             except (KeyError, ValueError, IndexError) as e:
                 logging.error(
                     f"解析响应 JSON 失败: {e}, "
                     f"完整内容: {response_json}"
                 )
-                openai_images = []
+                response_data = {
+                    "created": int(time.time()),
+                    "data": []
+                }
+
 
             logging.info(
                 f"使用的key: {api_key}, "
@@ -1111,10 +1138,7 @@ def handsome_images_generations():
                 request_timestamps.append(time.time())
                 token_counts.append(0)  # Image generation doesn't use tokens
 
-            return jsonify({
-                "created": int(time.time()),
-                "data": openai_images
-            })
+            return jsonify(response_data)
 
         except requests.exceptions.RequestException as e:
             logging.error(f"请求转发异常: {e}")