from flask import Flask, render_template, request, jsonify
from api.generate import create

app = Flask(__name__)

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

@app.route("/api/generate", methods=["POST"])
def generate():
    data = request.json
    prompt = data["prompt"]

    image_path = create(prompt)

    return jsonify({"image": image_path})

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)