Update app.py

app.py

@@ -1,46 +1,34 @@
-# worker_app.py
-from flask import Flask, request, jsonify
-import ray
+from fastapi import FastAPI
+import os
 
-app = Flask(__name__)
+app = FastAPI()
 
-# ====== Yeh apna FIXED head node address hai ======
-HEAD_NODE_ADDRESS = "ray://192.168.1.100:10001"
-# ====================================================
+# Head node address
+RAY_HEAD_ADDRESS = "ray://<HEAD-NODE-IP>:10001"  # <-- Yahan apna address daal dena
 
-connected = False  # Worker connection status
+# Worker connect status
+connected = False
 
-@app.route('/worker', methods=['POST'])
-def connect_worker():
+@app.post("/worker")
+async def connect_worker():
     global connected
-    
-    if connected:
-        return jsonify({"message": "Already connected to Ray head node."}), 200
-
-    try:
-        ray.init(address=HEAD_NODE_ADDRESS)  # Use the fixed head node address
+    if not connected:
+        os.system(f"ray start --address='{RAY_HEAD_ADDRESS}'")
         connected = True
-        return jsonify({"message": f"Worker connected successfully to head node at {HEAD_NODE_ADDRESS}."}), 200
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        return {"message": "Worker connected to Ray Head Node"}
+    else:
+        return {"message": "Worker already connected"}
 
-@app.route('/noworker', methods=['POST'])
-def disconnect_worker():
+@app.post("/noworker")
+async def disconnect_worker():
     global connected
-    
-    if not connected:
-        return jsonify({"message": "Worker is already disconnected."}), 200
-
-    try:
-        ray.shutdown()
+    if connected:
+        os.system("ray stop")
         connected = False
-        return jsonify({"message": "Worker disconnected successfully."}), 200
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
-@app.route('/')
-def home():
-    return "Worker Flask App Running."
+        return {"message": "Worker disconnected from Ray Head Node"}
+    else:
+        return {"message": "No worker was connected"}
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=5000)
+@app.get("/")
+async def root():
+    return {"message": "Worker Node Ready"}