Ethanfel/Comfyui-JSON-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

seed

Browse Source
This commit is contained in:
ethanfel
2025-12-30 17:25:01 +01:00
committed by GitHub
parent 75362e1246
commit d9b16584dd
1 changed files with 35 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
stream_editor.py
View File

@@ -2,6 +2,7 @@ import streamlit as st
import json
import os
import time
import random
from pathlib import Path
# --- Configuration ---
@@ -14,6 +15,7 @@ st.set_page_config(layout="wide", page_title="AI Settings Manager")
DEFAULTS = {
"camera": "Camera stand still. Motion starts immediately.",
"flf": 0,
"seed": 0, # <--- NEW KEY
"frame_to_skip": 81,
"input_a_frames": "",
"input_b_frames": "",
@@ -22,7 +24,7 @@ DEFAULTS = {
"vace schedule": 1,
"video file path": "",
"reference image path": "",
"flf image path": "", # <--- NEW KEY
"flf image path": "",
"current_prompt": "",
"negative": "",
"lora 1 high": "", "lora 1 low": "",
@@ -91,7 +93,6 @@ def generate_templates(directory):
elif "global" in filename:
data.update({"video file path": "", "negative": GLOBAL_NEGATIVE})
elif "i2v" in filename:
# Updated I2V Template
data.update({"reference image path": "", "flf image path": ""})
save_json(path, data)
@@ -193,6 +194,7 @@ if selected_file_name:
st.session_state.last_mtime = mtime
st.session_state.loaded_file = str(file_path)
if 'append_prompt' in st.session_state: del st.session_state.append_prompt
if 'rand_seed' in st.session_state: del st.session_state.rand_seed
st.session_state.edit_history_idx = None
else:
data = st.session_state.data_cache
@@ -209,6 +211,24 @@ if selected_file_name:
new_prompt = st.text_area("Current Prompt", value=current_prompt_val, height=150)
new_negative = st.text_area("Negative Prompt", value=data.get("negative", ""), height=100)
# --- SEED SECTION ---
col_seed_val, col_seed_btn = st.columns([4, 1])
with col_seed_btn:
st.write("") # Spacer
st.write("")
if st.button("🎲 Randomize"):
st.session_state.rand_seed = random.randint(0, 999999999999)
st.rerun()
with col_seed_val:
# Use randomized value if triggered, otherwise data value
seed_val = st.session_state.get('rand_seed', int(data.get("seed", 0)))
# IMPORTANT: Sync session state back to None to allow manual edits later if needed,
# but usually keeping it in session state until save is fine.
new_seed = st.number_input("Seed", value=seed_val, step=1, min_value=0, format="%d")
# Update cache immediately so it persists if we change other inputs
data["seed"] = new_seed
st.subheader("LoRAs")
st.code("<lora::1.0>", language="text")
@@ -230,7 +250,6 @@ if selected_file_name:
elif "global" in fname:
fields += ["video file path"]
elif "i2v" in fname:
# I2V Specific Fields Update
fields += ["reference image path", "flf image path"]
for f in fields:
@@ -249,6 +268,7 @@ if selected_file_name:
if c_col1.button("Force Overwrite", type="primary"):
data["current_prompt"] = new_prompt
data["negative"] = new_negative
data["seed"] = new_seed
data.update(loras)
data.update(spec_fields)
st.session_state.last_mtime = save_json(file_path, data)
@@ -264,6 +284,7 @@ if selected_file_name:
if st.button("💾 Update File", use_container_width=True):
data["current_prompt"] = new_prompt
data["negative"] = new_negative
data["seed"] = new_seed
data.update(loras)
data.update(spec_fields)
st.session_state.last_mtime = save_json(file_path, data)
@@ -277,6 +298,7 @@ if selected_file_name:
entry = {
"prompt": new_prompt,
"negative": new_negative,
"seed": new_seed,
"note": archive_note if archive_note else f"Snapshot {len(data.get('prompt_history', [])) + 1}",
"loras": loras,
**spec_fields
@@ -294,7 +316,6 @@ if selected_file_name:
st.subheader("Media Preview")
preview_path = None
# Add flf image path to preview logic
for k in ["reference path", "video file path", "reference image path", "flf image path"]:
if spec_fields.get(k):
preview_path = spec_fields[k]
@@ -336,12 +357,18 @@ if selected_file_name:
with st.expander(f"📝 EDITING: {note_title}", expanded=True):
st.info("Editing History Entry (In-Place)")
edit_note = st.text_input("Note", value=h.get('note', ''), key=f"edit_note_{idx}")
# --- ADDED SEED EDITING HERE ---
ec_seed1, ec_seed2 = st.columns([1, 3])
edit_seed = ec_seed1.number_input("Seed", value=int(h.get('seed', 0)), step=1, key=f"edit_seed_{idx}")
edit_prompt = st.text_area("Prompt", value=h.get('prompt', ''), height=150, key=f"edit_prompt_{idx}")
edit_negative = st.text_area("Negative", value=h.get('negative', ''), height=80, key=f"edit_neg_{idx}")
ec1, ec2 = st.columns([1, 4])
if ec1.button("💾 Save", key=f"save_edit_{idx}", type="primary"):
h['note'] = edit_note
h['seed'] = edit_seed
h['prompt'] = edit_prompt
h['negative'] = edit_negative
st.session_state.last_mtime = save_json(file_path, data)
@@ -358,11 +385,11 @@ if selected_file_name:
col_h1, col_h2 = st.columns([3, 1])
with col_h1:
st.caption("📝 Prompt")
st.caption(f"📝 Prompt (Seed: {h.get('seed', 0)})")
st.text(h.get('prompt', ''))
st.caption("🧩 LoRAs & Files")
info_dict = {k:v for k,v in h.items() if k not in ['prompt', 'negative', 'note'] and v}
info_dict = {k:v for k,v in h.items() if k not in ['prompt', 'negative', 'note', 'seed'] and v}
if 'loras' in h and isinstance(h['loras'], dict):
info_dict.update({k:v for k,v in h['loras'].items() if v})
if 'loras' in info_dict: del info_dict['loras']
@@ -375,10 +402,11 @@ if selected_file_name:
else:
data["current_prompt"] = h.get("prompt", "")
data["negative"] = h.get("negative", "")
data["seed"] = int(h.get("seed", 0))
if "loras" in h and isinstance(h["loras"], dict):
data.update(h["loras"])
for k, v in h.items():
if k not in ["note", "prompt", "loras", "negative"]:
if k not in ["note", "prompt", "loras", "negative", "seed"]:
data[k] = v
st.session_state.last_mtime = save_json(file_path, data)