Updated generation pages.
This commit is contained in:
25
routes/quick.py
Normal file
25
routes/quick.py
Normal file
@@ -0,0 +1,25 @@
|
||||
import logging
|
||||
from flask import render_template
|
||||
from models import Preset
|
||||
from services.file_io import get_available_checkpoints
|
||||
from services.comfyui import get_loaded_checkpoint
|
||||
from services.workflow import _get_default_checkpoint
|
||||
|
||||
logger = logging.getLogger('gaze')
|
||||
|
||||
|
||||
def register_routes(app):
|
||||
|
||||
@app.route('/quick')
|
||||
def quick_generator():
|
||||
presets = Preset.query.order_by(Preset.name).all()
|
||||
checkpoints = get_available_checkpoints()
|
||||
|
||||
# Default to whatever is currently loaded in ComfyUI, then settings default
|
||||
selected_ckpt = get_loaded_checkpoint()
|
||||
if not selected_ckpt:
|
||||
default_path, _ = _get_default_checkpoint()
|
||||
selected_ckpt = default_path
|
||||
|
||||
return render_template('quick.html', presets=presets,
|
||||
checkpoints=checkpoints, selected_ckpt=selected_ckpt)
|
||||
Reference in New Issue
Block a user