mirror of
https://github.com/karl0ss/ai_image_frame_server.git
synced 2025-07-05 05:59:08 +01:00
live reload of topics and models
This commit is contained in:
parent
9d80741da1
commit
d791c8b4ed
1
.vscode/launch.json
vendored
1
.vscode/launch.json
vendored
@ -11,6 +11,7 @@
|
||||
"program": "${file}",
|
||||
"console": "integratedTerminal",
|
||||
"justMyCode": false,
|
||||
"env": {"SECRET_KEY":"dkdkdk"}
|
||||
// "args": [
|
||||
// "--num_inference_steps",
|
||||
// "6",
|
||||
|
@ -60,7 +60,7 @@ def login():
|
||||
if request.method == 'POST':
|
||||
if request.form['password'] == user_config["frame"]["password_for_auth"]:
|
||||
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:
|
||||
return redirect(url_for('login'))
|
||||
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'):
|
||||
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":
|
||||
|
@ -100,16 +100,14 @@ def get_current_version():
|
||||
return None
|
||||
|
||||
def load_models_from_config():
|
||||
flux_models = user_config["comfyui:flux"]["models"].split(",")
|
||||
sdxl_models = user_config["comfyui"]["models"].split(",")
|
||||
flux_models = load_config()["comfyui:flux"]["models"].split(",")
|
||||
sdxl_models = load_config()["comfyui"]["models"].split(",")
|
||||
all_models = flux_models + sdxl_models
|
||||
return all_models
|
||||
|
||||
def load_topics_from_config():
|
||||
topics = user_config["comfyui"]["topics"].split(", ")
|
||||
topics = load_config()["comfyui"]["topics"].split(", ")
|
||||
return topics
|
||||
|
||||
|
||||
|
||||
user_config = load_config()
|
||||
output_folder = user_config["comfyui"]["output_dir"]
|
Loading…
x
Reference in New Issue
Block a user