Dalle2Api / app.py
Rooni's picture
Update app.py
d57b22c
raw
history blame
961 Bytes
from flask import Flask, request, jsonify
import openai
# Инициализация Flask приложения
app = Flask(__name__)
@app.route('/generate-image', methods=['POST'])
def generate_image():
# Проверка наличия авторизационного ключа в заголовках запроса
auth_key = request.headers.get('Authorization')
openai.api_key = auth_key
# Получение описания изображения из тела запроса
description = request.json.get('description')
if not description:
return jsonify({"error": "No description provided"}), 400
response = openai.Image.create(prompt=description, n=1, model="dall-e-3", size="1024x1024", quality="hd")
image_url = response['data'][0]['url'] # Пример получения URL изображения
return jsonify({"image_url": image_url})
if __name__ == '__main__':
app.run(debug=True)