if cookie_jar:
    session.cookies.update(cookie_jar)

# Get session cookies and CSRF token
csrf_token, csrf_cookie, session_cookie = login()

# Save cookies to the jar
cookie_jar.set('_csrf', csrf_cookie)
cookie_jar.set('StumbleChatV2', session_cookie)
session.cookies.update(cookie_jar)

# Headers for getting WebSocket endpoint
headers = {
    'content-type': 'application/json;charset=UTF-8',
    'cookie': f'_csrf={csrf_cookie}; StumbleChatV2={session_cookie}',
    'csrf-token': csrf_token