from flask import Flask, render_template, request, send_file
from diffusers import DiffusionPipeline
from PIL import Image
import requests
from io import BytesIO

app = Flask(__name__)

pipeline = DiffusionPipeline.from_pretrained("dataautogpt3/OpenDalleV1.1")

@app.route("/")
def index():
    return render_template('index.html')

@app.route("/generate", methods=["POST"])
def generate():
    prompt = request.form['prompt']
    results = []
    for _ in range(4):
        response = requests.post(pipeline, json={"prompt": prompt})
        image_bytes = BytesIO(response.content)
        img = Image.open(image_bytes)
        results.append(img)

    return render_template('results.html', results=results)

if __name__ == "__main__":
    app.run(debug=True, host='0.0.0.0')