Update app.py

app.py

@@ -274,76 +274,92 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
-            # Initialize the query key in session state if it doesn't exist
+            # Initialize session states
             if 'query' not in st.session_state:
                 st.session_state.query = ''
             
-            # Create the text input with a key
-            query = st.text_input(
-                "",
-                value=st.session_state.query,
-                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-                key="query_input"
-            )
-
-            # Button layout
-            col1, col2, col3 = st.columns([1, 1, 4])
-        
-            with col1:
-                send_query = st.button(
-                    "📤 ส่งคำถาม",
-                    type="primary",
-                    use_container_width=True,
-                    key="send_query_button"
-                )
-        
-            with col2:
-                clear_history = st.button(
-                    "🗑️ ล้างประวัติ",
-                    type="secondary",
-                    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"])
+            def clear_text():
+                st.session_state.query = ''
+                st.session_state.query_input = ''
+            
+            def handle_submit():
+                # Get the current query value
+                query = st.session_state.query_input
+                if query:
+                    if st.session_state.pipeline is None:
+                        st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
+                        return
                         
-                        # Clear the input after processing
-                        st.session_state.query = ''
-                        st.rerun()
+                    add_to_history("user", query)
                         
-                except Exception as e:
-                    st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
-            
-            elif send_query and not query:
-                st.warning("⚠️ กรุณาระบุคำถาม")
+                    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"])
+                            
+                            # Clear the input
+                            clear_text()
+                            st.rerun()
+                            
+                    except Exception as e:
+                        st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
+                else:
+                    st.warning("⚠️ กรุณาระบุคำถาม")
 
-        # Handle clear history action
-        if clear_history:
-            st.session_state.chat_history = []
-            st.rerun()
+            chat_col, info_col = st.columns([7, 3])
+            
+            with chat_col:
+                with st.container():
+                    # Display chat history
+                    display_chat_history()
+                    
+                    # Query input section
+                    st.markdown("""
+                        <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
+                            <span style="color: #ffffff; border-left: 4px solid #ffffff; padding-left: 0.8rem;">
+                                โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:
+                            </span>
+                        </label>
+                    """, unsafe_allow_html=True)
+                    
+                    # Text input with callback
+                    query = st.text_input(
+                        "",
+                        key="query_input",
+                        placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+                        on_change=handle_submit
+                    )
+        
+                    # Button layout
+                    col1, col2, col3 = st.columns([1, 1, 4])
+                
+                    with col1:
+                        st.button(
+                            "📤 ส่งคำถาม",
+                            type="primary",
+                            use_container_width=True,
+                            on_click=handle_submit
+                        )
+                
+                    with col2:
+                        st.button(
+                            "🗑️ ล้างประวัติ",
+                            type="secondary",
+                            use_container_width=True,
+                            on_click=lambda: setattr(st.session_state, 'chat_history', []) or st.rerun()
+                        )
 
     with info_col:
         # System information