Update config_provider.py

config_provider.py

@@ -20,6 +20,14 @@ class GlobalConfig(BaseModel):
 
     def get_plain_token(self) -> Optional[str]:
         return decrypt(self.cloud_token) if self.cloud_token else None
+    
+    def is_cloud_mode(self) -> bool:
+        """Check if running in cloud mode (hfcloud or cloud)"""
+        return self.work_mode in ("hfcloud", "cloud")
+    
+    def is_on_premise(self) -> bool:
+        """Check if running in on-premise mode"""
+        return self.work_mode == "on-premise"
 
 # ---------------- Global -----------------
 class UserConfig(BaseModel):
@@ -166,19 +174,38 @@ class ConfigProvider:
         if cls._instance is None:
             cls._instance = cls._load()
             cls._instance.build_index()
+            cls._check_environment_setup()
         return cls._instance
 
-    # -------- Internal helpers ------------
     @classmethod
-    def _load(cls) -> ServiceConfig:
-        log(f"📥 Loading service config from {cls._CONFIG_PATH.name} …")
-        raw = cls._CONFIG_PATH.read_text(encoding="utf-8")
-        # Use commentjson instead of custom parser
-        try:
-            data = commentjson.loads(raw)
-            cfg = ServiceConfig.model_validate(data)
-            log("✅ Service config loaded.")
-            return cfg
-        except (commentjson.JSONLibraryException, ValidationError) as exc:
-            log(f"❌ Config validation error: {exc}")
-            raise
+    def _check_environment_setup(cls):
+        """Check if environment is properly configured based on work_mode"""
+        config = cls._instance.global_config
+        
+        if config.is_cloud_mode():
+            # Cloud mode - check for HuggingFace Secrets
+            missing_secrets = []
+            
+            if not os.getenv("JWT_SECRET"):
+                missing_secrets.append("JWT_SECRET")
+            if not os.getenv("FLARE_TOKEN_KEY"):
+                missing_secrets.append("FLARE_TOKEN_KEY")
+            if not os.getenv("SPARK_TOKEN"):
+                missing_secrets.append("SPARK_TOKEN")
+            
+            if missing_secrets:
+                log(f"⚠️ Running in {config.work_mode} mode. Missing secrets: {', '.join(missing_secrets)}")
+                log("📌 Add these in HuggingFace Space Settings → Repository secrets")
+            else:
+                log(f"✅ Running in {config.work_mode} mode with all required secrets")
+                
+        elif config.is_on_premise():
+            # On-premise mode - check for .env file
+            env_path = Path(".env")
+            if env_path.exists():
+                from dotenv import load_dotenv
+                load_dotenv()
+                log("✅ Running in on-premise mode with .env file")
+            else:
+                log("⚠️ Running in on-premise mode but .env file not found")
+                log("📌 Copy .env.example to .env and configure it")