|
|
from flask import Flask, request, send_file |
|
|
import io |
|
|
from PIL import Image, ImageDraw |
|
|
from openai import OpenAI |
|
|
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() |
