Update app.py

app.py

@@ -251,10 +251,15 @@ def handle_submit(user_query: str):
             
             # Process query
             result = st.session_state.pipeline.process_query(query_with_context)
-            assistant_answer = result["answer"]
+            
+            # Create response dictionary with answer and documents
+            response_dict = {
+                "answer": result.get("answer", ""),
+                "documents": result.get("documents", [])
+            }
             
             # Update chat history and context
-            st.session_state.chat_history.append(("assistant", assistant_answer))
+            st.session_state.chat_history.append(("assistant", response_dict))
             st.session_state.context_memory.append({"query": user_query, "answer": assistant_answer})
             
             # Save to QA history