K-means-Clustering-Visualization/session.py at main · Ad2m1109/K-means-Clustering-Visualization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

from fastapi import Request
from kmeans import KMeansVisualizer
import uuid

sessions = {}

def get_session(request: Request):
    session_id = request.cookies.get("session_id")
    if session_id and session_id in sessions:
        return sessions[session_id]

    new_session_id = str(uuid.uuid4())
    kmeans = KMeansVisualizer()
    sessions[new_session_id] = kmeans
    return kmeans

def get_session_id(request: Request):
    session_id = request.cookies.get("session_id")
    if session_id and session_id in sessions:
        return session_id

    new_session_id = str(uuid.uuid4())
    kmeans = KMeansVisualizer()
    sessions[new_session_id] = kmeans
    return new_session_id