From 38a10e8f2d854ada745703d2969e3cb4a411da32 Mon Sep 17 00:00:00 2001
From: Clivia <132346501+Yanyutin753@users.noreply.github.com>
Date: Wed, 7 Aug 2024 11:35:11 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=91=80=20=E6=94=AF=E6=8C=81=E4=BC=A0?=
 =?UTF-8?q?=E5=85=A5size=EF=BC=8C=E6=8E=A7=E5=88=B6=E7=94=BB=E5=9B=BE?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E8=BE=93=E5=87=BA=E5=9B=BE=E7=89=87=E7=9A=84?=
 =?UTF-8?q?=E5=A4=A7=E5=B0=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/main.py b/main.py
index 040018c..1414ecf 100644
--- a/main.py
+++ b/main.py
@@ -1977,12 +1977,14 @@ def images_generations():
     if model not in accessible_model_list and not 'gpt-4-gizmo-' in model:
         return jsonify({"error": "model is not accessible"}), 401
 
-    prompt = data.get('prompt', '')
-
-    prompt = DALLE_PROMPT_PREFIX + prompt
-
     # 获取请求中的response_format参数，默认为"url"
     response_format = data.get('response_format', 'url')
+    # 获取请求中的size参数，默认为"1024x1024"
+    response_size = data.get('size', '1024x1024')
+
+    prompt = data.get('prompt', '')
+
+    prompt = DALLE_PROMPT_PREFIX + '\nprompt:' + prompt + '\nsize:' + response_size
 
     # stream = data.get('stream', False)