Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,233 +1,250 @@
 import streamlit as st
 import time
-import os
 
-# --- 1. PAGE CONFIGURATION ---
+# ==============================
+# PAGE CONFIG
+# ==============================
 st.set_page_config(
-    page_title="Voice of the Sign", 
-    page_icon="🦅", 
-    layout="wide", 
+    page_title="Voice of the Sign",
+    page_icon="🦅",
+    layout="wide",
     initial_sidebar_state="expanded"
 )
 
-# --- 2. BACKEND LOADING ---
+# ==============================
+# LOAD BACKEND
+# ==============================
 backend_loaded = False
 try:
     from app import get_rag_chain, search_archives
     backend_loaded = True
-except ImportError as e:
-    st.error(f"❌ System Error: Could not load the Brain.\n\nDetails: {e}")
+except Exception as e:
+    st.error(f"❌ Backend failed to load:\n\n{e}")
 
-# --- 3. HELPER: SMART LINK GENERATOR ---
-def generate_sermon_link(filename):
+# ==============================
+# MESSAGEHUB LINK BUILDER
+# ==============================
+def messagehub_link(filename: str):
     """
-    Converts '65_1128M_Gods_Only_Provided_Place.PDF' 
-    into 'https://churchages.net/en/sermon/branham/65-1128m-gods-only-provided-place'
+    Example:
+    62-0909E In His Presence.pdf
+    → https://www.messagehub.info/en/read.do?ref_num=62-0909E
     """
-    if not filename: return "#"
-    
-    try:
-        # 1. Remove file extension
-        clean_name = filename.replace(".PDF", "").replace(".pdf", "")
-        
-        # 2. Replace underscores with dashes (ChurchAges format)
-        clean_name = clean_name.replace("_", "-")
-        
-        # 3. Lowercase everything
-        slug = clean_name.lower()
-        
-        # 4. Build URL
-        return f"https://churchages.net/en/sermon/branham/{slug}"
-    except:
+    if not filename:
         return "#"
 
-# --- 4. ADAPTIVE CSS ---
+    name = filename.replace(".pdf", "").replace(".PDF", "").strip()
+    code = name.split()[0]  # first token is sermon code
+    return f"https://www.messagehub.info/en/read.do?ref_num={code}"
+
+
+# ==============================
+# STYLING
+# ==============================
 st.markdown("""
 <style>
-    /* FONTS & BASICS */
-    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&family=Playfair+Display:ital,wght@0,600;1,600&display=swap');
-    html, body, [class*="css"] { font-family: 'Inter', sans-serif; }
-    h1, h2, h3 { font-family: 'Playfair Display', serif !important; }
-
-    /* CHAT BUBBLES */
-    div[data-testid="stChatMessage"][data-test-role="user"] {
-        background-color: rgba(128, 128, 128, 0.1);
-        border-radius: 20px 20px 5px 20px;
-    }
-    div[data-testid="stChatMessage"][data-test-role="assistant"] {
-        background-color: rgba(212, 175, 55, 0.05);
-        border-radius: 20px 20px 20px 5px;
-        border-left: 4px solid #D4AF37;
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600&family=Playfair+Display:wght@600&display=swap');
+
+html, body, [class*="css"] {
+    font-family: 'Inter', sans-serif;
+}
+
+h1, h2, h3 {
+    font-family: 'Playfair Display', serif;
+}
+
+div[data-testid="stChatMessage"][data-test-role="user"] {
+    background-color: rgba(128,128,128,0.08);
+    border-radius: 20px 20px 5px 20px;
+}
+
+div[data-testid="stChatMessage"][data-test-role="assistant"] {
+    background-color: rgba(212,175,55,0.05);
+    border-left: 4px solid #D4AF37;
+    border-radius: 20px 20px 20px 5px;
+}
+
+/* Improve markdown spacing */
+.markdown-text-container p {
+    margin-bottom: 0.9em;
+    line-height: 1.7;
+}
+
+.markdown-text-container ul {
+    margin-left: 1.2em;
+}
+
+.markdown-text-container h3 {
+    margin-top: 1.2em;
+    color: #D4AF37;
+}
+
+/* Reference cards */
+.quote-card {
+    padding: 18px;
+    margin-bottom: 14px;
+    border-radius: 12px;
+    border-left: 5px solid #D4AF37;
+}
+
+.quote-meta {
+    font-weight: 600;
+    margin-bottom: 6px;
+}
+
+.quote-meta a {
+    color: #D4AF37;
+    text-decoration: none;
+}
+
+.quote-text {
+    font-family: 'Playfair Display', serif;
+    font-style: italic;
+    line-height: 1.6;
+}
+
+@media (prefers-color-scheme: dark) {
+    .quote-card {
+        background-color: rgba(255,255,255,0.05);
+        border: 1px solid rgba(255,255,255,0.08);
     }
+}
 
-    /* QUOTE CARDS (SEARCH) */
+@media (prefers-color-scheme: light) {
     .quote-card {
-        padding: 20px;
-        margin-bottom: 15px;
-        border-radius: 12px;
-        border-left: 5px solid #D4AF37;
-        transition: transform 0.2s;
-    }
-    .quote-meta {
-        font-size: 0.9rem;
-        margin-bottom: 8px;
-        font-weight: 600;
-    }
-    .quote-meta a {
-        text-decoration: none;
-        color: #D4AF37 !important; /* Gold Link */
-    }
-    .quote-text {
-        font-family: 'Playfair Display', serif;
-        font-size: 1.1rem;
-        line-height: 1.6;
-        font-style: italic;
+        background: #ffffff;
+        border: 1px solid #e6e6e6;
+        box-shadow: 0 4px 6px rgba(0,0,0,0.05);
     }
+}
 
-    /* DARK/LIGHT ADAPTATION */
-    @media (prefers-color-scheme: dark) {
-        .quote-card { background-color: rgba(255, 255, 255, 0.05); border: 1px solid rgba(255, 255, 255, 0.1); }
-        .quote-text { color: #E0E0E0; }
-    }
-    @media (prefers-color-scheme: light) {
-        .quote-card { background-color: #FFFFFF; border: 1px solid #E0E0E0; box-shadow: 0 4px 6px rgba(0,0,0,0.05); }
-        .quote-text { color: #2C3E50; }
-    }
-    
-    header {visibility: hidden;}
-    #MainMenu {visibility: hidden;}
+header, #MainMenu { visibility: hidden; }
 </style>
 """, unsafe_allow_html=True)
 
