Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,205 +2,210 @@ import streamlit as st
 import time
 import os
 
-# --- 1. PAGE CONFIGURATION (MUST BE FIRST) ---
+# --- 1. PAGE CONFIGURATION ---
 st.set_page_config(
-    page_title="Voice of the Sign", 
+    page_title="The Voice of the Sign", 
     page_icon="🦅", 
-    layout="centered", 
+    layout="wide", 
     initial_sidebar_state="expanded"
 )
 
-# --- 2. IMPORTS & ERROR HANDLING ---
+# --- 2. BACKEND LOADING ---
+# We try to import the brain. If it fails, we show a pretty error.
+backend_loaded = False
 try:
-    from app import get_rag_chain, search_archives, build_index_on_server, check_files, raw_text_search
+    from app import get_rag_chain, search_archives
+    backend_loaded = True
 except ImportError as e:
-    st.error(f"❌ CRITICAL ERROR: Could not load the Brain.\n\nDetails: {e}")
-    st.stop()
+    st.error(f"❌ System Error: Could not load the Brain.\n\nDetails: {e}")
 
-# --- 3. PROFESSIONAL "EAGLE & SCROLL" THEME ---
+# --- 3. CUSTOM CSS (The "Appealing" Look) ---
 st.markdown("""
 <style>
-    /* 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;
+    /* IMPORT FONTS */
+    @import url('https://fonts.googleapis.com/css2?family=Merriweather:ital,wght@0,300;0,400;0,700;1,300&family=Inter:wght@300;400;600&display=swap');
+    
+    /* GLOBAL THEME */
+    .stApp {
+        background-color: #FDFBF7; /* Parchment White */
     }
-
-    /* HEADERS */
+    
     h1, h2, h3 {
-        font-family: 'Merriweather', serif;
+        font-family: 'Merriweather', serif !important;
         color: #2C3E50;
     }
+    
+    p, div, label {
+        font-family: 'Inter', sans-serif;
+        color: #34495E;
+    }
 
     /* CHAT BUBBLES - USER */
-    .stChatMessage[data-testid="stChatMessage"]:nth-child(odd) {
-        background-color: #E8F4F8; /* Soft Blue-Grey */
-        border-left: 4px solid #3498DB;
-        color: #2C3E50 !important;
+    div[data-testid="stChatMessage"][data-test-role="user"] {
+        background-color: #EAECEE;
+        border-radius: 20px 20px 5px 20px;
+        padding: 15px;
+        box-shadow: 0 2px 5px rgba(0,0,0,0.05);
     }
 
-    /* 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;
+    /* CHAT BUBBLES - ASSISTANT */
+    div[data-testid="stChatMessage"][data-test-role="assistant"] {
+        background-color: #FFFFFF;
+        border: 1px solid #D4AF37; /* Gold Border */
+        border-radius: 20px 20px 20px 5px;
+        padding: 20px;
+        box-shadow: 0 4px 10px rgba(0,0,0,0.05);
     }
 
-    /* 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;
+    /* AVATARS */
+    .stChatMessage .stChatMessageAvatar {
+        background-color: #2C3E50 !important; /* Navy Blue */
+        color: white !important;
     }
 
-    /* CITATION BOXES */
-    .source-box {
+    /* QUOTE CARDS (SEARCH MODE) */
+    .quote-card {
         background-color: #FFFFFF;
-        border: 1px solid #D1D1D1;
+        border-left: 6px solid #D4AF37; /* Gold Accent */
+        padding: 20px;
+        margin-bottom: 20px;
         border-radius: 8px;
-        padding: 12px;
-        margin-top: 10px;
-        font-family: 'Merriweather', serif;
-        font-size: 0.9em;
-        box-shadow: 0 2px 4px rgba(0,0,0,0.05);
+        box-shadow: 0 4px 6px rgba(0,0,0,0.05);
+        transition: transform 0.2s;
+    }
+    .quote-card:hover {
+        transform: translateY(-2px);
+        box-shadow: 0 8px 15px rgba(0,0,0,0.1);
     }
-    .source-title {
-        color: #8B5E3C;
-        font-weight: bold;
-        border-bottom: 1px solid #EEE;
-        padding-bottom: 5px;
-        margin-bottom: 5px;
+    .quote-meta {
+        font-family: 'Inter', sans-serif;
+        font-size: 0.85rem;
+        color: #7F8C8D;
+        text-transform: uppercase;
+        letter-spacing: 1px;
+        margin-bottom: 8px;
+        font-weight: 600;
     }
-    .source-text {
-        font-style: italic;
-        color: #555;
+    .quote-text {
+        font-family: 'Merriweather', serif;
+        font-size: 1.05rem;
+        line-height: 1.6;
+        color: #2C3E50;
     }
 
     /* SIDEBAR STYLING */
     section[data-testid="stSidebar"] {
-        background-color: #F8F9FA;
+        background-color: #F4F6F7;
+        border-right: 1px solid #E5E7E9;
     }
     
-    /* REMOVE DEFAULT BRANDING */
-    #MainMenu {visibility: hidden;}
-    footer {visibility: hidden;}
-    
+    /* INPUT BOX */
+    .stChatInput textarea {
+        border-radius: 15px !important;
+        border: 1px solid #BDC3C7 !important;
+    }
 </style>
 """, unsafe_allow_html=True)
 
