Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import time
 import os
 
-# --- 1. PAGE CONFIGURATION ---
+# --- 1. PAGE CONFIGURATION (MUST BE FIRST) ---
 st.set_page_config(
     page_title="Voice of the Sign", 
     page_icon="🦅", 
@@ -10,180 +10,229 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# --- 2. BACKEND LOADING ---
-backend_loaded = False
+# --- 2. IMPORTS & ERROR HANDLING ---
 try:
-    from app import get_rag_chain, search_archives
-    backend_loaded = True
+    from app import get_rag_chain, search_archives, build_index_on_server, check_files, raw_text_search
 except ImportError as e:
-    st.error(f"❌ CRITICAL SYSTEM ERROR: Could not load the Brain.\n\nDetails: {e}")
+    st.error(f"❌ CRITICAL ERROR: Could not load the Brain.\n\nDetails: {e}")
+    st.stop()
 
-# --- 3. CSS STYLING ---
+# --- 3. PROFESSIONAL "EAGLE & SCROLL" THEME ---
 st.markdown("""
 <style>
-    @import url('https://fonts.googleapis.com/css2?family=Merriweather:wght@300;400;700&family=Inter:wght@300;400;600&display=swap');
-    html, body, [class*="css"] { font-family: 'Inter', sans-serif; }
-    h1, h2, h3 { font-family: 'Merriweather', serif !important; }
-    #MainMenu {visibility: hidden;}
-    footer {visibility: hidden;}
-    header {visibility: hidden;}
-    
-    /* CHAT BUBBLES */
-    .stChatMessage { background-color: transparent !important; border: none !important; }
-    div[data-testid="stChatMessage"][data-test-role="user"] {
-        background-color: #E8F0FE;
-        border: 1px solid #D0E1FD;
-        border-radius: 15px 15px 0px 15px;
-        padding: 20px;
-        color: #1a1a1a;
+    /* GLOBAL FONTS */
+    @import url('https://fonts.googleapis.com/css2?family=Merriweather:ital,wght@0,300;0,400;0,700;1,300&family=Lato:wght@400;700&display=swap');
+
+    html, body, [class*="css"] {
+        font-family: 'Lato', sans-serif;
     }
-    div[data-testid="stChatMessage"][data-test-role="assistant"] {
-        background-color: #FFFFFF;
-        border: 1px solid #E0E0E0;
-        border-radius: 15px 15px 15px 0px;
-        padding: 20px;
-        color: #1a1a1a;
+
+    /* HEADERS */
+    h1, h2, h3 {
+        font-family: 'Merriweather', serif;
+        color: #2C3E50;
     }
-    @media (prefers-color-scheme: dark) {
-        div[data-testid="stChatMessage"][data-test-role="user"] {
-            background-color: #2b313e !important;
-            border: 1px solid #3b4252 !important;
-            color: #FFFFFF !important;
-        }
-        div[data-testid="stChatMessage"][data-test-role="assistant"] {
-            background-color: #262730 !important;
-            border: 1px solid #3b4252 !important;
-            color: #FFFFFF !important;
-        }
+
+    /* CHAT BUBBLES - USER */
+    .stChatMessage[data-testid="stChatMessage"]:nth-child(odd) {
+        background-color: #E8F4F8; /* Soft Blue-Grey */
+        border-left: 4px solid #3498DB;
+        color: #2C3E50 !important;
     }
-    /* QUOTE CARDS */
-    .quote-card {
-        background-color: #FFFFFF !important;
-        border-left: 5px solid #D4AF37;
-        padding: 20px;
-        margin-bottom: 15px;
-        border-radius: 8px;
+
+    /* CHAT BUBBLES - ASSISTANT (BRANHAM) */
+    .stChatMessage[data-testid="stChatMessage"]:nth-child(even) {
+        background-color: #FDF5E6; /* Old Lace / Vintage Paper */
+        border-left: 4px solid #8B5E3C; /* Saddle Brown */
+        color: #2C3E50 !important;
     }
-    .quote-text {
+
+    /* TEXT COLOR OVERRIDE (For Dark Mode Compatibility) */
+    div[data-testid="stChatMessage"] p, 
+    div[data-testid="stChatMessage"] div {
+        color: #2C3E50 !important; /* Dark Slate for readability */
+        font-size: 1.05em;
+        line-height: 1.6;
+    }
+
+    /* CITATION BOXES */
+    .source-box {
+        background-color: #FFFFFF;
+        border: 1px solid #D1D1D1;
+        border-radius: 8px;
+        padding: 12px;
+        margin-top: 10px;
         font-family: 'Merriweather', serif;
-        font-style: italic;
-        color: #1a1a1a !important;
+        font-size: 0.9em;
+        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
     }
-    .quote-meta {
+    .source-title {
+        color: #8B5E3C;
         font-weight: bold;
-        color: #D4AF37 !important;
-        margin-bottom: 8px;
+        border-bottom: 1px solid #EEE;
+        padding-bottom: 5px;
+        margin-bottom: 5px;
+    }
+    .source-text {
+        font-style: italic;
+        color: #555;
+    }
+
+    /* SIDEBAR STYLING */
+    section[data-testid="stSidebar"] {
+        background-color: #F8F9FA;
     }
+    
+    /* REMOVE DEFAULT BRANDING */
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
+    
 </style>
 """, unsafe_allow_html=True)
 
-# --- 4. NAVIGATION (Main Page) ---
-col1, col2 = st.columns([3, 1])
-with col1:
+# --- 4. SIDEBAR DASHBOARD ---
+with st.sidebar:
+    st.image("https://img.icons8.com/ios-filled/100/8B5E3C/eagle.png", width=60)
+    st.title("Voice of the Sign")
+    st.markdown("---")
+    
+    # Mode Selector
     mode = st.radio(
-        "Select Mode:", 
-        ["🗣️ Chat with the Message", "🔍 Search Quotes Only"], 
-        horizontal=True,
-        label_visibility="collapsed"
+        "Select Operation Mode:", 
+        ["🗣️ Chat with Brother Branham", "🔍 Quote Search Only"],
+        captions=["AI Persona (1965 Era)", "Exact Text Search"]
     )
-with col2:
-    if st.button("🔄 Reset", use_container_width=True):
+    
+    st.markdown("---")
+    
+    # Admin Panel (Collapsible)
+    with st.expander("🛠️ Admin & Diagnostics"):
+        st.caption("Server Status")
+        file_count = check_files()
+        st.info(f"📚 Index Size: {file_count} Sermons")
+        
+        if st.button("🔄 Rebuild Index (Admin Only)"):
+            with st.status("Rebuilding Knowledge Base...", expanded=True) as status:
+                st.write("Extracting Text...")
+                res = build_index_on_server()
+                st.write(res)
+                status.update(label="Index Rebuilt!", state="complete", expanded=False)
+                time.sleep(1)
+                st.rerun()
+
+        st.caption("Keyword Check")
+        test_word = st.text_input("Search Raw Text", placeholder="e.g. Coleman")
+        if test_word:
+            hits = raw_text_search(test_word)
+            if hits:
+                st.success(f"Found in: {hits[0]}...")
+            else:
+                st.error("Not found in raw text.")
+
+    if st.button("🧹 Clear Chat History", type="primary", use_container_width=True):
         st.session_state.messages = []
         st.rerun()
 
-st.divider()
-
-# --- 5. STATE & LOAD ---
-if "messages" not in st.session_state: st.session_state.messages = []
+# --- 5. MAIN CONTENT AREA ---
 
-@st.cache_resource(show_spinner=False)
-def load_system():
-    if backend_loaded: return get_rag_chain()
-    return None
+# Initialize History
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-# --- 6. HEADER ---
-if mode.startswith("🗣️"):
-    st.markdown("<h2 style='text-align: center;'>🦅 The 7th Handle</h2>", unsafe_allow_html=True)
-    st.caption("Interactive AI Archive • Ask any question")
-else:
-    st.markdown("<h2 style='text-align: center;'>🔍 The Table</h2>", unsafe_allow_html=True)
-    st.caption("Direct Quote Search • Search the tapes")
+# --- MODE A: CHAT (PERSISTENT HISTORY) ---
+if mode == "🗣️ Chat with Brother Branham":
+    st.markdown("<h1 style='text-align: center; color: #8B5E3C;'>🦅 THE 7TH HANDLE</h1>", unsafe_allow_html=True)
+    st.markdown("<p style='text-align: center; color: #7F8C8D;'><i>Interactive Doctrinal Archive • Speaking as the Prophet (1965)</i></p>", unsafe_allow_html=True)
+    st.divider()
 
-if not backend_loaded: st.stop()
+    # 1. DISPLAY EXISTING HISTORY
+    # We loop through history FIRST so it stays on screen when new input arrives
+    if not st.session_state.messages:
+        with st.chat_message("assistant", avatar="📖"):
+            st.markdown("God bless you. I am here to search the Message for you. What is on your heart?")
 
-# --- 7. MAIN LOGIC ---
-placeholder = "Ask a question..." if mode.startswith("🗣️") else "Search for a quote (e.g., 'Coleman')..."
+    for msg in st.session_state.messages:
+        with st.chat_message(msg["role"], avatar="📖" if msg["role"] == "assistant" else "👤"):
+            st.markdown(msg["content"])
+            # Render Sources nicely if they exist
+            if "sources" in msg and msg["sources"]:
+                with st.expander("📜 View Sermon References"):
+                    for src in msg["sources"]:
+                        st.markdown(f"**📼 {src}**")
 
-if prompt := st.chat_input(placeholder):
-    
-    # === CHAT MODE ===
-    if mode.startswith("🗣️"):
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user", avatar="👤"): st.markdown(prompt)
+    # 2. HANDLE NEW INPUT
+    if prompt := st.chat_input("Ask a question on doctrine..."):
         
+        # A. Append User Message
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user", avatar="👤"):
+            st.markdown(prompt)
+
+        # B. Generate & Append Assistant Message
         with st.chat_message("assistant", avatar="📖"):
-            with st.spinner("Searching..."):
+            with st.spinner("Searching the storehouse..."):
                 try:
-                    chain = load_system()
+                    # Run the Brain
+                    chain = get_rag_chain()
+                    if not chain:
+                        st.error("⚠️ System Index is missing. Please Rebuild Index in Sidebar.")
+                        st.stop()
+                        
                     response = chain.invoke({"question": prompt})
+                    answer_text = response['result']
                     
-                    result_text = response['result']
-                    source_docs = response.get('source_documents', [])
-                    
-                    # Format Sources
-                    unique_sources = list(set([doc.metadata.get('source', 'Unknown') for doc in source_docs]))
-                    
-                    # Stream Output
-                    placeholder_box = st.empty()
+                    # Extract Sources
+                    raw_docs = response.get('source_documents', [])
+                    unique_sources = list(set([d.metadata.get('source', 'Unknown') for d in raw_docs]))
+
+                    # Simulate Typing Effect
+                    message_placeholder = st.empty()
                     full_response = ""
-                    for chunk in result_text.split():
+                    for chunk in answer_text.split():
                         full_response += chunk + " "
                         time.sleep(0.04)
-                        placeholder_box.markdown(full_response + "▌")
-                    placeholder_box.markdown(full_response)
-                    
+                        message_placeholder.markdown(full_response + "▌")
+                    message_placeholder.markdown(full_response)
+
+                    # Show Sources
+                    if unique_sources:
+                        with st.expander("📜 View Sermon References"):
+                            for src in unique_sources:
+                                st.markdown(f"- {src}")
+
+                    # Save to State (So it persists next run!)
                     st.session_state.messages.append({
                         "role": "assistant", 
                         "content": full_response,
                         "sources": unique_sources
                     })
                     
-                    if unique_sources:
-                        with st.expander("📚 Sermon References"):
-                            for src in unique_sources: st.markdown(f"- *{src}*")
-                            
                 except Exception as e:
-                    st.error(f"Error: {e}")
+                    st.error(f"Something went wrong: {e}")
 
-    # === SEARCH MODE (The Table) ===
-    else:
-        st.markdown(f"### 🔎 Results for: *'{prompt}'*")
-        
-        with st.spinner("Scanning archives..."):
-            # UNPACKING THE TUPLE (This fixes your error)
-            docs, debug_log = search_archives(prompt)
+# --- MODE B: SEARCH (EPHEMERAL) ---
+else:
+    st.markdown("<h1 style='text-align: center; color: #2F4F4F;'>🔍 SERMON SEARCH</h1>", unsafe_allow_html=True)
+    st.caption("Search for exact quotes without the AI Persona.")
+    
+    search_query = st.chat_input("Enter a keyword or phrase...")
+    
+    if search_query:
+        st.markdown(f"### Results for: *'{search_query}'*")
+        with st.spinner("Scanning tapes..."):
+            docs = search_archives(search_query)
             
-            # 1. SHOW DIAGNOSTICS
-            with st.expander("🛠️ System Diagnostics", expanded=False):
-                if debug_log:
-                    for line in debug_log:
-                        st.write(line)
-                else:
-                    st.write("No diagnostic info available.")
-
-            # 2. SHOW RESULTS
             if not docs:
                 st.warning("No results found.")
             
             for doc in docs:
-                # Safe Metadata Access
-                src = doc.metadata.get('source', 'Tape')
-                para = doc.metadata.get('paragraph', '')
-                txt = doc.page_content
+                source = doc.metadata.get('source', 'Unknown')
+                text = doc.page_content
                 
+                # Render as a nice card
                 st.markdown(f"""
-                <div class="quote-card">
-                    <div class="quote-meta">📼 {src} {para}</div>
-                    <div class="quote-text">"{txt}"</div>
+                <div class="source-box">
+                    <div class="source-title">📼 {source}</div>
+                    <div class="source-text">"...{text}..."</div>
                 </div>
                 """, unsafe_allow_html=True)