Update app.py

app.py

@@ -274,6 +274,10 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
+            # Initialize the session state for the input box if not set
+            if "query_input" not in st.session_state:
+                st.session_state.query_input = ""
+            
             query = st.text_input(
                 "",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
@@ -282,7 +286,7 @@ def main():
 
             # Button layout inside main() function
             col1, col2, col3 = st.columns([1, 1, 4])
-        
+            
             with col1:
                 send_query = st.button(
                     "📤 ส่งคำถาม",
@@ -290,7 +294,7 @@ def main():
                     use_container_width=True,
                     key="send_query_button"
                 )
-        
+            
             with col2:
                 clear_history = st.button(
                     "🗑️ ล้างประวัติ",
@@ -298,35 +302,34 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
+            
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
-                    return
-                    
-                add_to_history("user", query)
-                    
-                try:
-                    with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
-                        result = st.session_state.pipeline.process_query(query)
-                        add_to_history("assistant", result["answer"])
-                        
-                        with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
-                            for i, doc in enumerate(result["documents"], 1):
-                                st.markdown(f"""
-                                    <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
-                                        <strong>เอกสารที่ {i}:</strong><br>
-                                        {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)}")
+                else:
+                    add_to_history("user", query)
+                    try:
+                        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
+                            
+                            with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
+                                for i, doc in enumerate(result["documents"], 1):
+                                    st.markdown(f"""
+                                        <div style="padding: 1rem; background-color: #F9FAFB; border-radius: 8px; margin: 0.5rem 0;">
+                                            <strong>เอกสารที่ {i}:</strong><br>
+                                            {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("⚠️ กรุณาระบุคำถาม")