selenium fetchpdf checkpoint

apps/SeleniumService/agent.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
-from selenium_worker import AutomationMassDHP
+from selenium_worker import AutomationMassHealth
 
 app = FastAPI()
 
@@ -12,15 +12,38 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-@app.post("/run")
-async def run_bot(request: Request):
+
+# Endpoint: Step 1 — Start the automation
+@app.post("/start-workflow")
+async def start_workflow(request: Request):
     data = await request.json()
     try:
-        bot = AutomationMassDHP(data)
-        result = bot.main_workflow("https://providers.massdhp.com/providers_login.asp")
+        bot = AutomationMassHealth(data)
+        result = bot.main_workflow_upto_step2("https://providers.massdhp.com/providers_login.asp")
         return result
     except Exception as e:
         return {"status": "error", "message": str(e)}
 
+# Endpoint: Step 2 — Extract the PDF content after manual submission
+@app.post("/fetch-pdf")
+async def fetch_pdf():
+    try:
+        bot = AutomationMassHealth().get_last_instance()
+        if not bot:
+            return {"status": "error", "message": "No running automation session"}
+        
+        pdf_data = bot.reach_to_pdf()
+        if pdf_data.get("status") != "success":
+            return {"status": "error", "message": pdf_data.get("message")}
+
+        return {
+            "status": "success",
+            "pdf_url": pdf_data["pdf_url"],
+            "pdf_base64": pdf_data["pdf_bytes"]
+        }
+    except Exception as e:
+        return {"status": "error", "message": str(e)}
+    
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=5002)