Update app.py

app.py

@@ -221,54 +221,52 @@ def display_chat_history():
                 </div>
             """, unsafe_allow_html=True)
 
+if 'context_memory' not in st.session_state:
+    st.session_state.context_memory = []
+
 def submit():
     if not st.session_state.query_input:
         st.warning("⚠️ กรุณาระบุคำถาม")
         return
 
-    st.session_state.processing_query = True
-
     user_query = st.session_state.query_input
+    st.session_state.query_input = ""  # Clear input
+
     st.session_state.chat_history.append(("user", user_query))
 
-    # Show loading message
-    st.session_state.chat_history.append(("assistant", {
-        "answer": "🔍 กำลังค้นหาคำตอบ...",
-        "documents": [],
-        "query_info": {}
-    }))
-    
-    st.session_state.query_input = ""  # Clear input
+    # Maintain a rolling context of past interactions
+    if len(st.session_state.context_memory) > 5:  # Limit to last 5 Q&A pairs
+        st.session_state.context_memory.pop(0)
 
     try:
-        print("Processing query:", user_query)  # Debugging
-        result = st.session_state.pipeline.process_query(user_query)
+        # Include past interactions for better understanding
+        query_with_context = "\n".join(
+            [f"Q: {qa['query']}\nA: {qa['answer']}" for qa in st.session_state.context_memory]
+        ) + f"\nQ: {user_query}"
+
+        print("Processing query with context:", query_with_context)  # Debugging
+        result = st.session_state.pipeline.process_query(query_with_context)
         print("Query processed successfully")  # Debugging
 
-        # Update the answer
-        st.session_state.chat_history[-1] = ("assistant", {
-            "answer": result["answer"],
-            "documents": result["documents"],
-            "query_info": result["query_info"]
-        })
+        assistant_answer = result["answer"]
+
+        # Store in chat history
+        st.session_state.chat_history.append(("assistant", assistant_answer))
 
-        # Save QA history
-        add_to_qa_history(user_query, result["answer"])
+        # Save the new QA pair into context memory
+        st.session_state.context_memory.append({"query": user_query, "answer": assistant_answer})
+
+        # Save the QA history
+        add_to_qa_history(user_query, assistant_answer)
 
     except Exception as e:
         print("Error processing query:", str(e))  # Debugging
-        st.session_state.chat_history[-1] = ("assistant", {
-            "answer": f"❌ เกิดข้อผิดพลาด: {str(e)}",
-            "documents": [],
-            "query_info": {}
-        })
+        st.session_state.chat_history.append(("assistant", f"❌ เกิดข้อผิดพลาด: {str(e)}"))
         st.error(f"Query processing error: {e}")
 
     finally:
         st.session_state.processing_query = False  # Reset flag
 
-    # st.rerun()  # Refresh UI
-
 def main():
     # Page config
     st.set_page_config(
@@ -369,6 +367,7 @@ def main():
             
             with col2:
                 if st.button("🗑️ ล้างประวัติ", type="secondary", use_container_width=True):
+                    st.session_state.context_memory = []
                     st.session_state.chat_history = []
                     st.rerun()