Update admin_routes.py

admin_routes.py

@@ -224,6 +224,46 @@ def add_activity_log(config: dict, username: str, action: str,
     if len(config["activity_log"]) > 1000:
         config["activity_log"] = config["activity_log"][-1000:]
 
+async def notify_spark(project: dict, version: dict, global_config: dict):
+    """Notify Spark about published version"""
+    import httpx
+    
+    spark_endpoint = global_config.get("spark_endpoint", "").rstrip("/")
+    if not spark_endpoint:
+        log("⚠️ Spark endpoint not configured")
+        return
+    
+    work_mode = global_config.get("work_mode", "hfcloud")
+    cloud_token = global_config.get("cloud_token", "")
+    
+    # Decrypt token if needed
+    if cloud_token and cloud_token.startswith("enc:"):
+        from encryption_utils import decrypt
+        cloud_token = decrypt(cloud_token)
+    
+    payload = {
+        "work_mode": work_mode,
+        "cloud_token": cloud_token,
+        "project_name": project["name"],
+        "project_version": version["id"],
+        "repo_id": version["llm"]["repo_id"],
+        "generation_config": version["llm"]["generation_config"],
+        "use_fine_tune": version["llm"]["use_fine_tune"],
+        "fine_tune_zip": version["llm"]["fine_tune_zip"] if version["llm"]["use_fine_tune"] else None
+    }
+    
+    log(f"🚀 Notifying Spark about {project['name']} v{version['id']}")
+    
+    async with httpx.AsyncClient(timeout=30) as client:
+        try:
+            response = await client.post(spark_endpoint + "/startup", json=payload)
+            response.raise_for_status()
+            result = response.json()
+            log(f"✅ Spark notification successful: {result.get('message', 'OK')}")
+        except Exception as e:
+            log(f"❌ Spark notification failed: {e}")
+            raise
+
 # ===================== Auth Endpoints =====================
 @router.post("/login", response_model=LoginResponse)
 async def login(request: LoginRequest):
@@ -725,6 +765,15 @@ async def publish_version(
     save_config(config)
     
     log(f"✅ Version {version_id} published for project '{project['name']}' by {username}")
+
+    # Notify Spark if project is enabled
+    if project.get("enabled", False):
+        try:
+            await notify_spark(project, version, config.get("config", {}))
+        except Exception as e:
+            log(f"⚠️ Failed to notify Spark: {e}")
+            # Don't fail the publish 
+
     return {"success": True}
 
 @router.delete("/projects/{project_id}/versions/{version_id}")