Update src/app.py

src/app.py

@@ -4,11 +4,7 @@ import sys
 import streamlit as st
 from dotenv import load_dotenv
 
-# --- 1. CONFIGURATION ---
-INDEX_NAME = "branham-index"
-CHUNKS_FILE = "sermon_chunks.pkl"
-
-# --- 2. IMPORTS ---
+# --- 1. IMPORTS ---
 from langchain_google_genai import ChatGoogleGenerativeAI, GoogleGenerativeAIEmbeddings
 from langchain_community.retrievers import BM25Retriever
 from langchain_pinecone import PineconeVectorStore
@@ -18,19 +14,35 @@ from langchain.retrievers import EnsembleRetriever
 
 load_dotenv()
 
+# --- 2. CONFIGURATION ---
+INDEX_NAME = "branham-index"
+CHUNKS_FILE = "sermon_chunks.pkl"
+
 def get_rag_chain():
-    # A. Auth (Check Secrets first, then local .env)
-    pinecone_key = st.secrets.get("PINECONE_API_KEY") or os.getenv("PINECONE_API_KEY")
-    google_key = st.secrets.get("GOOGLE_API_KEY") or os.getenv("GOOGLE_API_KEY")
+    # --- A. ROBUST AUTHENTICATION ---
+    # 1. Try Hugging Face Environment Variables FIRST (This is where HF stores Settings)
+    pinecone_key = os.environ.get("PINECONE_API_KEY")
+    google_key = os.environ.get("GOOGLE_API_KEY")
 
+    # 2. If not found in Env, try Streamlit Secrets (Local fallback)
+    if not pinecone_key or not google_key:
+        try:
+            if not pinecone_key:
+                pinecone_key = st.secrets.get("PINECONE_API_KEY")
+            if not google_key:
+                google_key = st.secrets.get("GOOGLE_API_KEY")
+        except Exception:
+            pass
+
+    # 3. Final Check
     if not pinecone_key or not google_key:
-        raise ValueError("❌ Missing API Keys. Set PINECONE_API_KEY and GOOGLE_API_KEY.")
+        raise ValueError("❌ Missing API Keys. Please add PINECONE_API_KEY and GOOGLE_API_KEY to Settings > Secrets.")
 
-    # Set environment variables for the libraries to see
+    # Set environment variables for the libraries to use
     os.environ["PINECONE_API_KEY"] = pinecone_key
     os.environ["GOOGLE_API_KEY"] = google_key
 
-    # B. Pinecone Vector Search (Cloud)
+    # --- B. CLOUD VECTOR SEARCH (Pinecone) ---
     print("🔌 Connecting to Pinecone...")
     embeddings = GoogleGenerativeAIEmbeddings(model="models/text-embedding-004")
     
@@ -40,7 +52,7 @@ def get_rag_chain():
     )
     vector_retriever = vector_store.as_retriever(search_kwargs={"k": 5})
 
-    # C. Local Keyword Search (File)
+    # --- C. LOCAL KEYWORD SEARCH (Direct .pkl Load) ---
     print("🔌 Loading Keyword Search...")
     keyword_retriever = None
     
@@ -55,7 +67,7 @@ def get_rag_chain():
     except Exception as e:
         print(f"❌ Failed to load keyword file: {e}")
 
-    # D. Hybrid Merge
+    # --- D. HYBRID MERGE ---
     if keyword_retriever:
         print("🔗 Linking Hybrid System...")
         final_retriever = EnsembleRetriever(
@@ -65,14 +77,14 @@ def get_rag_chain():
     else:
         final_retriever = vector_retriever
 
-    # E. Model
+    # --- E. MODEL ---
     llm = ChatGoogleGenerativeAI(
         model="gemini-1.5-flash",
         temperature=0.3,
         convert_system_message_to_human=True
     )
 
-    # F. Prompt
+    # --- F. PROMPT ---
     template = """You are William Marion Branham.
 
 INSTRUCTIONS: