Update app.py

app.py

@@ -1,8 +1,8 @@
 """
 app.py — HuggingFace Spaces entry point
-Gradio provides the public UI + keeps the Space alive.
-FastAPI handles the Slack webhook at /slack/events.
-Scheduler runs in the background thread.
+Pure FastAPI + uvicorn. No Gradio.
+Slack webhook at /slack/events.
+Admin endpoints at /admin/*.
 """
 
 import os
@@ -13,8 +13,7 @@ import threading
 import json
 from contextlib import asynccontextmanager
 
-import gradio as gr
-from fastapi import FastAPI, Request, Response
+from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 
@@ -22,16 +21,16 @@ import router
 import scheduler as sched
 import sheets
 
+
 # ─────────────────────────────────────────────
-# FASTAPI — Slack webhook
+# LIFESPAN
 # ─────────────────────────────────────────────
 
 @asynccontextmanager
-async def lifespan(app):
-    # Run setup in background — don't block startup or hammer Sheets API
+async def lifespan(app: FastAPI):
     import asyncio
     async def delayed_setup():
-        await asyncio.sleep(5)  # wait 5s for Space to fully start
+        await asyncio.sleep(5)
         try:
             sheets.setup_sheets()
         except Exception as e:
@@ -40,13 +39,29 @@ async def lifespan(app):
     sched.start_scheduler()
     yield
 
-app = FastAPI(lifespan=lifespan)
+
+# ─────────────────────────────────────────────
+# APP
+# ─────────────────────────────────────────────
+
+app = FastAPI(
+    title="Intern Management Agent",
+    description="Slack agent managing intern pipeline — Week 0 through Probation.",
+    version="1.0.0",
+    lifespan=lifespan,
+)
 
 SLACK_SIGNING_SECRET = os.environ.get("SLACK_SIGNING_SECRET", "")
 CHANNEL_ID = os.environ.get("SLACK_CHANNEL_ID", "")
 
 
+# ─────────────────────────────────────────────
+# SLACK SIGNATURE VERIFICATION
+# ─────────────────────────────────────────────
+
 def verify_slack_signature(body: bytes, timestamp: str, signature: str) -> bool:
+    if not timestamp or not signature:
+        return False
     if abs(time.time() - int(timestamp)) > 300:
         return False
     sig_basestring = f"v0:{timestamp}:{body.decode('utf-8')}"
@@ -58,31 +73,32 @@ def verify_slack_signature(body: bytes, timestamp: str, signature: str) -> bool:
     return hmac.compare_digest(computed, signature)
 
 
-@app.post("/slack/events")
+# ─────────────────────────────────────────────
+# SLACK EVENTS
+# ─────────────────────────────────────────────
+
+@app.post("/slack/events", summary="Slack event webhook")
 async def slack_events(request: Request):
     try:
         body = await request.body()
-
         if not body:
             return JSONResponse({"ok": True})
 
         payload = json.loads(body)
 
-        # One-time URL verification — must respond before anything else
-        # Slack sends this unsigned to verify the endpoint exists
+        # URL verification challenge — must respond before signature check
         if payload.get("type") == "url_verification":
             return JSONResponse({"challenge": payload["challenge"]})
 
-        # All other events — verify signature
+        # Verify all other requests
         timestamp = request.headers.get("X-Slack-Request-Timestamp", "")
         signature = request.headers.get("X-Slack-Signature", "")
-
         if not verify_slack_signature(body, timestamp, signature):
             return JSONResponse({"error": "Invalid signature"}, status_code=403)
 
         if payload.get("type") == "event_callback":
             event = payload.get("event", {})
-            if event.get("type") == "message" and event.get("channel") == CHANNEL_ID:
+            if event.get("type") == "message":
                 thread = threading.Thread(target=router.route_event, args=(event,))
                 thread.daemon = True
                 thread.start()
@@ -95,6 +111,10 @@ async def slack_events(request: Request):
         return JSONResponse({"error": str(e)}, status_code=500)
 
 
+# ─────────────────────────────────────────────
+# ADMIN — ADD CANDIDATE
+# ─────────────────────────────────────────────
+
 class CandidateIn(BaseModel):
     slack_user_id: str
     name: str
@@ -102,9 +122,12 @@ class CandidateIn(BaseModel):
     ft_slack_id: str
     hr_slack_id: str
     week0_start_date: str
-    stage: str = "week0"  # week0 | intern | probation_w1 | probation_w2
+    stage: str = "week0"         # week0 | intern | probation_w1 | probation_w2
+    department: str = "general"  # ai | hr | sales | product | general
+    channel_id: str = ""         # Slack channel ID where intern posts reports
+
 
-@app.post("/admin/add_candidate")
+@app.post("/admin/add_candidate", summary="Register a new intern")
 async def add_candidate(data: CandidateIn):
     try:
         candidate = sheets.create_candidate(
@@ -115,130 +138,43 @@ async def add_candidate(data: CandidateIn):
             hr_slack_id=data.hr_slack_id,
             week0_start_date=data.week0_start_date,
             stage=data.stage,
+            department=data.department,
+            channel_id=data.channel_id,
         )
         return JSONResponse({"ok": True, "candidate_id": candidate["candidate_id"]})
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
 
