Dalle2Api / app.py
Rooni's picture
Create app.py
c67f33e
raw
history blame
1.65 kB
from flask import Flask, request, jsonify
import openai
# Инициализация Flask приложения
app = Flask(__name__)
# Загрузите ваш API-ключ OpenAI в безопасное место
# Например, из переменных окружения или файла конфигурации
openai.api_key = 'your-openai-api-key'
@app.route('/generate-image', methods=['POST'])
def generate_image():
# Проверка наличия авторизационного ключа в заголовках запроса
auth_key = request.headers.get('Authorization')
if auth_key != 'Your-Secret-Key':
return jsonify({"error": "Unauthorized"}), 401
# Получение описания изображения из тела запроса
description = request.json.get('description')
if not description:
return jsonify({"error": "No description provided"}), 400
# Вызов API DALL-E 2 для генерации изображения
# Здесь может понадобиться дополнительная логика для работы с API DALL-E 2
# Например, загрузка сгенерированного изображения на хранилище и получение URL
response = openai.Image.create(prompt=description, n=1)
image_url = response['data'][0]['url'] # Пример получения URL изображения
# Возвращение URL сгенерированного изображения
return jsonify({"image_url": image_url})
if __name__ == '__main__':
app.run(debug=True)