support auto image regeneration at specific time

ai_frame_image_server.py

@@ -1,22 +1,38 @@
-from flask import Flask, render_template, send_from_directory, redirect, url_for, request, jsonify
+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('/images/<filename>')
+@app.route("/")
+def index():
+    return render_template(
+        "index.html",
+        image="./image.png",
+        reload_interval=user_config["frame"]["reload_interval"],
+    )
+
+
+@app.route("/images/<filename>")
 def images(filename):
     return send_from_directory(image_folder, filename)
 
-@app.route('/create', methods=["GET", "POST"])
+
+@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
@@ -27,7 +43,21 @@ def create():
     return redirect(url_for("index"))
 
 
-if __name__ == '__main__':
-    os.makedirs(image_folder, exist_ok=True)  # Ensure the folder exists
-    app.run(host="0.0.0.0", port=user_config["frame"]["port"], debug=True)
+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

user_config.cfg.sample

@@ -1,5 +1,7 @@
 [frame]
 reload_interval = 30000
+auto_regen = True
+regen_time = 03:00
 port = 5000
 
 [comfyui]