Update app.py
Browse files
app.py
CHANGED
@@ -4,29 +4,20 @@ import openai
|
|
4 |
# Инициализация Flask приложения
|
5 |
app = Flask(__name__)
|
6 |
|
7 |
-
# Загрузите ваш API-ключ OpenAI в безопасное место
|
8 |
-
# Например, из переменных окружения или файла конфигурации
|
9 |
-
openai.api_key = 'your-openai-api-key'
|
10 |
-
|
11 |
@app.route('/generate-image', methods=['POST'])
|
12 |
def generate_image():
|
13 |
# Проверка наличия авторизационного ключа в заголовках запроса
|
14 |
auth_key = request.headers.get('Authorization')
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
# Получение описания изображения из тела запроса
|
19 |
description = request.json.get('description')
|
20 |
if not description:
|
21 |
return jsonify({"error": "No description provided"}), 400
|
22 |
|
23 |
-
|
24 |
-
# Здесь может понадобиться дополнительная логика для работы с API DALL-E 2
|
25 |
-
# Например, загрузка сгенерированного изображения на хранилище и получение URL
|
26 |
-
response = openai.Image.create(prompt=description, n=1)
|
27 |
image_url = response['data'][0]['url'] # Пример получения URL изображения
|
28 |
|
29 |
-
# Возвращение URL сгенерированного изображения
|
30 |
return jsonify({"image_url": image_url})
|
31 |
|
32 |
if __name__ == '__main__':
|
|
|
4 |
# Инициализация Flask приложения
|
5 |
app = Flask(__name__)
|
6 |
|
|
|
|
|
|
|
|
|
7 |
@app.route('/generate-image', methods=['POST'])
|
8 |
def generate_image():
|
9 |
# Проверка наличия авторизационного ключа в заголовках запроса
|
10 |
auth_key = request.headers.get('Authorization')
|
11 |
+
openai.api_key = auth_key
|
12 |
+
|
|
|
13 |
# Получение описания изображения из тела запроса
|
14 |
description = request.json.get('description')
|
15 |
if not description:
|
16 |
return jsonify({"error": "No description provided"}), 400
|
17 |
|
18 |
+
response = openai.Image.create(prompt=description, n=1, model="dall-e-3", size="1024x1024", quality="hd")
|
|
|
|
|
|
|
19 |
image_url = response['data'][0]['url'] # Пример получения URL изображения
|
20 |
|
|
|
21 |
return jsonify({"image_url": image_url})
|
22 |
|
23 |
if __name__ == '__main__':
|