-# --- 5. SESSION STATE ---
+
+# ==============================
+# SESSION STATE
+# ==============================
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
-# --- 6. SIDEBAR ---
+
+# ==============================
+# SIDEBAR
+# ==============================
 with st.sidebar:
     st.title("🦅 Controls")
-    mode = st.radio("Mode:", ["🗣️ Chat", "🔍 Search"], index=0, label_visibility="collapsed")
+    mode = st.radio("Mode", ["🗣️ Chat", "🔍 Search"], index=0, label_visibility="collapsed")
     st.divider()
+
     if st.button("🗑️ Clear Screen", use_container_width=True):
-        if mode.startswith("🗣️"):
-            st.session_state.chat_history = []
+        st.session_state.chat_history = []
         st.rerun()
 
-# --- 7. HEADER ---
-col1, col2 = st.columns([1, 15])
-with col1: st.markdown("# 🦅")
+
+# ==============================
+# HEADER
+# ==============================
+col1, col2 = st.columns([1, 14])
+with col1:
+    st.markdown("# 🦅")
 with col2:
-    if mode.startswith("🗣️"):
-        st.markdown("# The 7th Handle")
-    else:
-        st.markdown("# The Table")
+    st.markdown("# The 7th Handle" if mode.startswith("🗣️") else "# The Table")
+
 st.divider()
 
-if not backend_loaded: st.stop()
+if not backend_loaded:
+    st.stop()
+
 
-# --- 8. LOAD SYSTEM ---
+# ==============================
+# LOAD RAG SYSTEM
+# ==============================
 @st.cache_resource(show_spinner=False)
-def load_system():
+def load_chain():
     return get_rag_chain()
 
-# --- 9. MAIN LOGIC ---
 
-# === A. CHAT MODE ===
+# =========================================================
+# CHAT MODE
+# =========================================================
 if mode.startswith("🗣️"):
-    # Display History
-    for message in st.session_state.chat_history:
-        role = message["role"]
-        with st.chat_message(role, avatar="👤" if role == "user" else "🦅"):
-            st.markdown(message["content"])
-            if "sources" in message and message["sources"]:
+
+    # --- Render chat history ---
+    for msg in st.session_state.chat_history:
+        with st.chat_message(msg["role"], avatar="👤" if msg["role"] == "user" else "🦅"):
+            st.markdown(msg["content"], unsafe_allow_html=False)
+
+            if msg.get("sources"):
                 with st.expander("📚 References"):
