mirror of
https://github.com/karl0ss/ai_image_frame_server.git
synced 2025-04-28 11:13:41 +01:00
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
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
|