Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from datetime import datetime | |
| from ..database import database | |
| from ..core.agent import initialize_chat | |
| def render_sidebar(): | |
| with st.sidebar: | |
| st.title("🌐 AI Capabilities") | |
| use_web_search = st.toggle("Enable Live Web Search", value=False, help="Allows the AI to intelligently search Google to supplement your textbook notes with up-to-date real-world context.") | |
| st.divider() | |
| st.write("🔒 Secure Session") | |
| if st.button("🚪 Logout", use_container_width=True): | |
| for key in list(st.session_state.keys()): | |
| del st.session_state[key] | |
| st.rerun() | |
| st.divider() | |
| st.title("📚 Chat History") | |
| if st.button("➕ Start New Session", use_container_width=True): | |
| st.session_state.current_session_id = None | |
| st.session_state.notes = None | |
| st.session_state.pdf_text = None | |
| st.session_state.chat_session = None | |
| st.session_state.chat_history = [] | |
| st.session_state.current_file = None | |
| st.session_state.podcast_bytes = None | |
| st.session_state.anki_bytes = None | |
| st.rerun() | |
| st.divider() | |
| try: | |
| sessions = database.get_all_sessions(st.session_state.user_id) | |
| except Exception as e: | |
| sessions = [] | |
| st.error("Could not load chat history. Database connection failed.") | |
| for session in sessions: | |
| try: | |
| date_str = datetime.fromisoformat(session["timestamp"]).strftime("%b %d, %H:%M") | |
| except: | |
| date_str = "Unknown Date" | |
| col1, col2 = st.columns([4, 1]) | |
| btn_label = f"📄 {session['filename'][:15]}... ({date_str})" if len(session['filename']) > 15 else f"📄 {session['filename']} ({date_str})" | |
| with col1: | |
| if st.button(btn_label, key=f"load_{session['id']}"): | |
| st.session_state.current_session_id = session["id"] | |
| st.session_state.pdf_text = session["pdf_text"] | |
| st.session_state.notes = session["notes"] | |
| st.session_state.chat_history = session["chat_history"] | |
| st.session_state.chat_session = initialize_chat(session["pdf_text"], session["chat_history"], use_web_search=use_web_search) | |
| st.session_state.current_file = session["filename"] | |
| st.session_state.podcast_bytes = None | |
| st.session_state.anki_bytes = None | |
| st.rerun() | |
| with col2: | |
| if st.button("🗑️", key=f"del_{session['id']}"): | |
| try: | |
| database.delete_session(session["id"]) | |
| if st.session_state.current_session_id == session["id"]: | |
| st.session_state.current_session_id = None | |
| st.session_state.notes = None | |
| st.session_state.pdf_text = None | |
| st.session_state.chat_session = None | |
| st.session_state.chat_history = [] | |
| st.session_state.current_file = None | |
| st.session_state.podcast_bytes = None | |
| st.session_state.anki_bytes = None | |
| st.rerun() | |
| except Exception as e: | |
| st.error("Failed to delete session from the database.") | |
| st.divider() | |
| st.title("🛠️ Customization") | |
| st.write("Tweak the output of your study notes!") | |
| tone = st.selectbox("Tone", ["Academic", "Simple/Beginner", "Conversational"]) | |
| focus = st.selectbox("Focus area", ["General Summary", "Flashcards", "Code/Technical Examples", "Exam Prep Questions"]) | |
| length = st.select_slider("Length", options=["Short", "Medium", "Detailed"], value="Medium") | |
| return use_web_search, tone, focus, length | |