Enabling logger for capturing storage error

persistence/storage_service.py

@@ -2,15 +2,12 @@
 
 import json
 import os
-import logging
 import tempfile
 
 from huggingface_hub import HfApi, hf_hub_download
 
 from state import UserData
 
-logger = logging.getLogger(__name__)
-
 
 class StorageService:
     REPO_ID = "Group-1-5010/notebooklm-data"
@@ -21,7 +18,7 @@ class StorageService:
     def _get_token():
         token = os.environ.get("HF_TOKEN")
         if not token:
-            logger.warning("HF_TOKEN not found — persistence disabled")
+            raise RuntimeError("HF_TOKEN not found in environment. Add it as a Secret in your HF Space settings.")
         return token
 
     @staticmethod
@@ -29,28 +26,27 @@ class StorageService:
         """Create the dataset repo if it doesn't already exist (once per process)."""
         if StorageService._repo_ensured:
             return
-        try:
-            api.create_repo(
-                repo_id=StorageService.REPO_ID,
-                repo_type=StorageService.REPO_TYPE,
-                private=True,
-                exist_ok=True,
-            )
-            StorageService._repo_ensured = True
-            logger.info("Ensured dataset repo exists: %s", StorageService.REPO_ID)
-        except Exception as e:
-            logger.error("Failed to create dataset repo: %s", e)
+        api.create_repo(
+            repo_id=StorageService.REPO_ID,
+            repo_type=StorageService.REPO_TYPE,
+            private=True,
+            exist_ok=True,
+        )
+        StorageService._repo_ensured = True
+        print(f"[StorageService] Ensured dataset repo exists: {StorageService.REPO_ID}")
 
     @staticmethod
     def save_user_data(user_data: UserData) -> None:
-        """Serialize UserData to JSON and upload to HF Dataset repo."""
+        """Serialize UserData to JSON and upload to HF Dataset repo.
+
+        Raises on failure so the caller can show the error to the user.
+        """
         token = StorageService._get_token()
-        if not token:
-            return
 
-        try:
-            data_json = json.dumps(user_data.to_dict(), ensure_ascii=False, indent=2)
+        data_json = json.dumps(user_data.to_dict(), ensure_ascii=False, indent=2)
 
+        tmp_path = None
+        try:
             with tempfile.NamedTemporaryFile(
                 mode="w", suffix=".json", delete=False
             ) as tmp:
@@ -65,14 +61,13 @@ class StorageService:
                 repo_id=StorageService.REPO_ID,
                 repo_type=StorageService.REPO_TYPE,
             )
-            logger.info("Saved user data for '%s'", user_data.user_id)
-        except Exception as e:
-            logger.error("Failed to save user data: %s", e)
+            print(f"[StorageService] Saved user data for '{user_data.user_id}'")
         finally:
-            try:
-                os.unlink(tmp_path)
-            except Exception:
-                pass
+            if tmp_path:
+                try:
+                    os.unlink(tmp_path)
+                except Exception:
+                    pass
 
     @staticmethod
     def load_user_data(user_id: str, user_name: str) -> UserData | None:
@@ -80,8 +75,10 @@ class StorageService:
 
         Returns None if the file doesn't exist (new user).
         """
-        token = StorageService._get_token()
-        if not token:
+        try:
+            token = StorageService._get_token()
+        except RuntimeError:
+            print("[StorageService] HF_TOKEN not set — skipping load")
             return None
 
         try:
@@ -93,9 +90,9 @@ class StorageService:
             )
             with open(path, "r") as f:
                 data = json.load(f)
-            logger.info("Loaded user data for '%s'", user_id)
+            print(f"[StorageService] Loaded user data for '{user_id}'")
             return UserData.from_dict(data)
         except Exception as e:
             # EntryNotFoundError or network errors — treat as new user
-            logger.info("No existing data for '%s': %s", user_id, e)
+            print(f"[StorageService] No existing data for '{user_id}': {e}")
             return None