Update to chat interface with conversation history

src/ui/components.py

@@ -4,15 +4,21 @@ from src.config import Settings
 from src.retrieval.search import retrieve, format_citations
 from src.llm.answer import compose_answer
 
-def _ask(query: str, settings, embedder, vstore) -> tuple[str, List[dict]]:
+def _ask(query: str, history: List, settings, embedder, vstore) -> tuple[List, List[dict]]:
+    """Process a chat message and return updated history + citations."""
     if not query or not query.strip():
-        return "Please enter a question about the documents.", []
+        history.append((query, "Please enter a question about the documents."))
+        return history, []
+    
     hits: List[Tuple[str, float]] = retrieve(query, vstore=vstore, embedder=embedder, k=settings.top_k)
     if not hits:
-        return "No relevant passages found. Try adjusting your query.", []
+        history.append((query, "No relevant passages found. Try adjusting your query."))
+        return history, []
+    
     answer = compose_answer(query, hits, settings)
     citations = format_citations(hits, max_items=settings.top_k)
-    return answer, citations
+    history.append((query, answer))
+    return history, citations
 
 def build_app(settings: Settings, embedder, vstore):
     with gr.Blocks(title=settings.title) as demo:
@@ -23,17 +29,44 @@ def build_app(settings: Settings, embedder, vstore):
                else "— retrieval + summarizer enabled.")
         )
 
-        query = gr.Textbox(label="Your question", placeholder="e.g., Orders about Sector 4 in 1963?")
-        ask_btn = gr.Button("Search", variant="primary")
-        answer = gr.Markdown("Ask a question to see excerpts.")
-        with gr.Accordion("Citations (top matches)", open=False):
+        # Chat interface
+        chatbot = gr.Chatbot(label="Conversation", height=400)
+        with gr.Row():
+            query = gr.Textbox(
+                label="",
+                placeholder="Ask a question about the documents...",
+                scale=4,
+                container=False
+            )
+            submit_btn = gr.Button("Send", variant="primary", scale=1)
+        
+        # Citations accordion below chat
+        with gr.Accordion("Citations (top matches from last query)", open=False):
             citations = gr.JSON(label="Source & similarity")
+        
+        # Clear button
+        clear_btn = gr.Button("Clear Chat", size="sm")
 
-        def on_ask(q):
-            a, c = _ask(q, settings, embedder, vstore)
-            return a, c
+        def on_submit(message, history):
+            """Handle message submission."""
+            new_history, new_citations = _ask(message, history or [], settings, embedder, vstore)
+            return new_history, new_citations, ""  # Return empty string to clear input
+        
+        def clear_chat():
+            """Clear the chat history."""
+            return [], []
 
-        ask_btn.click(on_ask, inputs=[query], outputs=[answer, citations])
-        query.submit(on_ask, inputs=[query], outputs=[answer, citations])
+        # Wire up the interactions
+        submit_btn.click(
+            on_submit,
+            inputs=[query, chatbot],
+            outputs=[chatbot, citations, query]
+        )
+        query.submit(
+            on_submit,
+            inputs=[query, chatbot],
+            outputs=[chatbot, citations, query]
+        )
+        clear_btn.click(clear_chat, outputs=[chatbot, citations])
     
     return demo