Spaces:
Running
Running
File size: 5,125 Bytes
a5b3518 a51206c a5b3518 4d475d7 a5b3518 a51206c a5b3518 a51206c 4d475d7 a51206c a5b3518 a51206c a5b3518 a51206c a5b3518 | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 | # Path: frontend/components/sidebar.py
# Sidebar component for navigation and additional options.
import streamlit as st
from utils.api import get_available_models
from backend.src.database.db_manager import get_db_connection
from components.auth import render_auth_dialog, save_chat_history, save_userDB, load_userDB
@st.cache_data(ttl=60)
def check_db_connection():
"""
Ping the database to check connectivity status
Cached for 60 seconds to avoid DB overloading on Streamlit rerun
"""
try:
conn = get_db_connection()
# Close connection immediately after successful ping to free resources
if hasattr(conn, 'close'):
conn.close()
return True, None
except Exception as e:
return False, str(e)
@st.dialog("Delete Confirmation")
def confirm_delete(idx, session_title):
st.warning(f"Are you sure you want to delete '{session_title}'?")
st.write(":material/warning: This action cannot be undone!")
col1, col2 = st.columns(2)
with col1:
if st.button("Cancel", use_container_width=True):
st.rerun()
with col2:
if st.button("Delete", type="primary", use_container_width=True):
deleted_session = st.session_state.chat_sessions.pop(idx)
st.toast("Item permanently deleted!", icon=":material/delete:")
if st.session_state.messages == deleted_session.get("messages", []):
st.session_state.messages = []
db = load_userDB()
if st.session_state.username in db:
db[st.session_state.username]["sessions"] = st.session_state.chat_sessions
save_userDB(db)
st.rerun()
def render_sidebar():
with st.sidebar:
if st.session_state.auth_stat == 'guest':
st.markdown("## :material/emoji_people: Welcome, Guest!")
else:
display_name = st.session_state.username.capitalize()
st.markdown(f"## :material/emoji_people: Hi, {display_name}!")
st.write("")
if st.session_state.auth_stat == 'guest':
if st.button(":material/login: Login / Sign Up", type="secondary", use_container_width=True):
render_auth_dialog()
else:
if st.button(":material/logout: Logout", use_container_width=True):
st.session_state.auth_stat = 'guest'
st.session_state.username = 'guest'
st.session_state.messages = []
if "chat_sessions" in st.session_state:
del st.session_state.chat_sessions
st.rerun()
if st.button(":material/add: New Chat", use_container_width=True):
if st.session_state.auth_stat != 'guest':
save_chat_history(st.session_state.username, st.session_state.messages)
st.session_state.messages = []
st.rerun()
st.divider()
st.subheader("Database Connection")
is_connected, error_msg = check_db_connection()
if is_connected:
st.success(icon=":material/database:", body="Connected")
else:
st.error(icon=":material/database_off:", body="Disconnected")
st.caption("Connection failed. Please check your configuration.")
st.subheader("Model Selector")
available_models = get_available_models()
st.session_state.current_model = st.selectbox(
"Select model:",
available_models,
label_visibility="collapsed"
)
st.divider()
st.subheader(":material/history: History")
if st.session_state.auth_stat == 'guest':
st.info("You are in Guest Mode. Your conversation history will not be saved.")
else:
st.caption(f"Authenticated as: {st.session_state.username}")
if "chat_sessions" not in st.session_state:
db = load_userDB()
st.session_state.chat_sessions = db.get(st.session_state.username, {}).get("sessions", [])
if st.session_state.messages:
st.button(f":material/chat_bubble: Active Session Logs", disabled=True, use_container_width=True)
if not st.session_state.chat_sessions:
st.caption('Query your database now')
else:
for idx, session in enumerate(st.session_state.chat_sessions):
col1, col2 = st.columns([0.80, 0.20])
with col1:
session_title = session.get("title", f"Chat {idx + 1}")
if st.button(
f"{session_title[:25]}...",
key=f"load_session_{idx}",
use_container_width=True
):
st.session_state.messages = session.get("messages", [])
st.rerun()
with col2:
if st.button(":material/delete:", key=f"delete_session_{idx}"):
confirm_delete(idx, session_title) |