Update spark_startup.py

spark_startup.py

@@ -1,7 +1,7 @@
 """
 Flare – Spark startup notifier
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-Projeler açılırken Spark’a /startup çağrısı yapar.
+Projeler açılırken Spark'a /startup çağrısı yapar.
 """
 
 from __future__ import annotations
@@ -13,15 +13,32 @@ import os
 
 cfg = ConfigProvider.get()
 
+def _get_spark_token() -> Optional[str]:
+    """Get Spark token based on work_mode"""
+    if cfg.global_config.is_cloud_mode():
+        # Cloud mode - use HuggingFace Secrets
+        token = os.getenv("SPARK_TOKEN")
+        if not token:
+            log("❌ SPARK_TOKEN not found in HuggingFace Secrets!")
+        return token
+    else:
+        # On-premise mode - use .env file
+        from dotenv import load_dotenv
+        load_dotenv()
+        return os.getenv("SPARK_TOKEN")
 
 def _select_live_version(p: ProjectConfig) -> VersionConfig | None:
     """Yayınlanmış en güncel versiyonu getir."""
     published = [v for v in p.versions if v.published]
     return max(published, key=lambda v: v.id) if published else None
 
-
 def notify_startup():
     spark_base = str(cfg.global_config.spark_endpoint).rstrip("/")
+    spark_token = _get_spark_token()
+    
+    if not spark_token:
+        log("⚠️ SPARK_TOKEN not configured. Skipping Spark startup notification.")
+        return
 
     for p in cfg.projects:
         if not p.enabled:
@@ -45,12 +62,6 @@ def notify_startup():
             "fine_tune_zip": v.llm.fine_tune_zip,
         }
 
-        # Hugging Face token'ı ortam değişkeninden al
-        spark_token = os.getenv("SPARK_TOKEN")
-        if not spark_token:
-            log("❌ SPARK_TOKEN environment variable is missing.")
-            return
-    
         headers = {
             "Authorization": f"Bearer {spark_token}",
             "Content-Type": "application/json"
@@ -66,6 +77,5 @@ def notify_startup():
         except Exception as e:
             log(f"⚠️ Spark startup failed: {e}")
 
-
 def run_in_thread():
     threading.Thread(target=notify_startup, daemon=True).start()