eligibility check - checkpoint1

apps/SeleniumService/agent.py

@@ -1,9 +1,20 @@
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
-from selenium_worker import AutomationMassHealth
+import asyncio
+from selenium_claimSubmitWorker import AutomationMassHealth
+from selenium_eligibilityCheckWorker import AutomationMassHealthEligibilityCheck
+
 
 app = FastAPI()
+# Allow 1 selenium session at a time
+semaphore = asyncio.Semaphore(1)
+
+# Manual counters to track active & queued jobs
+active_jobs = 0
+waiting_jobs = 0
+lock = asyncio.Lock()  # To safely update counters
+
 
 app.add_middleware(
     CORSMiddleware,
@@ -12,21 +23,71 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-# Endpoint: Step 1 — Start the automation
-@app.post("/start-workflow")
+# Endpoint: Step 1 — Start the automation of submitting Claim.
+@app.post("/claimsubmit")
 async def start_workflow(request: Request):
+    global active_jobs, waiting_jobs
     data = await request.json()
-    try:
-        bot = AutomationMassHealth(data)
-        result = bot.main_workflow("https://providers.massdhp.com/providers_login.asp")
 
-        if result.get("status") != "success":
-            return {"status": "error", "message": result.get("message")}
-        
-        return result
-    except Exception as e:
-        return {"status": "error", "message": str(e)}
+    async with lock:
+        waiting_jobs += 1
+
+    async with semaphore: 
+        async with lock:
+            waiting_jobs -= 1
+            active_jobs += 1
+
+        try:
+            bot = AutomationMassHealth(data)
+            result = bot.main_workflow("https://providers.massdhp.com/providers_login.asp")
+
+            if result.get("status") != "success":
+                return {"status": "error", "message": result.get("message")}
+            
+            return result
+        except Exception as e:
+            return {"status": "error", "message": str(e)}
+        finally:
+            async with lock:
+                active_jobs -= 1
+    
+# Endpoint: Step 2 — Start the automation of cheking eligibility
+@app.post("/eligibility-check")
+async def start_workflow(request: Request):
+    global active_jobs, waiting_jobs
+    data = await request.json()
+
+    async with lock:
+        waiting_jobs += 1
+
+    async with semaphore:
+        async with lock:
+            waiting_jobs -= 1
+            active_jobs += 1
+        try:
+            print(data)
+            bot = AutomationMassHealthEligibilityCheck(data)
+            result = bot.main_workflow("https://providers.massdhp.com/providers_login.asp")
+
+            if result.get("status") != "success":
+                return {"status": "error", "message": result.get("message")}
+            
+            return result
+        except Exception as e:
+            return {"status": "error", "message": str(e)}
+        finally:
+            async with lock:
+                active_jobs -= 1
+
+# ✅ Status Endpoint
+@app.get("/status")
+async def get_status():
+    async with lock:
+        return {
+            "active_jobs": active_jobs,
+            "queued_jobs": waiting_jobs,
+            "status": "busy" if active_jobs > 0 or waiting_jobs > 0 else "idle"
+        }
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=5002)