Update app.py

app.py

@@ -26,11 +26,13 @@ db = FAISS.from_documents(texts, embeddings)
 # --- 4️⃣ Build retriever-based chatbot ---
 retriever = db.as_retriever(search_kwargs={"k": 3})
 
-# --- Hugging Face Hub LLM setup with secret token ---
+# --- 5️⃣ Hugging Face LLM setup using secret ---
+hf_token = os.environ["HUGGINGFACEHUB_API_TOKEN"]  # Must be set in Space Secrets
+
 llm = HuggingFaceHub(
-    repo_id="google/flan-t5-base",   # smaller for faster response
+    repo_id="google/flan-t5-base",
     model_kwargs={"temperature": 0},
-    huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN")
+    huggingfacehub_api_token=hf_token
 )
 
 qa = ConversationalRetrievalChain.from_llm(
@@ -38,15 +40,17 @@ qa = ConversationalRetrievalChain.from_llm(
     retriever=retriever
 )
 
-# --- 5️⃣ Respond function ---
-def respond(message, chat_history):
-    chat_history = chat_history or []
-    chat_history = chat_history[-6:]  # keep last 3 exchanges
-    result = qa({"question": message, "chat_history": chat_history})
-    chat_history.append((message, result["answer"]))
-    return chat_history, chat_history
+# --- 6️⃣ Chat history ---
+chat_history = []
+
+# --- 7️⃣ Respond function ---
+def respond(message, history):
+    history = history[-6:]  # Keep last 3 exchanges
+    result = qa({"question": message, "chat_history": history})
+    history.append((message, result["answer"]))
+    return history, history
 
-# --- 6️⃣ Gradio UI with Entry Warning ---
+# --- 8️⃣ Gradio UI with Entry Warning ---
 with gr.Blocks() as demo:
     with gr.Column():
         # Warning message
@@ -92,6 +96,7 @@ with gr.Blocks() as demo:
             gr.update(visible=False)   # exit_btn
         )
 
+    # Connect buttons
     enter_btn.click(
         enter_case,
         inputs=None,
@@ -110,7 +115,7 @@ with gr.Blocks() as demo:
         outputs=[chatbot, chatbot]
     )
 
-# --- 7️⃣ Launch ---
+# --- 9️⃣ Launch ---
 if __name__ == "__main__":
     demo.launch(share=True, enable_queue=True)