Update app.py

app.py

@@ -9,10 +9,110 @@ from utils import (
 )
 from tab_single import render_single_editor
 from tab_batch import render_batch_processor
-from tab_timeline import render_timeline_tab  # <--- IMPORT NEW TAB
+from tab_timeline import render_timeline_tab  # <--- NEW IMPORT
 
-# ... (Keep all setup code: set_page_config, session state init, sidebar) ...
+# --- Page Setup ---
-# ... [Use the app.py code from previous response, just change the Main App Logic section below] ...
+st.set_page_config(layout="wide", page_title="AI Settings Manager")
+
+# --- Init Session State ---
+if 'config' not in st.session_state:
+    st.session_state.config = load_config()
+    st.session_state.current_dir = Path(st.session_state.config.get("last_dir", Path.cwd()))
+if 'snippets' not in st.session_state: st.session_state.snippets = load_snippets()
+if 'loaded_file' not in st.session_state: st.session_state.loaded_file = None
+if 'last_mtime' not in st.session_state: st.session_state.last_mtime = 0
+if 'edit_history_idx' not in st.session_state: st.session_state.edit_history_idx = None
+if 'single_editor_cache' not in st.session_state: st.session_state.single_editor_cache = DEFAULTS.copy()
+
+# Version Token for forcing UI refreshes (Critical for Branching/Restoring)
+if 'ui_reset_token' not in st.session_state: st.session_state.ui_reset_token = 0
+
+# --- Sidebar ---
+with st.sidebar:
+    st.header("📂 Navigator")
+    
+    # Path Navigator
+    new_path = st.text_input("Current Path", value=str(st.session_state.current_dir))
+    if new_path != str(st.session_state.current_dir):
+        p = Path(new_path)
+        if p.exists() and p.is_dir():
+            st.session_state.current_dir = p
+            st.session_state.config['last_dir'] = str(p)
+            save_config(st.session_state.current_dir, st.session_state.config['favorites'])
+            st.rerun()
+
+    # Favorites
+    if st.button("📌 Pin Current Folder"):
+        if str(st.session_state.current_dir) not in st.session_state.config['favorites']:
+            st.session_state.config['favorites'].append(str(st.session_state.current_dir))
+            save_config(st.session_state.current_dir, st.session_state.config['favorites'])
+            st.rerun()
+
+    fav_selection = st.radio("Jump to:", ["Select..."] + st.session_state.config['favorites'], index=0, label_visibility="collapsed")
+    if fav_selection != "Select..." and fav_selection != str(st.session_state.current_dir):
+        st.session_state.current_dir = Path(fav_selection)
+        st.rerun()
+
+    st.markdown("---")
+    
+    # Snippets
+    st.subheader("🧩 Snippet Library")
+    with st.expander("Add New Snippet"):
+        snip_name = st.text_input("Name", placeholder="e.g. Cinematic")
+        snip_content = st.text_area("Content", placeholder="4k, high quality...")
+        if st.button("Save Snippet"):
+            if snip_name and snip_content:
+                st.session_state.snippets[snip_name] = snip_content
+                save_snippets(st.session_state.snippets)
+                st.success(f"Saved '{snip_name}'")
+                st.rerun()
+
+    if st.session_state.snippets:
+        st.caption("Click to Append to Prompt:")
+        for name, content in st.session_state.snippets.items():
+            col_s1, col_s2 = st.columns([4, 1])
+            if col_s1.button(f"➕ {name}", use_container_width=True):
+                st.session_state.append_prompt = content
+                st.rerun()
+            if col_s2.button("🗑️", key=f"del_snip_{name}"):
+                del st.session_state.snippets[name]
+                save_snippets(st.session_state.snippets)
+                st.rerun()
+
+    st.markdown("---")
+    
+    # File List
+    json_files = sorted(list(st.session_state.current_dir.glob("*.json")))
+    json_files = [f for f in json_files if f.name != ".editor_config.json" and f.name != ".editor_snippets.json"]
+
+    if not json_files:
+        if st.button("Generate Templates"):
+            generate_templates(st.session_state.current_dir)
+            st.rerun()
+    
+    # Create New
+    with st.expander("Create New JSON"):
+        new_filename = st.text_input("Filename", placeholder="my_prompt_vace")
+        is_batch = st.checkbox("Is Batch File?")
+        if st.button("Create"):
+            if not new_filename.endswith(".json"): new_filename += ".json"
+            path = st.session_state.current_dir / new_filename
+            if is_batch:
+                data = {"batch_data": []}
+            else:
+                data = DEFAULTS.copy()
+                if "vace" in new_filename: data.update({"frame_to_skip": 81, "vace schedule": 1, "video file path": ""})
+                elif "i2v" in new_filename: data.update({"reference image path": "", "flf image path": ""})
+            save_json(path, data)
+            st.rerun()
+
+    # File Selector
+    if 'file_selector' not in st.session_state:
+        st.session_state.file_selector = json_files[0].name if json_files else None
+    if st.session_state.file_selector not in [f.name for f in json_files] and json_files:
+        st.session_state.file_selector = json_files[0].name
+    
+    selected_file_name = st.radio("Select File", [f.name for f in json_files], key="file_selector")
 
 # --- Main App Logic ---
 if selected_file_name:
@@ -32,8 +132,7 @@ if selected_file_name:
 
     st.title(f"Editing: {selected_file_name}")
 
-    # --- TABS CONFIGURATION ---
+    # --- TABS ---
-    # We add the 3rd Tab here
     tab_single, tab_batch, tab_timeline = st.tabs(["📝 Single Editor", "🚀 Batch Processor", "🕒 Timeline"])
     
     with tab_single:
@@ -43,7 +142,4 @@ if selected_file_name:
         render_batch_processor(data, file_path, json_files, st.session_state.current_dir, selected_file_name)
         
     with tab_timeline:
-        # Check if batch file, as requested "only for batch"
-        # But honestly, it's useful for single too. I'll enable it for both, 
-        # but if you STRICTLY want batch only, uncomment the if statement inside render_timeline_tab
         render_timeline_tab(data, file_path)