From 8fbc9f402cd9109ee8e392ac93b98b90babfd0fc Mon Sep 17 00:00:00 2001 From: ethanfel Date: Fri, 2 Jan 2026 18:16:40 +0100 Subject: [PATCH] Update utils.py --- utils.py | 160 +++++++++++++++++++++++++++---------------------------- 1 file changed, 80 insertions(+), 80 deletions(-) diff --git a/utils.py b/utils.py index 1da0c6e..a370ac5 100644 --- a/utils.py +++ b/utils.py @@ -1,99 +1,99 @@ import json -import os +import time from pathlib import Path +import streamlit as st + +# Default structure for new files +DEFAULTS = { + "positive_prompt": "", + "negative_prompt": "", + "seed": -1, + "steps": 20, + "cfg": 7.0, + "sampler_name": "euler", + "scheduler": "normal", + "denoise": 1.0, + "model_name": "v1-5-pruned-emaonly.ckpt", + "vae_name": "vae-ft-mse-840000-ema-pruned.ckpt", + # I2V / VACE Specifics + "frame_to_skip": 81, + "vace schedule": 1, + "video file path": "", + "reference image path": "", + "flf": 0.0, + "camera": "static", + # LoRAs + "lora 1 high": "", "lora 1 low": "", + "lora 2 high": "", "lora 2 low": "", + "lora 3 high": "", "lora 3 low": "" +} CONFIG_FILE = Path(".editor_config.json") SNIPPETS_FILE = Path(".editor_snippets.json") -# --- Defaults --- -DEFAULTS = { - "camera": "Camera stand still. Motion starts immediately.", - "flf": 0, - "seed": 0, - "frame_to_skip": 81, - "input_a_frames": 0, - "input_b_frames": 0, - "reference path": "", - "reference switch": 1, - "vace schedule": 1, - "video file path": "", - "reference image path": "", - "flf image path": "", - - # --- PROMPTS --- - "general_prompt": "", - "general_negative": "Vivid tones, overexposed, static, blurry details, subtitles, style, artwork, painting, picture, still image, overall gray, worst quality, low quality, JPEG compression artifacts, ugly, deformed, extra fingers, poorly drawn hands, poorly drawn face, distorted, disfigured, malformed limbs, fused fingers, unmoving frame, cluttered background, three legs,", - "current_prompt": "", - "negative": "", - - # --- LORAS --- - "lora 1 high": "", "lora 1 low": "", - "lora 2 high": "", "lora 2 low": "", - "lora 3 high": "", "lora 3 low": "", - "prompt_history": [] -} - -GENERIC_TEMPLATES = ["prompt_i2v.json", "prompt_vace_extend.json", "batch_i2v.json", "batch_vace.json"] - -# --- I/O Functions --- -def get_file_mtime(path): - if path.exists(): return os.path.getmtime(path) - return 0 - -def load_json(path): - if not path.exists(): return DEFAULTS.copy(), 0 - try: - with open(path, 'r') as f: - data = json.load(f) - return data, get_file_mtime(path) - except: - return DEFAULTS.copy(), 0 - -def save_json(path, data): - if path.exists(): - try: - with open(path, 'r') as f: - existing = json.load(f) - if isinstance(existing, dict) and isinstance(data, dict): - existing.update(data) - data = existing - except: pass - - with open(path, 'w') as f: - json.dump(data, f, indent=4) - return get_file_mtime(path) - def load_config(): + """Loads the main editor configuration (Favorites, Last Dir, Servers).""" if CONFIG_FILE.exists(): - try: - with open(CONFIG_FILE, 'r') as f: return json.load(f) - except: pass - return {"last_dir": str(Path.cwd()), "favorites": []} + try: + with open(CONFIG_FILE, 'r') as f: + return json.load(f) + except: + pass + return {"favorites": [], "last_dir": str(Path.cwd()), "comfy_instances": []} -def save_config(current_dir, favorites): +def save_config(current_dir, favorites, extra_data=None): + """Saves configuration to disk. Supports extra keys like 'comfy_instances'.""" + data = { + "last_dir": str(current_dir), + "favorites": favorites + } + # Merge existing config to prevent data loss + existing = load_config() + data.update(existing) + + # Update with new 'last_dir' and 'favorites' + data["last_dir"] = str(current_dir) + data["favorites"] = favorites + + # Update with any extra data passed (like server lists) + if extra_data: + data.update(extra_data) + with open(CONFIG_FILE, 'w') as f: - json.dump({"last_dir": str(current_dir), "favorites": favorites}, f, indent=4) + json.dump(data, f, indent=4) def load_snippets(): if SNIPPETS_FILE.exists(): - try: - with open(SNIPPETS_FILE, 'r') as f: return json.load(f) - except: pass + try: + with open(SNIPPETS_FILE, 'r') as f: + return json.load(f) + except: + pass return {} def save_snippets(snippets): with open(SNIPPETS_FILE, 'w') as f: json.dump(snippets, f, indent=4) -def generate_templates(directory): - for filename in GENERIC_TEMPLATES: - path = directory / filename - if "batch" in filename: - data = {"batch_data": []} - else: - data = DEFAULTS.copy() - if "vace" in filename: - data.update({"frame_to_skip": 81, "vace schedule": 1, "video file path": ""}) - elif "i2v" in filename: - data.update({"reference image path": "", "flf image path": ""}) - save_json(path, data) +def load_json(path): + path = Path(path) + if not path.exists(): + return DEFAULTS.copy(), 0 + try: + with open(path, 'r') as f: + data = json.load(f) + return data, path.stat().st_mtime + except Exception as e: + st.error(f"Error loading JSON: {e}") + return DEFAULTS.copy(), 0 + +def save_json(path, data): + with open(path, 'w') as f: + json.dump(data, f, indent=4) + +def generate_templates(current_dir): + """Creates dummy template files if folder is empty.""" + save_json(current_dir / "template_i2v.json", DEFAULTS) + + batch_data = {"batch_data": [DEFAULTS.copy(), DEFAULTS.copy()]} + save_json(current_dir / "template_batch.json", batch_data)