Update app.py

app.py

@@ -63,6 +63,40 @@ def add_to_history(role: str, message: str):
     """Add message to chat history"""
     st.session_state.chat_history.append((role, message))
 
+def process_query(query: str):
+    """Process a query and update the chat history"""
+    if not query:
+        st.warning("กรุณาระบุคำถาม")
+        return
+        
+    if st.session_state.pipeline is None:
+        st.error("ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
+        return
+        
+    # Add user query to history
+    add_to_history("user", query)
+        
+    try:
+        with st.spinner("กำลังค้นหาคำตอบ..."):
+            # Process query
+            result = st.session_state.pipeline.process_query(query)
+            
+            # Add assistant response to history
+            add_to_history("assistant", result["answer"])
+            
+            # Create containers for additional information
+            with st.expander("📚 แสดงข้อมูลอ้างอิง"):
+                for i, doc in enumerate(result["documents"], 1):
+                    st.markdown(f"**เอกสารที่ {i}:**")
+                    st.text(doc.content)
+                    st.markdown("---")
+            
+            with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม"):
+                st.json(result["query_info"])
+            
+    except Exception as e:
+        st.error(f"เกิดข้อผิดพลาด: {str(e)}")
+
 def main():
     # Header
     st.title("🎓 ระบบค้นหาข้อมูลปฏิทินการศึกษา")
@@ -90,50 +124,12 @@ def main():
         # Add query button and clear chat button in the same line
         col1, col2 = st.columns([1, 4])
         with col1:
-            send_query = st.button("ส่งคำถาม", type="primary", use_container_width=True)
+            if st.button("ส่งคำถาม", type="primary", use_container_width=True):
+                process_query(query)
         with col2:
             if st.button("ล้างประวัติการสนทนา", type="secondary", use_container_width=True):
                 st.session_state.chat_history = []
                 st.rerun()
-
-        # Process query when button is clicked
-        if send_query and query:
-            if st.session_state.pipeline is None:
-                st.error("ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
-                return
-                
-            # Add user query to history
-            add_to_history("user", query)
-                
-            try:
-                with st.spinner("กำลังค้นหาคำตอบ..."):
-                    # Process query
-                    result = st.session_state.pipeline.process_query(query)
-                    
-                    # Add assistant response to history
-                    add_to_history("assistant", result["answer"])
-                    
-                    # Create containers for additional information
-                    with st.expander("📚 แสดงข้อมูลอ้างอิง"):
-                        for i, doc in enumerate(result["documents"], 1):
-                            st.markdown(f"**เอกสารที่ {i}:**")
-                            st.text(doc.content)
-                            st.markdown("---")
-                    
-                    with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม"):
-                        st.json(result["query_info"])
-                    
-                    # Clear the input field
-                    st.session_state.query_input = ""
-                    
-                    # Rerun to update chat display
-                    st.rerun()
-                    
-            except Exception as e:
-                st.error(f"เกิดข้อผิดพลาด: {str(e)}")
-        
-        elif send_query and not query:
-            st.warning("กรุณาระบุคำถาม")
     
     with info_col:
         # System information