Enhance progress/result endpoints with query param support and code cleanup

api.py

@@ -22,16 +22,6 @@ logger = logging.getLogger("API")
 
 app = FastAPI(title="Antigravity SEO Scaler API")
 
-app.add_middleware(
-    CORS,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-from fastapi.staticfiles import StaticFiles
-
-# ...
-
 app.add_middleware(
     CORS,
     allow_origins=["*"],
@@ -221,26 +211,26 @@ def start_scan(payload: StartReq, bg: BackgroundTasks):
     return {"job_id": job_id}
 
 @app.get("/api/progress/{job_id}")
-def scan_progress(job_id: str):
+@app.get("/api/progress")
+def scan_progress(job_id: Optional[str] = None):
+    if not job_id:
+         return JSONResponse(status_code=400, content={"error": "job_id is required"})
+    
     job = JOBS.get(job_id)
     if not job:
-        return JSONResponse(status_code=404, content={"status": "not_found", "error": "Job ID not found"})
+        return JSONResponse(status_code=404, content={"status": "not_found", "error": f"Job ID {job_id} not found"})
     
     # Calculate Live Stats
     current_time = time.time()
     elapsed = int(current_time - job.get("start_time", current_time))
     
     # Simple ETA Calculation
-    # If we have scanned X pages in Y seconds, and need to scan Z total...
-    # ETA = (target_pages - scanned_pages) * (elapsed / scanned_pages)
     eta_seconds = None
     pages_scanned = job.get("pages_scanned", 0)
     percent = job.get("percent", 0)
     
     if job["status"] == "running" and pages_scanned > 0 and percent > 0:
         avg_per_page = elapsed / pages_scanned
-        # Assume limit is roughly the target (or use percent to back-calculate)
-        # Using percent is safer if limit wasn't reached
         estimated_total = (pages_scanned / percent) * 100
         remaining = estimated_total - pages_scanned
         eta_seconds = int(remaining * avg_per_page)
@@ -257,10 +247,14 @@ def scan_progress(job_id: str):
     }
 
 @app.get("/api/result/{job_id}")
-def scan_result(job_id: str):
+@app.get("/api/result")
+def scan_result(job_id: Optional[str] = None):
+    if not job_id:
+         return JSONResponse(status_code=400, content={"error": "job_id is required"})
+         
     job = JOBS.get(job_id)
     if not job:
-        return JSONResponse(status_code=404, content={"status": "not_found", "error": "Job ID not found"})
+        return JSONResponse(status_code=404, content={"status": "not_found", "error": f"Job ID {job_id} not found"})
     
     if job["status"] != "done":
         return {"status": job["status"], "message": "Result not ready yet"}