Update app.py

app.py

@@ -88,7 +88,15 @@ def chat(message, history):
     history.append([message, full])
     return history, ""
 #============starting extract_docx_text
-
+def respond(message, history):
+    answer = qa_chain.invoke(message)
+    docs = retriever.invoke(message)  # Global or pass as state
+    refs = [f"Page {d.metadata.get('page', 'N/A')}" for d in docs]
+    full_answer = f"{answer}\n\n**Refs:** {' | '.join(refs)}"
+    
+    history.append([message, full_answer])
+    return history, ""
+#====================
 def extract_docx_text(file_path):
     doc = Document(file_path)
     final_data = []
@@ -318,12 +326,12 @@ with gr.Blocks(css=css) as demo:
                             🚧 Coming Soon 🚧
                         </div>"""
                    )
-        with gr.TabItem("ChatBox-NRL maual-goods"):
-            gr.ChatInterface(
-                                chat,
-                                title="RAG Chatbot - NRL procurement manual of Goods",
-                                description="Ask questions!",
-                                examples=["what is GeM", "What is limited tender"],
-                                cache_examples=False  # Disable for dynamic retriever
-                            )
+        with gr.TabItem("ChatBot-NRL manual-goods"):
+            gr.Markdown("# RAG Chatbot - procurement manual of Goods")
+            chatbot = gr.Chatbot(height=500, type="tuples")  # Explicit tuples format
+            msg = gr.Textbox(placeholder="Ask about GeM...", scale=4)
+            clear = gr.Button("Clear")
+            
+            msg.submit(respond, [msg, chatbot], [chatbot, msg])
+            clear.click(lambda: ([], ""), None, chatbot)
 demo.queue().launch()