Set logged_in_user in login_by_token

src/data_storage.py

@@ -1,5 +1,6 @@
 import os
 from secrets import token_urlsafe
+import streamlit as st
 import sqlite3
 
 DATA_DIR = "/data"
@@ -85,6 +86,7 @@ def login_by_token(token: str):
         query_result = cur.execute(f"select user_id from tokens where token = '{token}'").fetchone()
         if query_result:
             user_id = query_result[0]
+            st.session_state["logged_in_user"] = user_id
         else:
             user_id = None
     return user_id