Update app.py

app.py

@@ -88,6 +88,10 @@ def load_custom_css():
             </style>
     """, unsafe_allow_html=True)
 
+# Define a callback to reset query_input
+def reset_query_input():
+    st.session_state.query_input = ""
+    
 def initialize_pipeline():
     """Initialize RAG pipeline with configurations"""
     try:
@@ -274,17 +278,16 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
-            # Initialize the session state for the input box if not set
+            # Inside the main function
             if "query_input" not in st.session_state:
                 st.session_state.query_input = ""
             
             query = st.text_input(
                 "",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-                key="query_input"
+                key="query_input",
             )
-
-            # Button layout inside main() function
+            
             col1, col2, col3 = st.columns([1, 1, 4])
             
             with col1:
@@ -313,8 +316,10 @@ def main():
                         with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
                             result = st.session_state.pipeline.process_query(query)
                             add_to_history("assistant", result["answer"])
-                            st.session_state.query_input = ""  # Clear input after submission
-                            
+            
+                            # Reset the query input
+                            reset_query_input()
+            
                             with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
                                 for i, doc in enumerate(result["documents"], 1):
                                     st.markdown(f"""
@@ -323,14 +328,13 @@ def main():
                                             {doc.content}
                                         </div>
                                     """, unsafe_allow_html=True)
-                            
+            
                             with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
                                 st.json(result["query_info"])
-                            
+            
                             st.rerun()
                     except Exception as e:
                         st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
-            
             elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")