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)