remember/save switch states

modules/ui.py

@@ -6,6 +6,8 @@ import cv2
 from cv2_enumerate_cameras import enumerate_cameras  # Add this import
 from PIL import Image, ImageOps
 import time
+import json
+
 import modules.globals
 import modules.metadata
 from modules.face_analyser import (
@@ -80,9 +82,49 @@ def init(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.CTk:
     return ROOT
 
 
+def save_switch_states():
+    switch_states = {
+        "keep_fps": modules.globals.keep_fps,
+        "keep_audio": modules.globals.keep_audio,
+        "keep_frames": modules.globals.keep_frames,
+        "many_faces": modules.globals.many_faces,
+        "map_faces": modules.globals.map_faces,
+        "color_correction": modules.globals.color_correction,
+        "nsfw_filter": modules.globals.nsfw_filter,
+        "live_mirror": modules.globals.live_mirror,
+        "live_resizable": modules.globals.live_resizable,
+        "fp_ui": modules.globals.fp_ui,
+        "show_fps": modules.globals.show_fps,
+    }
+    with open("switch_states.json", "w") as f:
+        json.dump(switch_states, f)
+
+
+def load_switch_states():
+    try:
+        with open("switch_states.json", "r") as f:
+            switch_states = json.load(f)
+        modules.globals.keep_fps = switch_states.get("keep_fps", True)
+        modules.globals.keep_audio = switch_states.get("keep_audio", True)
+        modules.globals.keep_frames = switch_states.get("keep_frames", False)
+        modules.globals.many_faces = switch_states.get("many_faces", False)
+        modules.globals.map_faces = switch_states.get("map_faces", False)
+        modules.globals.color_correction = switch_states.get("color_correction", False)
+        modules.globals.nsfw_filter = switch_states.get("nsfw_filter", False)
+        modules.globals.live_mirror = switch_states.get("live_mirror", False)
+        modules.globals.live_resizable = switch_states.get("live_resizable", False)
+        modules.globals.fp_ui = switch_states.get("fp_ui", {"face_enhancer": False})
+        modules.globals.show_fps = switch_states.get("show_fps", False)
+    except FileNotFoundError:
+        # If the file doesn't exist, use default values
+        pass
+
+
 def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.CTk:
     global source_label, target_label, status_label, show_fps_switch
 
+    load_switch_states()
+
     ctk.deactivate_automatic_dpi_awareness()
     ctk.set_appearance_mode("system")
     ctk.set_default_color_theme(resolve_relative_path("ui.json"))
@@ -125,8 +167,9 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
         text="Keep fps",
         variable=keep_fps_value,
         cursor="hand2",
-        command=lambda: setattr(
-            modules.globals, "keep_fps", not modules.globals.keep_fps
+        command=lambda: (
+            setattr(modules.globals, "keep_fps", keep_fps_value.get()),
+            save_switch_states(),
         ),
     )
     keep_fps_checkbox.place(relx=0.1, rely=0.6)
@@ -137,20 +180,23 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
         text="Keep frames",
         variable=keep_frames_value,
         cursor="hand2",
-        command=lambda: setattr(
-            modules.globals, "keep_frames", keep_frames_value.get()
+        command=lambda: (
+            setattr(modules.globals, "keep_frames", keep_frames_value.get()),
+            save_switch_states(),
         ),
     )
     keep_frames_switch.place(relx=0.1, rely=0.65)
 
-    # for FRAME PROCESSOR ENHANCER tumbler:
     enhancer_value = ctk.BooleanVar(value=modules.globals.fp_ui["face_enhancer"])
     enhancer_switch = ctk.CTkSwitch(
         root,
         text="Face Enhancer",
         variable=enhancer_value,
         cursor="hand2",
-        command=lambda: update_tumbler("face_enhancer", enhancer_value.get()),
+        command=lambda: (
+            update_tumbler("face_enhancer", enhancer_value.get()),
+            save_switch_states(),
+        ),
     )
     enhancer_switch.place(relx=0.1, rely=0.7)
 
@@ -160,7 +206,10 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
         text="Keep audio",
         variable=keep_audio_value,
         cursor="hand2",
-        command=lambda: setattr(modules.globals, "keep_audio", keep_audio_value.get()),
+        command=lambda: (
+            setattr(modules.globals, "keep_audio", keep_audio_value.get()),
+            save_switch_states(),
+        ),
     )
     keep_audio_switch.place(relx=0.6, rely=0.6)
 
@@ -170,19 +219,22 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
         text="Many faces",
         variable=many_faces_value,
         cursor="hand2",
-        command=lambda: setattr(modules.globals, "many_faces", many_faces_value.get()),
+        command=lambda: (
+            setattr(modules.globals, "many_faces", many_faces_value.get()),
+            save_switch_states(),
+        ),
     )
     many_faces_switch.place(relx=0.6, rely=0.65)
 
-    # Add color correction toggle button
     color_correction_value = ctk.BooleanVar(value=modules.globals.color_correction)
     color_correction_switch = ctk.CTkSwitch(
         root,
         text="Fix Blueish Cam\n(force cv2 to use RGB instead of BGR)",
         variable=color_correction_value,
         cursor="hand2",
-        command=lambda: setattr(
-            modules.globals, "color_correction", color_correction_value.get()
+        command=lambda: (
+            setattr(modules.globals, "color_correction", color_correction_value.get()),
+            save_switch_states(),
         ),
     )
     color_correction_switch.place(relx=0.6, rely=0.70)
@@ -197,18 +249,23 @@ def create_root(start: Callable[[], None], destroy: Callable[[], None]) -> ctk.C
         text="Map faces",
         variable=map_faces,
         cursor="hand2",
-        command=lambda: setattr(modules.globals, "map_faces", map_faces.get()),
+        command=lambda: (
+            setattr(modules.globals, "map_faces", map_faces.get()),
+            save_switch_states(),
+        ),
     )
     map_faces_switch.place(relx=0.1, rely=0.75)
 
-    # Add Show FPS switch
-    show_fps_value = ctk.BooleanVar(value=False)
+    show_fps_value = ctk.BooleanVar(value=modules.globals.show_fps)
     show_fps_switch = ctk.CTkSwitch(
         root,
         text="Show FPS",
         variable=show_fps_value,
         cursor="hand2",
-        command=lambda: setattr(modules.globals, "show_fps", show_fps_value.get()),
+        command=lambda: (
+            setattr(modules.globals, "show_fps", show_fps_value.get()),
+            save_switch_states(),
+        ),
     )
     show_fps_switch.place(relx=0.6, rely=0.75)
 
@@ -456,6 +513,7 @@ def update_pop_live_status(text: str) -> None:
 
 def update_tumbler(var: str, value: bool) -> None:
     modules.globals.fp_ui[var] = value
+    save_switch_states()
 
 
 def select_source_path() -> None: