File size: 959 Bytes
c849742
f5128e8
a228558
e98570f
 
f5128e8
 
 
 
e98570f
 
f5128e8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
c849742
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f5128e8
1c5f20a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
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)