|
from flask import Flask, request, jsonify |
|
import openai |
|
|
|
|
|
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'] |
|
|
|
return jsonify({"image_url": image_url}) |
|
|
|
if __name__ == '__main__': |
|
app.run(debug=True) |