Update app.py

app.py

@@ -13,7 +13,7 @@ st.set_page_config(
     initial_sidebar_state="collapsed"
 )
 
-# Enhanced mobile-responsive CSS
+# White background
 st.markdown("""
 <style>
     .stApp {
@@ -39,211 +39,6 @@ st.markdown("""
         font-size: 0.8em;
         font-style: italic;
     }
-    
-    .welcome-title {
-        text-align: center;
-        font-size: 2.5rem;
-        font-weight: bold;
-        margin-bottom: 0.5rem;
-    }
-    
-    .welcome-subtitle {
-        text-align: center;
-        font-size: 1.2rem;
-        color: #666;
-        margin-bottom: 2rem;
-    }
-    
-    /* Mobile optimizations */
-    @media (max-width: 768px) {
-        /* Reduce overall padding */
-        .main .block-container {
-            padding-top: 1rem;
-            padding-left: 1rem;
-            padding-right: 1rem;
-            padding-bottom: 1rem;
-        }
-        
-        /* Make titles smaller on mobile */
-        .welcome-title {
-            font-size: 1.8rem;
-            margin-bottom: 0.3rem;
-        }
-        
-        .welcome-subtitle {
-            font-size: 1rem;
-            margin-bottom: 1rem;
-        }
-        
-        /* Compact chat messages */
-        .stChatMessage {
-            margin-bottom: 0.5rem !important;
-            padding: 0.5rem !important;
-        }
-        
-        /* Smaller chat message content */
-        .stChatMessage > div {
-            padding: 0.3rem 0.5rem !important;
-            font-size: 0.9rem !important;
-            line-height: 1.3 !important;
-        }
-        
-        /* Compact chat input */
-        .stChatInput > div {
-            padding: 0.5rem !important;
-        }
-        
-        .stChatInput input {
-            font-size: 0.9rem !important;
-            padding: 0.5rem !important;
-        }
-        
-        /* Reduce spacing in chat container */
-        .stChatFloatingInputContainer {
-            padding: 0.5rem !important;
-        }
-        
-        /* Make chat avatars smaller */
-        .stChatMessage img {
-            width: 24px !important;
-            height: 24px !important;
-        }
-        
-        /* Compact attribution text */
-        .model-attribution {
-            font-size: 0.7rem !important;
-            margin-top: 0.2rem !important;
-        }
-        
-        /* Reduce paragraph spacing in messages */
-        .stChatMessage p {
-            margin-bottom: 0.5rem !important;
-        }
-        
-        /* Make buttons smaller */
-        .stButton button {
-            padding: 0.25rem 0.5rem !important;
-            font-size: 0.8rem !important;
-        }
-        
-        /* Compact sidebar */
-        .css-1d391kg {
-            padding: 0.5rem !important;
-        }
-        
-        /* Smaller text in sidebar */
-        .css-1d391kg .stMarkdown {
-            font-size: 0.85rem !important;
-        }
-        
-        /* Reduce spacing between elements */
-        .element-container {
-            margin-bottom: 0.5rem !important;
-        }
-        
-        /* Compact selectbox and other inputs */
-        .stSelectbox > div > div {
-            font-size: 0.85rem !important;
-        }
-        
-        /* Reduce caption size */
-        .stCaption {
-            font-size: 0.7rem !important;
-        }
-        
-        /* Make status indicators smaller */
-        .stSuccess, .stInfo, .stWarning, .stError {
-            font-size: 0.8rem !important;
-            padding: 0.3rem !important;
-        }
-        
-        /* Compact dividers */
-        hr {
-            margin: 0.5rem 0 !important;
-        }
-        
-        /* Reduce header sizes in sidebar */
-        .css-1d391kg h1, .css-1d391kg h2, .css-1d391kg h3 {
-            font-size: 1rem !important;
-            margin-bottom: 0.3rem !important;
-        }
-        
-        /* Make chat messages look more like mobile messages */
-        .stChatMessage[data-testid="user-message"] {
-            background-color: #007bff !important;
-            color: white !important;
-            border-radius: 1rem !important;
-            margin-left: 20% !important;
-        }
-        
-        .stChatMessage[data-testid="assistant-message"] {
-            background-color: #f1f1f1 !important;
-            color: black !important;
-            border-radius: 1rem !important;
-            margin-right: 20% !important;
-        }
-        
-        /* Force AI message text to be black */
-        .stChatMessage[data-testid="assistant-message"] * {
-            color: black !important;
-        }
-        
-        /* Make sure all AI message content is visible */
-        .stChatMessage[data-testid="assistant-message"] .stMarkdown,
-        .stChatMessage[data-testid="assistant-message"] p,
-        .stChatMessage[data-testid="assistant-message"] div {
-            color: black !important;
-        }
-        
-        /* Reduce overall font size for mobile */
-        body, .stApp {
-            font-size: 14px !important;
-        }
-        
-        /* Make expander more compact */
-        .streamlit-expanderHeader {
-            font-size: 0.8rem !important;
-            padding: 0.3rem !important;
-        }
-        
-        .streamlit-expanderContent {
-            padding: 0.3rem !important;
-        }
-        
-        /* Compact text areas */
-        .stTextArea textarea {
-            font-size: 0.8rem !important;
-        }
-    }
-    
-    /* Extra small screens (phones in portrait) */
-    @media (max-width: 480px) {
-        .main .block-container {
-            padding: 0.5rem;
-        }
-        
-        .welcome-title {
-            font-size: 1.5rem;
-        }
-        
-        .welcome-subtitle {
-            font-size: 0.9rem;
-        }
-        
-        .stChatMessage > div {
-            font-size: 0.85rem !important;
-        }
-        
-        .stChatInput input {
-            font-size: 0.85rem !important;
-        }
-        
-        /* Even more compact messages */
-        .stChatMessage {
-            margin-bottom: 0.3rem !important;
-            padding: 0.3rem !important;
-        }
-    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -251,8 +46,6 @@ st.markdown("""
 HISTORY_FILE = "chat_history.json"
 # NEW: File to store online users
 USERS_FILE = "online_users.json"
-# NEW: File to store chat sessions
-SESSIONS_FILE = "chat_sessions.json"
 
 
 def load_chat_history():
@@ -284,98 +77,6 @@ def clear_chat_history():
     except Exception as e:
         st.error(f"Error clearing chat history: {e}")
 
-
-# NEW: Chat Sessions Management
-def load_chat_sessions():
-    """Load all chat sessions"""
-    try:
-        if os.path.exists(SESSIONS_FILE):
-            with open(SESSIONS_FILE, 'r', encoding='utf-8') as f:
-                return json.load(f)
-    except Exception as e:
-        st.error(f"Error loading chat sessions: {e}")
-    return {}
-
-
-def save_chat_sessions(sessions):
-    """Save chat sessions to file"""
-    try:
-        with open(SESSIONS_FILE, 'w', encoding='utf-8') as f:
-            json.dump(sessions, f, ensure_ascii=False, indent=2)
-    except Exception as e:
-        st.error(f"Error saving chat sessions: {e}")
-
-
-def get_session_id():
-    """Get or create session ID"""
-    if 'session_id' not in st.session_state:
-        st.session_state.session_id = str(uuid.uuid4())
-    return st.session_state.session_id
-
-
-def get_chat_title(messages):
-    """Generate a title for the chat based on first user message"""
-    if not messages:
-        return "New Chat"
-    
-    for msg in messages:
-        if msg["role"] == "user":
-            content = msg["content"]
-            # Take first 30 characters and add ellipsis if longer
-            if len(content) > 30:
-                return content[:30] + "..."
-            return content
-    return "New Chat"
-
-
-def save_current_session():
-    """Save current chat session"""
-    if not st.session_state.messages:
-        return
-        
-    sessions = load_chat_sessions()
-    session_id = get_session_id()
-    
-    sessions[session_id] = {
-        "title": get_chat_title(st.session_state.messages),
-        "messages": st.session_state.messages,
-        "created_at": datetime.now().isoformat(),
-        "updated_at": datetime.now().isoformat()
-    }
-    
-    save_chat_sessions(sessions)
-
-
-def load_session(session_id):
-    """Load a specific chat session"""
-    sessions = load_chat_sessions()
-    if session_id in sessions:
-        st.session_state.messages = sessions[session_id]["messages"]
-        st.session_state.session_id = session_id
-        return True
-    return False
-
-
-def delete_session(session_id):
-    """Delete a chat session"""
-    sessions = load_chat_sessions()
-    if session_id in sessions:
-        del sessions[session_id]
-        save_chat_sessions(sessions)
-        return True
-    return False
-
-
-def start_new_chat():
-    """Start a new chat session"""
-    # Save current session if it has messages
-    if st.session_state.messages:
-        save_current_session()
-    
-    # Clear current chat and create new session
-    st.session_state.messages = []
-    st.session_state.session_id = str(uuid.uuid4())
-
 # NEW: User tracking functions
 
 
@@ -446,10 +147,6 @@ def get_online_count():
 if "messages" not in st.session_state:
     st.session_state.messages = load_chat_history()
 
-# Initialize session ID
-if "session_id" not in st.session_state:
-    st.session_state.session_id = str(uuid.uuid4())
-
 # Get API key
 OPENROUTER_API_KEY = os.environ.get("OPENROUTER_API_KEY")
 
@@ -545,282 +242,210 @@ def get_ai_response(messages, model="openai/gpt-3.5-turbo"):
 
 
 # Header
-st.markdown('<div class="welcome-title">Welcome back!</div>', unsafe_allow_html=True)
-st.markdown('<div class="welcome-subtitle">Which AI model do you want to chat with today?</div>', unsafe_allow_html=True)
-
-# Create main layout with chat history sidebar
-col1, col2 = st.columns([1, 3])
+st.title("Chat Flow 🕷")
+st.caption("10 powerful Models, one simple chat.")
+
+# Sidebar
+with st.sidebar:
+    st.header("Settings")
+
+    # API Status
+    status = check_api_status()
+    if status == "Connected":
+        st.success("🟢 API Connected")
+    elif status == "No API Key":
+        st.error("No API Key")
+    else:
+        st.warning("Connection Issue")
 
-# Chat History Sidebar (Left Column)
-with col1:
-    st.header("💬 Chat History")
-    
-    # New Chat Button
-    if st.button("➕ New Chat", use_container_width=True, type="primary"):
-        start_new_chat()
-        st.rerun()
-    
     st.divider()
-    
-    # Load and display chat sessions
-    sessions = load_chat_sessions()
-    current_session_id = get_session_id()
-    
-    if sessions:
-        st.subheader("Previous Chats")
-        
-        # Sort sessions by updated_at (most recent first)
-        sorted_sessions = sorted(
-            sessions.items(), 
-            key=lambda x: x[1].get("updated_at", x[1].get("created_at", "")), 
-            reverse=True
-        )
-        
-        for session_id, session_data in sorted_sessions:
-            # Create container for each chat item
-            chat_container = st.container()
-            
-            with chat_container:
-                # Show current chat with different styling
-                if session_id == current_session_id:
-                    st.markdown(f"🔹 **{session_data['title']}**")
-                else:
-                    col_load, col_delete = st.columns([3, 1])
-                    
-                    with col_load:
-                        if st.button(
-                            f"💭 {session_data['title']}", 
-                            key=f"load_{session_id}",
-                            use_container_width=True
-                        ):
-                            # Save current session before switching
-                            if st.session_state.messages:
-                                save_current_session()
-                            
-                            # Load selected session
-                            load_session(session_id)
-                            st.rerun()
-                    
-                    with col_delete:
-                        if st.button("🗑️", key=f"delete_{session_id}"):
-                            delete_session(session_id)
-                            # If deleted session was current, start new chat
-                            if session_id == current_session_id:
-                                start_new_chat()
-                            st.rerun()
-                
-                # Show session info
-                if "updated_at" in session_data:
-                    update_time = datetime.fromisoformat(session_data["updated_at"])
-                    st.caption(f"Updated: {update_time.strftime('%m/%d %H:%M')}")
-                
-                st.markdown("---")
-    
+
+    # NEW: Live Users Section
+    st.header("👥 Who's Online")
+
+    # Update that you're online
+    online_count = update_online_users()
+
+    # Show live count
+    if online_count == 1:
+        st.info("🟢 Just you online")
     else:
-        st.info("No previous chats yet")
-    
-    # Auto-save current session periodically
-    if st.session_state.messages:
-        save_current_session()
-
-# Main Chat Area (Right Column)
-with col2:
-    # Sidebar
-    with st.sidebar:
-        st.header("Settings")
-
-        # API Status
-        status = check_api_status()
-        if status == "Connected":
-            st.success("🟢 API Connected")
-        elif status == "No API Key":
-            st.error("No API Key")
+        st.success(f"🟢 {online_count} people online")
+
+    # Show your session
+    your_id = get_user_id()
+    st.caption(f"You: User-{your_id}")
+
+    # Quick refresh button
+    if st.button("Refresh", use_container_width=True):
+        st.rerun()
+
+    # === NEW: DEBUG SECTION ===
+    with st.expander("🔍 Debug Info"):
+        if os.path.exists(USERS_FILE):
+            with open(USERS_FILE, 'r') as f:
+                users = json.load(f)
+            st.write(f"Users in file: {len(users)}")
+            for uid, data in users.items():
+                last_seen_time = datetime.fromisoformat(data['last_seen'])
+                time_ago = datetime.now() - last_seen_time
+                minutes_ago = int(time_ago.total_seconds() / 60)
+                st.write(f"- {uid}: {minutes_ago} min ago")
         else:
-            st.warning("Connection Issue")
+            st.write("No users file yet")
+    # === END DEBUG SECTION ===
 
-        st.divider()
+    st.divider()
 
-        # NEW: Live Users Section
-        st.header("👥 Who's Online")
+    # All models including new ones
+    models = [
+        ("GPT-3.5 Turbo", "openai/gpt-3.5-turbo"),
+        ("LLaMA 3.1 8B", "meta-llama/llama-3.1-8b-instruct"),
+        ("LLaMA 3.1 70B", "meta-llama/llama-3.1-70b-instruct"),
+        ("DeepSeek Chat v3", "deepseek/deepseek-chat-v3-0324:free"),
+        ("DeepSeek R1", "deepseek/deepseek-r1-0528:free"),
+        ("Qwen3 Coder", "qwen/qwen3-coder:free"),
+        ("Microsoft MAI DS R1", "microsoft/mai-ds-r1:free"),
+        ("Gemma 3 27B", "google/gemma-3-27b-it:free"),
+        ("Gemma 3 4B", "google/gemma-3-4b-it:free"),
+        ("Auto (Best Available)", "openrouter/auto")
+    ]
+
+    model_names = [name for name, _ in models]
+    model_ids = [model_id for _, model_id in models]
+
+    selected_index = st.selectbox("Model", range(len(model_names)),
+                                  format_func=lambda x: model_names[x],
+                                  index=0)
+    selected_model = model_ids[selected_index]
+
+    # Show selected model ID in green
+    st.markdown(
+        f"**Model ID:** <span class='model-id'>{selected_model}</span>", unsafe_allow_html=True)
 
-        # Update that you're online
-        online_count = update_online_users()
+    st.divider()
 
-        # Show live count
-        if online_count == 1:
-            st.info("🟢 Just you online")
-        else:
-            st.success(f"🟢 {online_count} people online")
+    # Chat History Controls
+    st.header("Chat History")
 
-        # Show your session
-        your_id = get_user_id()
-        st.caption(f"You: User-{your_id}")
+    # Show number of messages
+    if st.session_state.messages:
+        st.info(f"Messages stored: {len(st.session_state.messages)}")
+
+    # Auto-save toggle
+    auto_save = st.checkbox("Auto-save messages", value=True)
+
+    # Manual save/load buttons
+    col1, col2 = st.columns(2)
+    with col1:
+        if st.button("Save History", use_container_width=True):
+            save_chat_history(st.session_state.messages)
+            st.success("History saved!")
 
-        # Quick refresh button
-        if st.button("Refresh", use_container_width=True):
+    with col2:
+        if st.button("Load History", use_container_width=True):
+            st.session_state.messages = load_chat_history()
+            st.success("History loaded!")
             st.rerun()
 
-        # === NEW: DEBUG SECTION ===
-        with st.expander("🔍 Debug Info"):
-            if os.path.exists(USERS_FILE):
-                with open(USERS_FILE, 'r') as f:
-                    users = json.load(f)
-                st.write(f"Users in file: {len(users)}")
-                for uid, data in users.items():
-                    last_seen_time = datetime.fromisoformat(data['last_seen'])
-                    time_ago = datetime.now() - last_seen_time
-                    minutes_ago = int(time_ago.total_seconds() / 60)
-                    st.write(f"- {uid}: {minutes_ago} min ago")
-            else:
-                st.write("No users file yet")
-        # === END DEBUG SECTION ===
-
-        st.divider()
-
-        # All models including new ones
-        models = [
-            ("GPT-3.5 Turbo", "openai/gpt-3.5-turbo"),
-            ("LLaMA 3.1 8B", "meta-llama/llama-3.1-8b-instruct"),
-            ("LLaMA 3.1 70B", "meta-llama/llama-3.1-70b-instruct"),
-            ("DeepSeek Chat v3", "deepseek/deepseek-chat-v3-0324:free"),
-            ("DeepSeek R1", "deepseek/deepseek-r1-0528:free"),
-            ("Qwen3 Coder", "qwen/qwen3-coder:free"),
-            ("Microsoft MAI DS R1", "microsoft/mai-ds-r1:free"),
-            ("Gemma 3 27B", "google/gemma-3-27b-it:free"),
-            ("Gemma 3 4B", "google/gemma-3-4b-it:free"),
-            ("Auto (Best Available)", "openrouter/auto")
-        ]
-
-        model_names = [name for name, _ in models]
-        model_ids = [model_id for _, model_id in models]
-
-        selected_index = st.selectbox("Model", range(len(model_names)),
-                                      format_func=lambda x: model_names[x],
-                                      index=0)
-        selected_model = model_ids[selected_index]
-
-        # Show selected model ID in green
-        st.markdown(
-            f"**Model ID:** <span class='model-id'>{selected_model}</span>", unsafe_allow_html=True)
-
-        st.divider()
-
-        # Chat History Controls
-        st.header("Chat History")
-
-        # Show number of messages
-        if st.session_state.messages:
-            st.info(f"Messages stored: {len(st.session_state.messages)}")
-
-        # Auto-save toggle
-        auto_save = st.checkbox("Auto-save messages", value=True)
-
-        # Manual save/load buttons
-        col1_sidebar, col2_sidebar = st.columns(2)
-        with col1_sidebar:
-            if st.button("Save History", use_container_width=True):
-                save_chat_history(st.session_state.messages)
-                st.success("History saved!")
-
-        with col2_sidebar:
-            if st.button("Load History", use_container_width=True):
-                st.session_state.messages = load_chat_history()
-                st.success("History loaded!")
-                st.rerun()
-
-        st.divider()
-
-        # View History
-        if st.button("View History File", use_container_width=True):
-            if os.path.exists(HISTORY_FILE):
-                with open(HISTORY_FILE, 'r', encoding='utf-8') as f:
-                    history_content = f.read()
-                st.text_area("Chat History (JSON)", history_content, height=200)
-            else:
-                st.warning("No history file found")
+    st.divider()
 
-        # Download History
+    # View History
+    if st.button("View History File", use_container_width=True):
         if os.path.exists(HISTORY_FILE):
-            with open(HISTORY_FILE, 'rb') as f:
-                st.download_button(
-                    label="Download History",
-                    data=f.read(),
-                    file_name=f"chat_history_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
-                    mime="application/json",
-                    use_container_width=True
-                )
-
-        st.divider()
-
-        # Clear controls
-        if st.button("Clear Chat", use_container_width=True, type="secondary"):
-            clear_chat_history()
-            st.success("Chat cleared!")
-            st.rerun()
+            with open(HISTORY_FILE, 'r', encoding='utf-8') as f:
+                history_content = f.read()
+            st.text_area("Chat History (JSON)", history_content, height=200)
+        else:
+            st.warning("No history file found")
+
+    # Download History
+    if os.path.exists(HISTORY_FILE):
+        with open(HISTORY_FILE, 'rb') as f:
+            st.download_button(
+                label="Download History",
+                data=f.read(),
+                file_name=f"chat_history_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
+                mime="application/json",
+                use_container_width=True
+            )
+
+    st.divider()
+
+    # Clear controls
+    if st.button("Clear Chat", use_container_width=True, type="secondary"):
+        clear_chat_history()
+        st.success("Chat cleared!")
+        st.rerun()
 
-    # Display chat messages
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            # Check if this is an assistant message with attribution
-            if message["role"] == "assistant" and "Response created by:" in message["content"]:
-                # Split content and attribution
-                parts = message["content"].split("\n\n---\n*Response created by:")
-                main_content = parts[0]
-                if len(parts) > 1:
-                    model_name = parts[1].replace("***", "").replace("**", "")
-                    st.markdown(main_content)
-                    st.markdown(
-                        f"<div class='model-attribution'>Response created by: <strong>{model_name}</strong></div>", unsafe_allow_html=True)
-                else:
-                    st.markdown(message["content"])
+# Show welcome message when no messages
+
+# Display chat messages
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        # Check if this is an assistant message with attribution
+        if message["role"] == "assistant" and "Response created by:" in message["content"]:
+            # Split content and attribution
+            parts = message["content"].split("\n\n---\n*Response created by:")
+            main_content = parts[0]
+            if len(parts) > 1:
+                model_name = parts[1].replace("***", "").replace("**", "")
+                st.markdown(main_content)
+                st.markdown(
+                    f"<div class='model-attribution'>Response created by: <strong>{model_name}</strong></div>", unsafe_allow_html=True)
             else:
                 st.markdown(message["content"])
+        else:
+            st.markdown(message["content"])
 
-    # Chat input
-    if prompt := st.chat_input("Chat Smarter. Chat many Brains"):
-        # NEW: Update online status when user sends message
-        update_online_users()
+# Chat input - MUST be at the main level, not inside sidebar or columns
+if prompt := st.chat_input("Chat Smarter. Chat many Brains"):
+    # NEW: Update online status when user sends message
+    update_online_users()
 
-        # Add user message
-        user_message = {"role": "user", "content": prompt}
-        st.session_state.messages.append(user_message)
+    # Add user message
+    user_message = {"role": "user", "content": prompt}
+    st.session_state.messages.append(user_message)
 
-        # Auto-save if enabled
-        if auto_save:
-            save_chat_history(st.session_state.messages)
+    # Auto-save if enabled
+    if 'auto_save' not in locals():
+        auto_save = True  # Default value if not set in sidebar
+    
+    if auto_save:
+        save_chat_history(st.session_state.messages)
 
-        # Display user message
-        with st.chat_message("user"):
-            st.markdown(prompt)
+    # Display user message
+    with st.chat_message("user"):
+        st.markdown(prompt)
 
-        # Get AI response
-        with st.chat_message("assistant"):
-            placeholder = st.empty()
+    # Get AI response
+    with st.chat_message("assistant"):
+        placeholder = st.empty()
 
-            full_response = ""
-            try:
-                for response in get_ai_response(st.session_state.messages, selected_model):
-                    full_response = response
-                    placeholder.markdown(full_response + "▌")
-
-                # Remove cursor and show final response
-                placeholder.markdown(full_response)
-
-            except Exception as e:
-                error_msg = f"An error occurred: {str(e)}"
-                placeholder.markdown(error_msg)
-                full_response = error_msg
-
-        # Add AI response to messages with attribution
-        full_response_with_attribution = full_response + \
-            f"\n\n---\n*Response created by: **{model_names[selected_index]}***"
-        assistant_message = {"role": "assistant",
-                             "content": full_response_with_attribution}
-        st.session_state.messages.append(assistant_message)
-
-        # Auto-save if enabled
-        if auto_save:
-            save_chat_history(st.session_state.messages)
-
-    # Show currently using model
-    st.caption(f"Currently using: **{model_names[selected_index]}**")
+        full_response = ""
+        try:
+            for response in get_ai_response(st.session_state.messages, selected_model):
+                full_response = response
+                placeholder.markdown(full_response + "▌")
+
+            # Remove cursor and show final response
+            placeholder.markdown(full_response)
+
+        except Exception as e:
+            error_msg = f"An error occurred: {str(e)}"
+            placeholder.markdown(error_msg)
+            full_response = error_msg
+
+    # Add AI response to messages with attribution
+    full_response_with_attribution = full_response + \
+        f"\n\n---\n*Response created by: **{model_names[selected_index]}***"
+    assistant_message = {"role": "assistant",
+                         "content": full_response_with_attribution}
+    st.session_state.messages.append(assistant_message)
+
+    # Auto-save if enabled
+    if auto_save:
+        save_chat_history(st.session_state.messages)
+
+# Show currently using model
+st.caption(f"Currently using: **{model_names[selected_index]}**")