Update app.py

app.py

@@ -224,16 +224,13 @@ def display_chat_history():
 if 'context_memory' not in st.session_state:
     st.session_state.context_memory = []
 
-def handle_submit():
+def handle_submit(user_query: str):
     """Handle form submission logic"""
-    if not st.session_state.query_input:
+    if not user_query:
         st.warning("⚠️ กรุณาระบุคำถาม")
         return
 
-    user_query = st.session_state.query_input.strip()
-    
-    # Important: Clear input immediately to prevent duplicate submissions
-    st.session_state.query_input = ""
+    user_query = user_query.strip()
     
     # Prevent duplicate submissions by checking last message
     if not st.session_state.chat_history or st.session_state.chat_history[-1][1] != user_query:
@@ -269,7 +266,7 @@ def handle_submit():
         
         finally:
             st.session_state.processing_query = False
-            st.rerun()  # Refresh the page to show new messages
+            st.rerun()
 
 def create_chat_input():
     """Create the chat input section with form handling"""
@@ -288,7 +285,7 @@ def create_chat_input():
             """, unsafe_allow_html=True)
             
             # Text input
-            st.text_input(
+            query = st.text_input(
                 "",
                 key="query_input",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?"
@@ -302,7 +299,7 @@ def create_chat_input():
             )
             
             if submitted:
-                handle_submit()
+                handle_submit(query)
     
     # Clear history button outside the form
     with clear_col:
@@ -333,8 +330,8 @@ def main():
     if 'context_memory' not in st.session_state:
         st.session_state.context_memory = []
         
-    if 'query' not in st.session_state:
-        st.session_state.query = ''
+    if 'processing_query' not in st.session_state:
+        st.session_state.processing_query = False
 
     # Load QA history at startup
     if 'qa_history_loaded' not in st.session_state: