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('/', methods=["GET", "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 изображения
    if request.method == "POST":
        return jsonify({"image_url": image_url})
    return render_template("app.py", user_input=[prompt_input, auth_input], response=image_output)

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