Update app.py

app.py

@@ -26,7 +26,8 @@ def load_embedding_model(model_name: str = DEFAULT_EMBED_MODEL):
 
 def save_uploaded_file(uploaded_file) -> str:
     tmpdir = tempfile.gettempdir()
-    temp_path = os.path.join(tmpdir, uploaded_file.name)
+    safe_name = os.path.basename(uploaded_file.name)
+    temp_path = os.path.join(tmpdir, safe_name)
     with open(temp_path, "wb") as f:
         f.write(uploaded_file.getbuffer())
     return temp_path
@@ -112,8 +113,8 @@ def load_index_and_metadata(index_path: str = INDEX_PATH, meta_path: str = METAD
     return None, None
 
 
-def get_groq_client_from_env(hardcoded_key: str = None) -> Groq | None:
-    key = os.environ.get("GROQ_API_KEY") or (hardcoded_key or None)
+def get_groq_client_from_env() -> Groq | None:
+    key = os.environ.get("GROQ_API_KEY") or os.environ.get("HF_GROQ_API_KEY") or os.environ.get("HF_API_TOKEN")
     if not key:
         return None
     return Groq(api_key=key)
@@ -142,13 +143,16 @@ def main():
     top_k = int(st.sidebar.number_input("Top-k results to retrieve", min_value=1, max_value=50, value=DEFAULT_TOP_K))
 
     st.sidebar.markdown("---")
-    st.sidebar.write("Groq API key (optional - recommended to use env var)")
-    hardcoded_key = st.sidebar.text_input("Paste GROQ key here (will be used if env var not set)", type="password", value="")
+    st.sidebar.write("Groq API key is loaded from Hugging Face secrets or environment variables. Do not paste keys in the UI.")
 
     with st.spinner("Loading embedding model..."):
         embed_model = load_embedding_model()
 
     index, metadata = load_index_and_metadata()
+    if index is None and "index" in st.session_state:
+        index = st.session_state["index"]
+        metadata = st.session_state.get("metadata")
+
     if index is None:
         st.info("No existing FAISS index found. Upload documents and click 'Ingest documents' to build index.")
     else:
@@ -200,8 +204,17 @@ def main():
                     embeddings = normalize_embeddings(embeddings)
 
                 index = build_faiss_index(embeddings)
-                save_index_and_metadata(index, metadata)
-                st.success(f"Index built and saved. {len(all_chunks)} chunks indexed.")
+                st.session_state["index"] = index
+                st.session_state["metadata"] = metadata
+                try:
+                    save_index_and_metadata(index, metadata)
+                    st.success(f"Index built and saved. {len(all_chunks)} chunks indexed.")
+                except Exception:
+                    st.success(f"Index built in memory. {len(all_chunks)} chunks indexed.")
+                st.info(f"Ingested {len(all_chunks)} chunks from {len(set(m['source'] for m in metadata))} file(s).")
+                if len(all_chunks) > 0:
+                    st.markdown("### Example chunk")
+                    st.write(all_chunks[0][:1000])
 
     st.markdown("---")
 
@@ -214,9 +227,11 @@ def main():
             if not query or not query.strip():
                 st.warning("Please enter a question.")
             else:
-                if index is None or metadata is None:
+                if "index" not in st.session_state or "metadata" not in st.session_state:
                     st.warning("No index available. Please ingest documents first.")
                 else:
+                    index = st.session_state["index"]
+                    metadata = st.session_state["metadata"]
                     q_emb = embed_model.encode([query], convert_to_numpy=True)
                     q_emb = normalize_embeddings(q_emb)
                     q_emb = np.ascontiguousarray(q_emb.astype('float32'))
@@ -234,9 +249,9 @@ def main():
                             meta = metadata[idx]
                             contexts.append(meta)
 
-                        client = get_groq_client_from_env(hardcoded_key if hardcoded_key else None)
+                        client = get_groq_client_from_env()
                         if client is None:
-                            st.error("Groq client could not be created. Set groq API key in env or sidebar.")
+                            st.error("Groq client could not be created. Set GROQ_API_KEY in your Hugging Face secrets or environment.")
                         else:
                             system_msg = {
                                 "role": "system",
@@ -273,9 +288,9 @@ def main():
 
     st.sidebar.markdown("---")
     st.sidebar.header("Index status")
-    if index is not None:
+    if "index" in st.session_state:
         try:
-            st.sidebar.write(f"Vectors in index: {index.ntotal}")
+            st.sidebar.write(f"Vectors in index: {st.session_state['index'].ntotal}")
         except Exception:
             st.sidebar.write("Vectors in index: unknown")
     else: