Update app.py

app.py

@@ -1,23 +1,39 @@
 from fastapi import FastAPI
 import os
+import subprocess
 
 app = FastAPI()
 
 # Head node address
-RAY_HEAD_ADDRESS = "ray://<HEAD-NODE-IP>:10001"  # <-- Yahan apna address daal dena
+RAY_HEAD_ADDRESS = "ray://<HEAD-NODE-IP>:10001"  # <-- Apna correct head node address yahan daalo
 
 # Worker connect status
 connected = False
 
+def get_ray_status():
+    """Check if ray is running or stopped."""
+    try:
+        output = subprocess.check_output(["ray", "status"], stderr=subprocess.STDOUT)
+        if b"cluster" in output or b"Connected" in output:
+            return "running"
+        else:
+            return "unknown"
+    except subprocess.CalledProcessError:
+        return "stopped"
+    except FileNotFoundError:
+        return "ray-not-installed"
+
 @app.post("/worker")
 async def connect_worker():
     global connected
     if not connected:
         os.system(f"ray start --address='{RAY_HEAD_ADDRESS}'")
         connected = True
-        return {"message": "Worker connected to Ray Head Node"}
-    else:
-        return {"message": "Worker already connected"}
+    return {
+        "message": "Worker connection attempt finished",
+        "connection": 1 if connected else 0,
+        "ray_status": get_ray_status()
+    }
 
 @app.post("/noworker")
 async def disconnect_worker():
@@ -25,10 +41,16 @@ async def disconnect_worker():
     if connected:
         os.system("ray stop")
         connected = False
-        return {"message": "Worker disconnected from Ray Head Node"}
-    else:
-        return {"message": "No worker was connected"}
+    return {
+        "message": "Worker disconnection attempt finished",
+        "connection": 1 if connected else 0,
+        "ray_status": get_ray_status()
+    }
 
 @app.get("/")
 async def root():
-    return {"message": "Worker Node Ready"}
+    return {
+        "message": "Worker Node Ready",
+        "connection": 1 if connected else 0,
+        "ray_status": get_ray_status()
+    }