-# --- 4. SIDEBAR DASHBOARD ---
+# --- 4. SESSION STATE INITIALIZATION (Fixes the "Clearing" Issue) ---
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# --- 5. SIDEBAR CONTROLS ---
 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("---")
+    st.markdown("### 🦅 Controls")
     
-    # Mode Selector
     mode = st.radio(
-        "Select Operation Mode:", 
-        ["🗣️ Chat with Brother Branham", "🔍 Quote Search Only"],
-        captions=["AI Persona (1965 Era)", "Exact Text Search"]
+        "Select Mode:", 
+        ["🗣️ Chat with Bro. Branham", "🔍 Search Quotes Only"], 
+        captions=["Ask theological questions.", "Find exact paragraphs."],
+        index=0
     )
     
     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):
+    # CLEAR BUTTON (Resets Memory)
+    if st.button("🗑️ Clear Conversation", use_container_width=True):
         st.session_state.messages = []
         st.rerun()
-
-# --- 5. MAIN CONTENT AREA ---
-
-# Initialize History
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
-# --- 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()
-
-    # 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?")
-
-    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}**")
-
-    # 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})
+    st.markdown("---")
+    st.info("💡 **Tip:** Ask specific questions like *'What is the Third Pull?'* or search names like *'Coleman'*.")
+
+# --- 6. MAIN HEADER ---
+col1, col2 = st.columns([1, 6])
+with col1:
+    st.markdown("# 🦅") # Simple Icon
+with col2:
+    if mode.startswith("🗣️"):
+        st.markdown("# The 7th Handle\n*Interactive AI Archive*")
+    else:
+        st.markdown("# The Table\n*Direct Quote Search*")
+
+st.divider()
+
+if not backend_loaded: st.stop()
+
+# --- 7. LOAD SYSTEM (Cached) ---
+@st.cache_resource(show_spinner=False)
+def load_system():
+    return get_rag_chain()
+
+# --- 8. DISPLAY CHAT HISTORY (Persistent) ---
+# This block ensures old messages stay on screen
+for message in st.session_state.messages:
+    # We differentiate between "Search Results" and "Chat Messages" visually
+    if message["role"] == "user":
         with st.chat_message("user", avatar="👤"):
