ai-frame-image-server/ai_frame_image_server.py

from flask import Flask
from libs.generic import load_config
import os

from routes import (
    auth_routes,
    gallery_routes,
    image_routes,
    index_routes,
    job_routes,
    create_routes,
    settings_routes
)

user_config = load_config()

app = Flask(__name__)
app.secret_key = os.environ.get("SECRET_KEY")

# Inject config into routes that need it
create_routes.init_app(user_config)
auth_routes.init_app(user_config)

# Register blueprints
app.register_blueprint(index_routes.bp)
app.register_blueprint(auth_routes.bp)
app.register_blueprint(gallery_routes.bp)
app.register_blueprint(image_routes.bp)
app.register_blueprint(job_routes.bp)
app.register_blueprint(create_routes.bp)
app.register_blueprint(settings_routes.bp)

# Optional: scheduler setup
from apscheduler.schedulers.background import BackgroundScheduler
import time
from libs.comfyui import create_image

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":
    if os.environ.get("WERKZEUG_RUN_MAIN") == "true":
        scheduler = BackgroundScheduler()
        h, m = user_config["frame"]["regen_time"].split(":")
        scheduler.add_job(scheduled_task, "cron", hour=h, minute=m, id="scheduled_task", max_instances=1, replace_existing=True)
        scheduler.start()

os.makedirs("./output", exist_ok=True)
app.run(host="0.0.0.0", port=user_config["frame"]["port"], debug=True)