26 lines
869 B
Python
26 lines
869 B
Python
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)
|