| from fastapi import APIRouter, HTTPException
|
| from pydantic import BaseModel
|
| from backend.state import session_store
|
| from backend.services.clusterService import get_clusters
|
|
|
| router = APIRouter()
|
|
|
| class SessionRequest(BaseModel):
|
| session_id: str
|
|
|
| @router.post("/clustering")
|
| async def clusters(request: SessionRequest):
|
| session_id = request.session_id
|
|
|
| if session_id not in session_store:
|
| raise HTTPException(status_code=400, detail="session code not found")
|
|
|
| df = session_store[session_id]["df"]
|
|
|
| try:
|
| return get_clusters(df)
|
| except ValueError as e:
|
| raise HTTPException(status_code=400, detail=str(e))
|
|
|
