Update app.py

app.py

@@ -327,16 +327,16 @@ async def pnl_route(file: UploadFile = File(...), use_rlhf: bool = Query(False))
 
 
 @router.post("/bs")
-async def bs_route(file: UploadFile = File(...), use_rlhf: bool = Query(False)):
+async def bs_route(file: UploadFile = File(...), use_rlhf: bool = Query(False), user_api_key: Optional[str] = Form(None)):
     file_path = f"data/input/{file.filename}"
     os.makedirs("data/input", exist_ok=True)
     with open(file_path, "wb") as buffer:
         shutil.copyfileobj(file.file, buffer)
     
     if use_rlhf:
-        result = run_rlhf_workflow(file_path, "bs")
+        result = run_rlhf_workflow(file_path, "bs", user_api_key=user_api_key)
     else:
-        result = run_workflow(file_path, "bs")
+        result = run_workflow(file_path, "bs", user_api_key=user_api_key)
     
     if result["status"] == "success":
         output_file = result["result"].get("output_path")