Update app.py

app.py

@@ -102,6 +102,7 @@ class CandidateIn(BaseModel):
     ft_slack_id: str
     hr_slack_id: str
     week0_start_date: str
+    stage: str = "week0"  # week0 | intern | probation_w1 | probation_w2
 
 @app.post("/admin/add_candidate")
 async def add_candidate(data: CandidateIn):
@@ -113,6 +114,7 @@ async def add_candidate(data: CandidateIn):
             ft_slack_id=data.ft_slack_id,
             hr_slack_id=data.hr_slack_id,
             week0_start_date=data.week0_start_date,
+            stage=data.stage,
         )
         return JSONResponse({"ok": True, "candidate_id": candidate["candidate_id"]})
     except Exception as e: