Update app.py

app.py

@@ -38,6 +38,7 @@ def get_default_config():
             "index_directory": "./index",
         },
         "models": {
+            # You can also use "all-MiniLM-L6-v2" here, but this path works well on HF
             "embedding": "sentence-transformers/all-MiniLM-L6-v2",
             "qa": "deepset/roberta-base-squad2",
         },
@@ -402,13 +403,32 @@ print("=" * 50)
 # -----------------------------
 
 def rag_respond(message, history):
-    """Handle chat messages for chatbot UI"""
+    """Handle chat messages for chatbot UI (messages format)"""
+    if history is None:
+        history = []
+
     if not message or not str(message).strip():
+        # Keep history unchanged, just clear input
         return "", history
 
     user_msg = str(message)
+
+    # Append user message
+    history.append({
+        "role": "user",
+        "content": user_msg,
+    })
+
+    # Get bot reply
     bot_reply = rag_index.answer(user_msg)
-    history = history + [[user_msg, bot_reply]]
+
+    # Append assistant message
+    history.append({
+        "role": "assistant",
+        "content": bot_reply,
+    })
+
+    # Clear textbox, return updated history
     return "", history
 
 
@@ -425,7 +445,6 @@ def upload_to_kb(files):
 
     for f in files:
         # Gradio File object or temp file path
-        # In older Gradio, `f.name` is the temp file path
         src_path = getattr(f, "name", None) or str(f)
         if not os.path.exists(src_path):
             continue
@@ -463,7 +482,7 @@ def rebuild_index():
     )
 
 
-# Description + examples
+# Description + (optional) examples
 description = WELCOME_MSG
 if not rag_index.initialized or rag_index.index is None or not rag_index.chunks:
     description += (
@@ -489,39 +508,20 @@ with gr.Blocks(title=CONFIG["client"]["name"]) as demo:
     gr.Markdown(description)
 
     with gr.Tab("Chat"):
-        chatbot = gr.Chatbot(label="RAG Chat")
+        chatbot = gr.Chatbot(label="RAG Chat")  # messages-format by default
+
         with gr.Row():
             txt = gr.Textbox(
                 show_label=False,
-                placeholder="Ask a question about your documents...",
+                placeholder="Ask a question about your documents and press Enter to send...",
                 lines=2,
             )
+
         with gr.Row():
             send_btn = gr.Button("Send")
             clear_btn = gr.Button("Clear")
 
-        # Pre-fill example buttons if available
-        if examples:
-            gr.Markdown("### Example questions")
-            example_btns = []
-            with gr.Row():
-                for ex in examples:
-                    example_btns.append(gr.Button(ex))
-
-            def use_example(example, history):
-                """When clicking an example, send it as a message"""
-                bot_reply = rag_index.answer(example)
-                history = history + [[example, bot_reply]]
-                return history
-
-            for btn, ex in zip(example_btns, examples):
-                btn.click(
-                    use_example,
-                    inputs=[gr.State(ex), chatbot],
-                    outputs=chatbot,
-                )
-
-        # Chat logic wiring
+        # Enter submits, Send button also submits
         txt.submit(rag_respond, [txt, chatbot], [txt, chatbot])
         send_btn.click(rag_respond, [txt, chatbot], [txt, chatbot])
         clear_btn.click(lambda: ([], ""), None, [chatbot, txt])