ai-frame-image-server/ai_frame_image_server.py
2025-03-29 12:24:46 +00:00

30 lines
825 B
Python

from flask import Flask, render_template, send_from_directory, redirect, url_for
import os
from lib import create_image, load_config
user_config = load_config()
app = Flask(__name__)
image_folder = "./output"
@app.route('/')
def index():
# latest_image = get_latest_image()
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')
def create():
"""Endpoint to create a new image."""
create_image()
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)