+# ─────────────────────────────────────────────
+# ADMIN — LOG COMMITMENT
+# ─────────────────────────────────────────────
+
 class CommitmentIn(BaseModel):
     candidate_id: str
     description: str
     due_date: str
     due_time: str
 
-@app.post("/admin/log_commitment")
-async def log_commitment(data: CommitmentIn):
-    commitment_id = sheets.log_commitment(
-        candidate_id=data.candidate_id,
-        description=data.description,
-        due_date=data.due_date,
-        due_time=data.due_time,
-    )
-    return JSONResponse({"ok": True, "commitment_id": commitment_id})
-
-
-@app.get("/health")
-async def health():
-    return JSONResponse({"status": "ok"})
-
-
-# ─────────────────────────────────────────────
-# GRADIO — Dashboard UI (keeps Space alive)
-# ─────────────────────────────────────────────
-
-def get_cohort_status():
-    """Pull live candidate data from Sheets for the dashboard."""
-    try:
-        candidates = sheets.get_active_candidates()
-        if not candidates:
-            return "No active candidates."
-        lines = ["**Active candidates**\n"]
-        for c in candidates:
-            lines.append(
-                f"- **{c['name']}** — Stage: `{c['stage']}` | "
-                f"Status: `{c['status']}` | Misses: `{c['miss_count']}`"
-            )
-        return "\n".join(lines)
-    except Exception as e:
-        return f"Error loading candidates: {e}"
-
-
-def get_recent_reports():
-    """Show last 10 report submissions across all candidates."""
-    try:
-        candidates = sheets.get_active_candidates()
-        all_reports = []
-        for c in candidates:
-            reports = sheets.get_reports_for_candidate(c["candidate_id"])
-            for r in reports:
-                r["candidate_name"] = c["name"]
-                all_reports.append(r)
-        all_reports.sort(key=lambda x: x.get("submitted_at", ""), reverse=True)
-        recent = all_reports[:10]
-        if not recent:
-            return "No reports yet."
-        lines = ["**Recent reports**\n"]
-        for r in recent:
-            flag = "✓" if r.get("format_valid") == "True" else "✗"
-            lines.append(
-                f"- {flag} **{r['candidate_name']}** — "
-                f"Day {r.get('day_number')} | "
-                f"Score: {r.get('quality_score', '-')} | "
-                f"{r.get('submitted_at', '')[:16]}"
-            )
-        return "\n".join(lines)
-    except Exception as e:
-        return f"Error loading reports: {e}"
 
-
-def get_pending_evals():
-    """Show evaluations awaiting HR confirmation."""
+@app.post("/admin/log_commitment", summary="Log a Week 2 delivery commitment")
+async def log_commitment(data: CommitmentIn):
     try:
-        candidates = sheets.get_active_candidates()
-        pending = []
-        for c in candidates:
-            rec = sheets.get_pending_recommendation(c["candidate_id"])
-            if rec:
-                pending.append(
-                    f"- **{c['name']}** — `{rec.get('eval_type')}` → "
-                    f"*{rec.get('recommendation')}* (awaiting HR)"
-                )
-        if not pending:
-            return "No pending evaluations."
-        return "**Pending HR confirmations**\n\n" + "\n".join(pending)
+        commitment_id = sheets.log_commitment(
+            candidate_id=data.candidate_id,
+            description=data.description,
+            due_date=data.due_date,
+            due_time=data.due_time,
+        )
+        return JSONResponse({"ok": True, "commitment_id": commitment_id})
     except Exception as e:
-        return f"Error loading evaluations: {e}"
-
-
-with gr.Blocks(title="ai_vidya Pipeline Agent") as gradio_app:
-    gr.Markdown("## ai_vidya Pipeline Agent")
-    gr.Markdown("Slack agent managing Week 0 → Probation for the AI Department intern pipeline.")
-
-    with gr.Row():
-        refresh_btn = gr.Button("Refresh", variant="secondary")
-
-    with gr.Row():
-        with gr.Column():
-            cohort_md = gr.Markdown(value=get_cohort_status())
-        with gr.Column():
-            pending_md = gr.Markdown(value=get_pending_evals())
-
-    with gr.Row():
-        reports_md = gr.Markdown(value=get_recent_reports())
-
-    def refresh():
-        return get_cohort_status(), get_pending_evals(), get_recent_reports()
-
-    refresh_btn.click(
-        fn=refresh,
-        outputs=[cohort_md, pending_md, reports_md],
-    )
+        return JSONResponse({"error": str(e)}, status_code=500)
 
 
 # ─────────────────────────────────────────────
-# MOUNT GRADIO ONTO FASTAPI + START SCHEDULER
+# HEALTH
 # ─────────────────────────────────────────────
 
-# Mount Gradio at root — FastAPI handles /slack/events and /admin/* separately
-app = gr.mount_gradio_app(app, gradio_app, path="/dashboard")
+@app.get("/health", summary="Health check")
+async def health():
+    return JSONResponse({"status": "ok"})