Dudil/sidebar.py at main · coder-royswarnajit/Dudil · GitHub

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
import streamlit as st
import os
from datetime import datetime
from utils import setup_gemini_api, create_new_chat, get_chat_title, save_chat_history

def render_sidebar():
    """Render the complete sidebar with API configuration and chat management"""
    with st.sidebar:
        st.title("🤖 Dudil Emotion-Aware Chat")

        api_key = os.getenv("GEMINI_API_KEY")
        handle_api_connection(api_key)

        st.divider()
        handle_new_chat()

        st.divider()
        render_chat_history()

    return api_key

def handle_api_connection(api_key):
    """Handle API connection logic"""
    if api_key and not st.session_state.api_connected:
        model = setup_gemini_api(api_key)
        if model:
            st.session_state.gemini_model = model
            st.session_state.api_connected = True
        else:
            st.session_state.gemini_model = None
            st.session_state.api_connected = False

    if api_key and not st.session_state.api_connected:
        st.error("❌ Failed to connect to Gemini API")
        if st.button("🔄 Retry Connection"):
            st.session_state.api_connected = False
            st.session_state.gemini_model = None
            st.rerun()

def handle_new_chat():
    """Handle new chat creation"""
    if st.button("➕ New Chat", use_container_width=True):
        if st.session_state.messages:
            st.session_state.chat_history[st.session_state.current_chat_id] = {
                'messages': st.session_state.messages,
                'title': get_chat_title(st.session_state.messages),
                'timestamp': datetime.now().isoformat()
            }
            save_chat_history(st.session_state.chat_history)

        st.session_state.current_chat_id = create_new_chat()
        st.session_state.messages = []
        st.rerun()

def render_chat_history():
    """Render chat history section"""
    st.subheader("📚 Chat History")

    if st.session_state.chat_history:
        if st.button("🗑️ Clear All History", use_container_width=True):
            if st.button("⚠️ Confirm Clear All", use_container_width=True, key="confirm_clear"):
                st.session_state.chat_history = {}
                save_chat_history(st.session_state.chat_history)
                st.success("All chat history cleared!")
                st.rerun()

        st.divider()

        recent_chats = list(st.session_state.chat_history.items())[-10:]
        for chat_id, chat_data in reversed(recent_chats):
            if not isinstance(chat_data, dict):
                continue
            render_chat_item(chat_id, chat_data)
    else:
        st.info("No chat history yet. Start chatting to create history!")

def render_chat_item(chat_id, chat_data):
    col1, col2 = st.columns([3, 1])

    with col1:
        chat_title = chat_data.get('title', 'Untitled Chat')
        timestamp = chat_data.get('timestamp', 'Unknown')[:16]

        if st.button(
            f"💬 {chat_title}",
            key=f"load_{chat_id}",
            use_container_width=True,
            help=f"Created: {timestamp}"
        ):
            if st.session_state.messages:
                st.session_state.chat_history[st.session_state.current_chat_id] = {
                    'messages': st.session_state.messages,
                    'title': get_chat_title(st.session_state.messages),
                    'timestamp': datetime.now().isoformat()
                }

            st.session_state.current_chat_id = chat_id
            st.session_state.messages = chat_data.get('messages', [])
            save_chat_history(st.session_state.chat_history)
            st.rerun()

    with col2:
        if st.button("🗑️", key=f"delete_{chat_id}", help="Delete this chat"):
            if chat_id == st.session_state.current_chat_id:
                st.session_state.current_chat_id = create_new_chat()
                st.session_state.messages = []

            del st.session_state.chat_history[chat_id]
            save_chat_history(st.session_state.chat_history)
            st.success("Chat deleted!")
            st.rerun()