Update user_management.py

user_management.py

@@ -320,6 +320,76 @@ class HFUserManager:
         
         return user.get("password_hash") == password_hash
 
+    # ============================================
+    # USER STORAGE FILE HELPERS
+    # ============================================
+    def _path_in_repo_for_user(self, user_id: str, filename: str) -> str:
+        return f"users/{user_id}/{filename}"
+
+    def save_user_json(self, user_id: str, filename: str, data: dict, commit_message: str = None) -> bool:
+        """
+        Save a JSON file into the user's storage folder in the HF repo.
+        """
+        if not self.enabled:
+            print("⚠️ HF User Manager disabled, cannot save user file")
+            return False
+
+        if commit_message is None:
+            commit_message = f"Update {user_id}/{filename}"
+
+        try:
+            # Create temp file
+            import tempfile
+            with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as tmp:
+                json.dump(data, tmp, indent=2, ensure_ascii=False)
+                tmp_path = tmp.name
+
+            path_in_repo = self._path_in_repo_for_user(user_id, filename)
+
+            try:
+                self.api.upload_file(
+                    path_or_fileobj=tmp_path,
+                    path_in_repo=path_in_repo,
+                    repo_id=self.hf_repo,
+                    token=self.hf_token,
+                    repo_type='model',
+                    commit_message=commit_message
+                )
+                print(f"✅ Saved {path_in_repo} to HF repo")
+                return True
+            finally:
+                if os.path.exists(tmp_path):
+                    os.remove(tmp_path)
+
+        except Exception as e:
+            print(f"❌ Failed to save user file to HF: {e}")
+            return False
+
+    def load_user_json(self, user_id: str, filename: str) -> Optional[dict]:
+        """
+        Load a JSON file from the user's storage folder in the HF repo.
+        Returns the parsed JSON dict or None if not found / error.
+        """
+        if not self.enabled:
+            return None
+
+        path_in_repo = self._path_in_repo_for_user(user_id, filename)
+        try:
+            downloaded_path = hf_hub_download(
+                repo_id=self.hf_repo,
+                filename=path_in_repo,
+                token=self.hf_token,
+                repo_type='model',
+                local_dir_use_symlinks=False
+            )
+
+            with open(downloaded_path, 'r', encoding='utf-8') as f:
+                return json.load(f)
+
+        except Exception as e:
+            # Not found or error
+            return None
+
 
 # ============================================
 # CONVENIENCE FUNCTIONS