Update services/auth.py

services/auth.py

@@ -1,12 +1,15 @@
+# services/auth.py
+
 import streamlit as st
 from streamlit_authenticator import Authenticate
 from config.settings import settings
 from repositories.user_repo import UserRepo
 from models.db import init_db
 
-# Initialize DB tables
+# 1) Ensure DB tables exist
 init_db()
 
+# 2) User repo
 user_repo = UserRepo(settings.database_url)
 
 def init_auth():
@@ -23,15 +26,22 @@ def init_auth():
         cookie_expiry_days=1,
     )
 
+# 3) Instantiate authenticator
 authenticator = init_auth()
 
 def require_login():
+    # 🔧 Initialize logout flag so login() can read it
+    if 'logout' not in st.session_state:
+        st.session_state['logout'] = False
+
     login_result = authenticator.login(location="sidebar")
-    # Bail out if login form didn't render a result
     if login_result is None:
         st.stop()
 
     name, authentication_status, username = login_result
     if not authentication_status:
         st.stop()
+
+    # 🛑 Render the logout button once logged in
+    authenticator.logout("Logout", location="sidebar")
     return username