File size: 657 Bytes
1067825 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | 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))
|