-                    for doc in message["sources"]:
-                        # Handle both string filenames and document objects
-                        if hasattr(doc, 'metadata'):
-                            fname = doc.metadata.get('source', 'Unknown')
-                            para = doc.metadata.get('paragraph', '')
-                        else:
-                            fname = str(doc)
-                            para = ""
-                        
-                        link = generate_sermon_link(fname)
-                        st.markdown(f"🔗 [{fname} (Para {para})]({link})")
-
-    # Handle Input
-    if prompt := st.chat_input("Ask a question..."):
-        st.session_state.chat_history.append({"role": "user", "content": prompt})
-        with st.chat_message("user", avatar="👤"):
-            st.markdown(prompt)
-
-        with st.chat_message("assistant", avatar="🦅"):
-            placeholder = st.empty()
-            full_response = ""
-            with st.spinner("Thinking..."):
-                try:
-                    chain = load_system()
-                    response = chain.invoke({"question": prompt})
-                    result_text = response['result']
-                    
-                    for chunk in result_text.split():
-                        full_response += chunk + " "
-                        time.sleep(0.02)
-                        placeholder.markdown(full_response + "▌")
-                    placeholder.markdown(full_response)
-                    
-                    # Process Sources
-                    source_docs = response.get('source_documents', [])
-                    unique_docs = {}
-                    for doc in source_docs:
-                        src = doc.metadata.get('source', 'Unknown')
-                        if src not in unique_docs:
-                            unique_docs[src] = doc 
-                    
-                    final_sources = list(unique_docs.values())
-                    
-                    if final_sources:
-                        with st.expander("📚 References"):
-                            for doc in final_sources:
-                                fname = doc.metadata.get('source', 'Unknown')
-                                para = doc.metadata.get('paragraph', '')
-                                link = generate_sermon_link(fname)
-                                st.markdown(f"🔗 [{fname} (Para {para})]({link})")
-                    
-                    st.session_state.chat_history.append({
-                        "role": "assistant", 
-                        "content": full_response, 
-                        "sources": final_sources
-                    })
-                except Exception as e:
-                    st.error(f"Error: {e}")
-
-# === B. SEARCH MODE ===
+                    for doc in msg["sources"]:
+                        src = doc.metadata.get("source", "")
+                        para = doc.metadata.get("paragraph", "")
+                        link = messagehub_link(src)
+                        st.markdown(f"🔗 [{src} (Para {para})]({link})")
+
+    # --- Input ---
+    prompt = st.chat_input("Ask a question...")
+
+    if prompt:
+        # Save user message
+        st.session_state.chat_history.append({
+            "role": "user",
+            "content": prompt
+        })
+
+        with st.spinner("Searching the tapes..."):
+            chain = load_chain()
+            response = chain.invoke({"question": prompt})
+
+        answer_text = response.get("result", "")
+        sources = response.get("source_documents", [])
+
+        # Save assistant message (FULLY formed)
+        st.session_state.chat_history.append({
+            "role": "assistant",
+            "content": answer_text,
+            "sources": sources
+        })
+
+        # Force rerender so references appear immediately
+        st.rerun()
+
+
+# =========================================================
+# SEARCH MODE
+# =========================================================
 else:
-    if prompt := st.chat_input("Search for a keyword..."):
-        st.subheader(f"Results for: '{prompt}'")
-        
+    query = st.chat_input("Search for a keyword...")
+
+    if query:
+        st.subheader(f"Results for: “{query}”")
+
         with st.spinner("Scanning archives..."):
-            docs, debug_log = search_archives(prompt)
-            
-            with st.expander("🛠️ Debug Info", expanded=False):
-                for line in debug_log: st.write(line)
-            
-            if not docs:
-                st.info("No exact records found.")
-            else:
-                for doc in docs:
-                    src = doc.metadata.get('source', 'Tape')
-                    para = doc.metadata.get('paragraph', '')
-                    txt = doc.page_content.replace(prompt, f"**{prompt}**")
-                    link = generate_sermon_link(src)
-                    
-                    # Render Card with Link
-                    st.markdown(f"""
-                    <div class="quote-card">
-                        <div class="quote-meta">
-                            <a href="{link}" target="_blank">
-                                📼 {src} (Para {para}) ↗
-                            </a>
-                        </div>
-                        <div class="quote-text">"{txt}"</div>
+            docs, debug_log = search_archives(query)
+
+        with st.expander("🛠 Debug Info"):
+            for line in debug_log:
+                st.write(line)
+
+        if not docs:
+            st.info("No exact records found.")
+        else:
+            for doc in docs:
+                src = doc.metadata.get("source", "")
+                para = doc.metadata.get("paragraph", "")
+                link = messagehub_link(src)
+
+                st.markdown(f"""
+                <div class="quote-card">
+                    <div class="quote-meta">
+                        <a href="{link}" target="_blank">
+                            📼 {src} (Para {para})
+                        </a>
+                    </div>
+                    <div class="quote-text">
+                        "{doc.page_content}"
                     </div>
-                    """, unsafe_allow_html=True)
+                </div>
+                """, unsafe_allow_html=True)