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)