Karl/ai-frame-image-server
1
0
Fork 0
mirror of https://github.com/karl0ss/ai_image_frame_server.git synced 2025-07-06 06:26:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

live reload of topics and models

Browse Source
This commit is contained in:
Karl 2025-06-20 10:21:58 +01:00
parent 9d80741da1
commit d791c8b4ed
3 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.vscode/launch.json vendored
View File

@ -11,6 +11,7 @@
"program": "${file}", "program": "${file}",
"console": "integratedTerminal", "console": "integratedTerminal",
"justMyCode": false, "justMyCode": false,
"env": {"SECRET_KEY":"dkdkdk"}
// "args": [ // "args": [
// "--num_inference_steps", // "--num_inference_steps",
// "6", // "6",

6
ai_frame_image_server.py
View File

@ -60,7 +60,7 @@ def login():
if request.method == 'POST': if request.method == 'POST':
if request.form['password'] == user_config["frame"]["password_for_auth"]: if request.form['password'] == user_config["frame"]["password_for_auth"]:
session['authenticated'] = True session['authenticated'] = True
return render_template("create_image.html", models=load_models_from_config()) return render_template("create_image.html", models=load_models_from_config(), topics=load_topics_from_config())
else: else:
return redirect(url_for('login')) return redirect(url_for('login'))
return render_template('login.html') return render_template('login.html')
@ -159,10 +159,8 @@ def create_image_endpoint() -> str:
""" """
if user_config["frame"]["create_requires_auth"] == "True" and not session.get('authenticated'): if user_config["frame"]["create_requires_auth"] == "True" and not session.get('authenticated'):
return redirect(url_for("login")) return redirect(url_for("login"))
models = load_models_from_config()
topics = load_topics_from_config()
return render_template("create_image.html", models=models, topics=topics) return render_template("create_image.html", models=load_models_from_config(), topics=load_topics_from_config())
if user_config["frame"]["auto_regen"] == "True": if user_config["frame"]["auto_regen"] == "True":

8
libs/generic.py
View File

@ -100,16 +100,14 @@ def get_current_version():
return None return None
def load_models_from_config(): def load_models_from_config():
flux_models = user_config["comfyui:flux"]["models"].split(",") flux_models = load_config()["comfyui:flux"]["models"].split(",")
sdxl_models = user_config["comfyui"]["models"].split(",") sdxl_models = load_config()["comfyui"]["models"].split(",")
all_models = flux_models + sdxl_models all_models = flux_models + sdxl_models
return all_models return all_models
def load_topics_from_config(): def load_topics_from_config():
topics = user_config["comfyui"]["topics"].split(", ") topics = load_config()["comfyui"]["topics"].split(", ")
return topics return topics
user_config = load_config() user_config = load_config()
output_folder = user_config["comfyui"]["output_dir"] output_folder = user_config["comfyui"]["output_dir"]