-            st.markdown(prompt)
-
-        # B. Generate & Append Assistant Message
-        with st.chat_message("assistant", avatar="📖"):
-            with st.spinner("Searching the storehouse..."):
+            st.markdown(message["content"])
+    elif message["role"] == "assistant":
+        with st.chat_message("assistant", avatar="🦅"):
+            st.markdown(message["content"])
+            # If sources exist, show them in a dropdown
+            if "sources" in message and message["sources"]:
+                with st.expander("📚 View Sermon References"):
+                    for src in message["sources"]:
+                        st.markdown(f"- *{src}*")
+
+# --- 9. INPUT HANDLER ---
+prompt = st.chat_input("Ask a question or search...")
+
+if prompt:
+    # 9a. SHOW USER MESSAGE IMMEDIATELY
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message("user", avatar="👤"):
+        st.markdown(prompt)
+
+    # 9b. PROCESS BASED ON MODE
+    
+    # === CHAT MODE ===
+    if mode.startswith("🗣️"):
+        with st.chat_message("assistant", avatar="🦅"):
+            message_placeholder = st.empty()
+            full_response = ""
+            
+            with st.spinner("Searching the archives..."):
                 try:
-                    # Run the Brain
-                    chain = get_rag_chain()
-                    if not chain:
-                        st.error("⚠️ System Index is missing. Please Rebuild Index in Sidebar.")
-                        st.stop()
-                        
+                    chain = load_system()
                     response = chain.invoke({"question": prompt})
-                    answer_text = response['result']
                     
-                    # 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 answer_text.split():
+                    result_text = response['result']
+                    source_docs = response.get('source_documents', [])
+                    unique_sources = list(set([doc.metadata.get('source', 'Unknown Tape') for doc in source_docs]))
+                    
+                    # TYPEWRITER EFFECT
+                    for chunk in result_text.split():
                         full_response += chunk + " "
-                        time.sleep(0.04)
+                        time.sleep(0.02) # Subtle typing speed
                         message_placeholder.markdown(full_response + "▌")
                     message_placeholder.markdown(full_response)
-
-                    # Show Sources
+                    
+                    # 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!)
+                        with st.expander("📚 Sermon References"):
+                            for src in unique_sources: st.markdown(f"- *{src}*")
+                    
+                    # SAVE TO HISTORY
                     st.session_state.messages.append({
                         "role": "assistant", 
                         "content": full_response,
@@ -208,31 +213,33 @@ if mode == "🗣️ Chat with Brother Branham":
                     })
                     
                 except Exception as e:
-                    st.error(f"Something went wrong: {e}")
+                    st.error(f"Error: {e}")
 
-# --- 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)
+    # === SEARCH MODE ===
+    else:
+        # For Search Mode, we don't save to chat history the same way.
+        # We just display the results clearly.
+        st.markdown(f"### 🔎 Results for: *'{prompt}'*")
+        
+        with st.spinner("Scanning local and cloud libraries..."):
+            docs, debug_log = search_archives(prompt)
             
-            if not docs:
-                st.warning("No results found.")
+            # DIAGNOSTICS (Collapsible)
+            with st.expander("🛠️ View System Diagnostics", expanded=False):
+                for line in debug_log: st.write(line)
             
-            for doc in docs:
-                source = doc.metadata.get('source', 'Unknown')
-                text = doc.page_content
-                
-                # Render as a nice card
-                st.markdown(f"""
-                <div class="source-box">
-                    <div class="source-title">📼 {source}</div>
-                    <div class="source-text">"...{text}..."</div>
-                </div>
-                """, unsafe_allow_html=True)
+            if not docs:
+                st.warning("No relevant records found.")
+            else:
+                for doc in docs:
+                    src = doc.metadata.get('source', 'Tape')
+                    para = doc.metadata.get('paragraph', '')
+                    # Highlight the keyword
+                    txt = doc.page_content.replace(prompt, f"**{prompt}**") 
+                    
+                    st.markdown(f"""
+                    <div class="quote-card">
+                        <div class="quote-meta">📼 {src} {para}</div>
+                        <div class="quote-text">"{txt}"</div>
+                    </div>
+                    """, unsafe_allow_html=True)