Update app.py

app.py

@@ -4,101 +4,84 @@ import requests
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-# ============ CONFIG ============
 BACKEND_URL = "https://careflownexus-production.up.railway.app"
-AGENT_ROLE = os.getenv("AGENT_ROLE", "CLEANER").upper()  # MASTER/BED/CLEANER/NURSE
-POLL_INTERVAL = 3   # seconds
+AGENT_ROLE = os.getenv("AGENT_ROLE", "CLEANER").upper()
 
-app = FastAPI(title=f"{AGENT_ROLE} Agent Service")
+POLL_INTERVAL = 4
+
+HEADERS = {
+    "User-Agent": f"{AGENT_ROLE}-agent",
+    "Accept": "application/json"
+}
+
+app = FastAPI(title=f"{AGENT_ROLE} Agent")
 
-# CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
-    allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
+    allow_credentials=True,
 )
 
-# Status
+
 @app.get("/")
 def status():
-    return {
-        "agent_role": AGENT_ROLE,
-        "polling": True,
-        "backend": BACKEND_URL
-    }
+    return {"status": "running", "agent": AGENT_ROLE, "backend": BACKEND_URL}
 
-# ============ AGENT LOGIC ============
 
 def poll_tasks():
-    """Fetch pending tasks for this agent."""
     try:
         url = f"{BACKEND_URL}/agent/tasks?role={AGENT_ROLE}"
-        response = requests.get(url)
-        response.raise_for_status()
-        tasks = response.json()
-        return tasks if isinstance(tasks, list) else []
+        res = requests.get(url, headers=HEADERS)
+        res.raise_for_status()
+        data = res.json()
+        return data if isinstance(data, list) else []
     except Exception as e:
-        print("Error polling tasks:", e)
+        print("❌ Error polling tasks:", e)
         return []
 
+
 def complete_task(task):
-    """Complete the task using the proper backend endpoint."""
     try:
         task_id = task["id"]
         task_type = task["type"]
         payload = {}
 
-        # ==== BED agent must choose a bed ====
         if AGENT_ROLE == "BED" and task_type == "bed_assignment":
-            available = task.get("context", {}).get("available_beds", [])
-            if not available:
-                print("No available beds!")
+            beds = task.get("context", {}).get("available_beds", [])
+            if not beds:
+                print("No beds available")
                 return
-
-            chosen_bed = available[0]["id"]
-            payload = {"bed_id": chosen_bed}
-
-        # CLEANER (cleaning / post_discharge_cleaning)
-        if AGENT_ROLE == "CLEANER":
-            payload = {}
-
-        # NURSE (nurse_assignment or discharge_work)
-        if AGENT_ROLE == "NURSE":
-            payload = {}
+            payload = {"bed_id": beds[0]["id"]}
 
         url = f"{BACKEND_URL}/agent/tasks/{task_id}/complete"
-        response = requests.post(url, json=payload)
-        response.raise_for_status()
+        res = requests.post(url, json=payload, headers=HEADERS)
+        res.raise_for_status()
 
-        print(f"✔ Completed task {task_id}: {task_type}")
-        print(response.json())
+        print(f"✔ Completed {task_id}: {task_type}")
 
     except Exception as e:
-        print("Error completing task:", e)
-
+        print("❌ Error completing task:", e)
 
-# Background polling task
-def run_agent_loop():
-    print(f"Starting {AGENT_ROLE} agent loop...")
 
+def run_loop():
+    print(f"🔄 {AGENT_ROLE} agent started...")
     while True:
         tasks = poll_tasks()
 
         if tasks:
-            print(f"🔔 {len(tasks)} task(s) available")
-            for task in tasks:
-                complete_task(task)
+            print("🔔 Tasks:", tasks)
+            for t in tasks:
+                complete_task(t)
         else:
-            print("No tasks available.")
+            print("No tasks.")
 
         time.sleep(POLL_INTERVAL)
 
 
-# Startup event
 @app.on_event("startup")
-def start_background_task():
+def start_loop():
     import threading
-    thread = threading.Thread(target=run_agent_loop, daemon=True)
-    thread.start()
+    time.sleep(2)  # HF boot delay fix
+    threading.Thread(target=run_loop, daemon=True).start()