ai-frame-image-server/ai_frame_image_server.py

from flask import (
    Flask,
    render_template,
    send_from_directory,
    redirect,
    url_for,
    request,
    jsonify,
)
import os
import time
from apscheduler.schedulers.background import BackgroundScheduler
from lib import create_image, load_config

user_config = load_config()
app = Flask(__name__)

image_folder = "./output"


@app.route("/")
def index():
    return render_template(
        "index.html",
        image="./image.png",
        reload_interval=user_config["frame"]["reload_interval"],
    )
    

@app.route("/gallery")
def gallery():
    images = [f for f in os.listdir(image_folder) if f.lower().endswith(('png', 'jpg', 'jpeg', 'gif'))]
    return render_template("gallery.html", images=images)


@app.route("/images/<filename>")
def images(filename):
    return send_from_directory(image_folder, filename)


@app.route("/create", methods=["GET", "POST"])
def create():
    """Endpoint to create a new image. Supports optional prompt via POST."""
    prompt = request.form.get("prompt") if request.method == "POST" else None
    create_image(prompt)  # Pass prompt to create_image()

    if request.method == "POST":
        return jsonify({"message": "Image created", "prompt": prompt}), 200
    return redirect(url_for("index"))


def scheduled_task():
    print(f"Executing scheduled task at {time.strftime('%Y-%m-%d %H:%M:%S')}")
    create_image(None)


if user_config["frame"]["auto_regen"] == "True":
    scheduler = BackgroundScheduler()

    regen_time = user_config["frame"]["regen_time"].split(":")
    scheduler.add_job(scheduled_task, "cron", hour=regen_time[0], minute=regen_time[1])
    scheduler.start()

if __name__ == "__main__":
    os.makedirs(image_folder, exist_ok=True)  # Ensure the folder exists
    try:
        app.run(host="0.0.0.0", port=user_config["frame"]["port"], debug=True)
    except KeyboardInterrupt:
        scheduler.shutdown()  # Ensure graceful shutdown of scheduler