Enhancement/make selector sticky

pyproject.toml

@@ -16,7 +16,7 @@ packages = [
 python = "^3.10"
 streamlit = "^1.24.0"
 st-pages = "^0.4.1"
-llama-index = "0.7.4"
+llama-index = "0.8.5.post2"
 pypdf = "^3.9.0"
 docx2txt = "^0.8"
 argon2-cffi = "^21.3.0"

source/docq/manage_users.py

@@ -103,7 +103,7 @@ def authenticate(username: str, password: str) -> Tuple[int, str, bool]:
             log.debug("User found: %s", selected)
             (id_, saved_password, fullname, is_admin) = selected
             try:
-                result = (id_, fullname, is_admin) if PH.verify(saved_password, password) else None
+                result = (id_, fullname, is_admin, username) if PH.verify(saved_password, password) else None
             except VerificationError as e:
                 log.warning("Failing to authenticate user: %s for [%s]", username, e)
                 return None

web/utils/constants.py

@@ -19,6 +19,7 @@ class SessionKeyNameForAuth(Enum):
     ID = "id"
     NAME = "name"
     ADMIN = "admin"
+    USERNAME = "username"
 
 
 class SessionKeyNameForSettings(Enum):

web/utils/handlers.py

@@ -1,6 +1,7 @@
 """Handlers for the web app."""
 
 import asyncio
+import hashlib
 import logging as log
 import math
 from datetime import datetime
@@ -32,13 +33,15 @@
 from .sessions import (
     get_authenticated_user_id,
     get_chat_session,
+    get_user_email,
     set_auth_session,
     set_chat_session,
     set_settings_session,
 )
 
 
 def handle_login(username: str, password: str) -> bool:
+    """Handle login."""
     result = manage_users.authenticate(username, password)
     log.info("Login result: %s", result)
     if result:
@@ -47,6 +50,7 @@ def handle_login(username: str, password: str) -> bool:
                 SessionKeyNameForAuth.ID.name: result[0],
                 SessionKeyNameForAuth.NAME.name: result[1],
                 SessionKeyNameForAuth.ADMIN.name: result[2],
+                SessionKeyNameForAuth.USERNAME.name: result[3],
             }
         )
         set_settings_session(
@@ -363,3 +367,12 @@ def prepare_for_chat(feature: domain.FeatureKey) -> None:
                 feature.type_,
                 SessionKeyNameForChat.HISTORY,
             )
+
+def handle_get_gravatar_url() -> str:
+    """Get Gravatar URL for the specified email."""
+    email = get_user_email()
+    if email is None:
+        email = "[email protected]"
+    size, default, rating = 200, "identicon", "g"
+    email_hash = hashlib.md5(email.lower().encode("utf-8")).hexdigest()  # noqa: S324
+    return f"https://www.gravatar.com/avatar/{email_hash}?s={size}&d={default}&r={rating}"