From 8064c90be9533f47ed79be7de11a1a93b1003dc7 Mon Sep 17 00:00:00 2001 From: Karl Hudgell Date: Sat, 29 Mar 2025 12:24:46 +0000 Subject: [PATCH] user config controls reload interval --- ai_frame_image_server.py | 8 +++++--- lib.py | 22 +++++++++++++--------- templates/index.html | 2 +- user_config.cfg.sample | 4 ++++ 4 files changed, 23 insertions(+), 13 deletions(-) diff --git a/ai_frame_image_server.py b/ai_frame_image_server.py index cac3535..7499931 100644 --- a/ai_frame_image_server.py +++ b/ai_frame_image_server.py @@ -1,6 +1,8 @@ from flask import Flask, render_template, send_from_directory, redirect, url_for import os -from lib import create_image +from lib import create_image, load_config + +user_config = load_config() app = Flask(__name__) @@ -9,7 +11,7 @@ image_folder = "./output" @app.route('/') def index(): # latest_image = get_latest_image() - return render_template("index.html", image="./image.png") + return render_template("index.html", image="./image.png", reload_interval=user_config["frame"]["reload_interval"]) @app.route('/images/') def images(filename): @@ -23,5 +25,5 @@ def create(): if __name__ == '__main__': os.makedirs(image_folder, exist_ok=True) # Ensure the folder exists - app.run(host="0.0.0.0", port=5000, debug=True) + app.run(host="0.0.0.0", port=user_config["frame"]["port"], debug=True) diff --git a/lib.py b/lib.py index 86bc982..eb9f9f8 100644 --- a/lib.py +++ b/lib.py @@ -8,15 +8,15 @@ import os from comfy_api_simplified import ComfyApiWrapper, ComfyWorkflowWrapper -user_config = configparser.ConfigParser() -try: - user_config.read("./user_config.cfg") - output_folder = user_config["comfyui"]["output_dir"] - logging.debug("Configuration loaded successfully.") -except KeyError as e: - logging.error(f"Missing configuration key: {e}") - sys.exit(1) - +def load_config(): + user_config = configparser.ConfigParser() + try: + user_config.read("./user_config.cfg") + logging.debug("Configuration loaded successfully.") + return user_config + except KeyError as e: + logging.error(f"Missing configuration key: {e}") + sys.exit(1) def rename_image(): """Rename 'image.png' to a timestamped filename if it exists in the output folder.""" @@ -84,3 +84,7 @@ def create_image(): prompt = send_prompt_to_openwebui(user_config["comfyui"]["prompt"]) print(f"Generated prompt: {prompt}") generate_image("image", prompt) + + +user_config = load_config() +output_folder = user_config["comfyui"]["output_dir"] diff --git a/templates/index.html b/templates/index.html index f51b744..d1279ac 100644 --- a/templates/index.html +++ b/templates/index.html @@ -26,7 +26,7 @@ diff --git a/user_config.cfg.sample b/user_config.cfg.sample index f9dc393..d57dd72 100644 --- a/user_config.cfg.sample +++ b/user_config.cfg.sample @@ -1,3 +1,7 @@ +[frame] +reload_interval = 30000 +port = 5000 + [comfyui] comfyui_url = http://comfyui model = zavychromaxl_v100.safetensors