Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from answer import answer_question
 
+
 # ---------- Helper: format retrieved chunks ----------
 def format_chunks(chunks):
     if not chunks:
@@ -22,26 +23,28 @@ def format_chunks(chunks):
 
 # ---------- Chat handler (tuple-based, HF compatible) ----------
 def chat_handler(message, chat_history):
+    """
+    message: str
+    chat_history: List[Tuple[user, assistant]]
+    """
     chat_history = chat_history or []
 
-    # Build history text for RAG
+    # Build history string for RAG backend
     history_text = ""
-    for msg in chat_history:
-        history_text += f"{msg['role'].capitalize()}: {msg['content']}\n"
+    for user_msg, bot_msg in chat_history:
+        history_text += f"User: {user_msg}\nAssistant: {bot_msg}\n"
 
     # Call RAG backend
     answer, chunks = answer_question(message, history_text)
 
-    # Append messages (DICT FORMAT)
-    chat_history.append({"role": "user", "content": message})
-    chat_history.append({"role": "assistant", "content": answer})
+    # Append new turn (TUPLE FORMAT)
+    chat_history.append((message, answer))
 
     chunk_display = format_chunks(chunks)
 
     return chat_history, chunk_display, ""
 
 
-
 # ---------- Gradio UI ----------
 with gr.Blocks(title="BlinkNow – DSA RAG Assistant") as demo:
 
@@ -54,11 +57,11 @@ with gr.Blocks(title="BlinkNow – DSA RAG Assistant") as demo:
     )
 
     with gr.Row():
+        # Left: Chat
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(
                 label="Chat",
-                height=500,
-                type="messages"
+                height=500
             )
 
             with gr.Row():
@@ -69,12 +72,14 @@ with gr.Blocks(title="BlinkNow – DSA RAG Assistant") as demo:
                 )
                 send_btn = gr.Button("Send", variant="primary", scale=1)
 
+        # Right: Retrieved Context
         with gr.Column(scale=3):
             gr.Markdown("### 🔍 Retrieved Context")
             retrieved_chunks = gr.Markdown(
                 value="*Retrieved knowledge chunks will appear here after your query.*"
             )
 
+    # ---------- Events ----------
     send_btn.click(
         fn=chat_handler,
         inputs=[user_input, chatbot],
@@ -87,5 +92,6 @@ with gr.Blocks(title="BlinkNow – DSA RAG Assistant") as demo:
         outputs=[chatbot, retrieved_chunks, user_input]
     )
 
+
 if __name__ == "__main__":
     demo.launch()