app.py

from flask import Flask, request
from openai import OpenAI
import os
client = OpenAI(api_key=os.getenv("API_KEY"))

app = Flask(__name__)

@app.route('/generate-image', methods=['POST'])
def generate_image():
    data = request.json
    prompt = data['prompt']

    response = client.images.generate(
    model="dall-e-3",
    prompt=prompt,
    size="1024x1024",
    quality="hd",
    n=1,
    )
    
    image_url = response.data[0].url
    print(image_url)
    return image_url

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