main.py

import os
from flask import Flask, request, jsonify
import openai

# Инициализация Flask приложения
app = Flask(__name__)

# Инициализация клиента OpenAI
openai.api_key = os.getenv("OPENAI_API_KEY")

@app.route('/generate-image', methods=['POST'])
def generate_image_api():
    # Проверка авторизации
    auth_key = request.headers.get('Authorization')
    if auth_key != f"Bearer {os.getenv('API_KEY')}":
        return jsonify({"error": "Unauthorized"}), 401

    # Получение описания изображения из запроса
    description = request.json.get('description')
    if not description:
        return jsonify({"error": "No description provided"}), 400

    # Генерация изображения с помощью DALL-E
    response = openai.Image.create(prompt=description, n=1)
    image_url = response['data'][0]['url']  # Получение URL изображения

    return jsonify({"image_url": image_url})

if __name__ == '__main__':
    app.run(debug=True)