Update app.py

app.py

@@ -260,10 +260,10 @@ def handle_submit(user_query: str):
             
             # Update chat history and context
             st.session_state.chat_history.append(("assistant", response_dict))
-            st.session_state.context_memory.append({"query": user_query, "answer": assistant_answer})
+            st.session_state.context_memory.append({"query": user_query, "answer": response_dict})
             
             # Save to QA history
-            add_to_qa_history(user_query, assistant_answer)
+            add_to_qa_history(user_query, response_dict)
             
         except Exception as e:
             st.session_state.chat_history.append(("assistant", f"❌ เกิดข้อผิดพลาด: {str(e)}"))