Create utils/session.py

utils/session.py

@@ -1,7 +1,31 @@
-import streamlit as st
+import json
+import os
 
-def get_canvas_data():
-    return st.session_state.get("canvas_data", {})
+STATE_FILE = "data/state.json"
 
-def save_canvas_data(canvas_data):
-    st.session_state["canvas_data"] = canvas_data
+def load_state():
+    """Load the app state from the JSON file."""
+    if os.path.exists(STATE_FILE):
+        try:
+            with open(STATE_FILE, "r", encoding="utf-8") as f:
+                return json.load(f)
+        except json.JSONDecodeError:
+            return {}
+    return {}
+
+def save_state(state: dict):
+    """Save the app state to the JSON file."""
+    os.makedirs(os.path.dirname(STATE_FILE), exist_ok=True)
+    with open(STATE_FILE, "w", encoding="utf-8") as f:
+        json.dump(state, f, indent=4)
+
+def update_state(key, value):
+    """Update a specific key in the app state."""
+    state = load_state()
+    state[key] = value
+    save_state(state)
+
+def get_state_value(key, default=None):
+    """Get a specific key from the app state, or return default."""
+    state = load_state()
+    return state.get(key, default)