Enabling logger with flush

app.py

@@ -362,16 +362,16 @@ with gr.Blocks(css=CUSTOM_CSS, theme=dark_theme, title="NotebookLM") as demo:
 
     # ── Auth: on page load ───────────────────────────────────────────────────
     def on_app_load(profile: gr.OAuthProfile | None):
-        print(f"[on_app_load] profile={profile}")
+        print(f"[on_app_load] profile={profile}", flush=True)
         if profile is None:
             return None, gr.update(visible=True), gr.update(visible=False)
-        print(f"[on_app_load] Loading data for user: {profile.username}")
+        print(f"[on_app_load] Loading data for user: {profile.username}", flush=True)
         state = StorageService.load_user_data(profile.username, profile.name)
         if state is None:
-            print(f"[on_app_load] No saved data found — creating default")
+            print(f"[on_app_load] No saved data found — creating default", flush=True)
             state = create_default_user_data(profile.username, profile.name)
         else:
-            print(f"[on_app_load] Loaded {len(state.notebooks)} notebook(s)")
+            print(f"[on_app_load] Loaded {len(state.notebooks)} notebook(s)", flush=True)
         return state, gr.update(visible=False), gr.update(visible=True)
 
     demo.load(
@@ -444,14 +444,14 @@ with gr.Blocks(css=CUSTOM_CSS, theme=dark_theme, title="NotebookLM") as demo:
 
     # ── Sidebar: Save ────────────────────────────────────────────────────────
     def handle_save(state):
-        print(f"[handle_save] Called, state={state is not None}")
+        print(f"[handle_save] Called, state={state is not None}", flush=True)
         if not state:
             return '<p style="color:#ef4444; font-size:0.8rem;">No data to save.</p>'
         try:
             StorageService.save_user_data(state)
             return '<p style="color:#22c55e; font-size:0.8rem;">Saved successfully!</p>'
         except Exception as e:
-            print(f"[handle_save] Error: {e}")
+            print(f"[handle_save] Error: {e}", flush=True)
             return f'<p style="color:#ef4444; font-size:0.8rem;">Save failed: {e}</p>'
 
     save_btn.click(

persistence/storage_service.py

@@ -33,7 +33,7 @@ class StorageService:
             exist_ok=True,
         )
         StorageService._repo_ensured = True
-        print(f"[StorageService] Ensured dataset repo exists: {StorageService.REPO_ID}")
+        print(f"[StorageService] Ensured dataset repo exists: {StorageService.REPO_ID}", flush=True)
 
     @staticmethod
     def save_user_data(user_data: UserData) -> None:
@@ -61,7 +61,7 @@ class StorageService:
                 repo_id=StorageService.REPO_ID,
                 repo_type=StorageService.REPO_TYPE,
             )
-            print(f"[StorageService] Saved user data for '{user_data.user_id}'")
+            print(f"[StorageService] Saved user data for '{user_data.user_id}'", flush=True)
         finally:
             if tmp_path:
                 try:
@@ -78,7 +78,7 @@ class StorageService:
         try:
             token = StorageService._get_token()
         except RuntimeError:
-            print("[StorageService] HF_TOKEN not set — skipping load")
+            print("[StorageService] HF_TOKEN not set — skipping load", flush=True)
             return None
 
         try:
@@ -90,9 +90,9 @@ class StorageService:
             )
             with open(path, "r") as f:
                 data = json.load(f)
-            print(f"[StorageService] Loaded user data for '{user_id}'")
+            print(f"[StorageService] Loaded user data for '{user_id}'", flush=True)
             return UserData.from_dict(data)
         except Exception as e:
             # EntryNotFoundError or network errors — treat as new user
-            print(f"[StorageService] No existing data for '{user_id}': {e}")
+            print(f"[StorageService] No existing data for '{user_id}': {e}", flush=True)
             return None