|
|
from flask import Flask, request, render_template_string |
|
|
from openai import OpenAI |
|
|
import os |
|
|
import logging |
|
|
|
|
|
client = OpenAI(api_key=os.getenv("API_KEY")) |
|
|
|
|
|
app = Flask(__name__) |
|
|
|
|
|
logging.basicConfig(level=logging.DEBUG) |
|
|
|
|
|
@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 |
|
|
|
|
|
html_template = """ |
|
|
<!doctype html> |
|
|
<html lang="en"> |
|
|
<head> |
|
|
<meta charset="utf-8"> |
|
|
<title>Test Button</title> |
|
|
</head> |
|
|
<body> |
|
|
<h1>Test Button for Flask App</h1> |
|
|
<button onclick="alert('Button clicked!')">Click me</button> |
|
|
</body> |
|
|
</html> |
|
|
""" |
|
|
|
|
|
@app.route('/') |
|
|
def index(): |
|
|
return render_template_string(html_template) |
|
|
|
|
|
|
|
|
if __name__ == '__main__': |
|
|
app.run(host='0.0.0.0', port=5000) |
