Update app.py

app.py

@@ -239,6 +239,9 @@ def main():
     
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
+        
+    if 'query' not in st.session_state:
+        st.session_state.query = ''
 
     # Load QA history at startup
     if 'qa_history_loaded' not in st.session_state:
@@ -258,6 +261,46 @@ def main():
         with st.spinner("กำลังเริ่มต้นระบบ..."):
             st.session_state.pipeline = initialize_pipeline()
 
+    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
+                
+            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"])
+                    
+                    # Clear the input
+                    clear_text()
+                    st.rerun()
+                    
+            except Exception as e:
+                st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
+        else:
+            st.warning("⚠️ กรุณาระบุคำถาม")
+
     chat_col, info_col = st.columns([7, 3])
     
     with chat_col:
@@ -274,92 +317,32 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
-            # Initialize session states
-            if 'query' not in st.session_state:
-                st.session_state.query = ''
-            
-            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
-                        
-                    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"])
-                            
-                            # Clear the input
-                            clear_text()
-                            st.rerun()
-                            
-                    except Exception as e:
-                        st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
-                else:
-                    st.warning("⚠️ กรุณาระบุคำถาม")
+            # Create the text input with callbacks
+            query = st.text_input(
+                "",
+                key="query_input",
+                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+                on_change=handle_submit
+            )
 
-            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])
         
-                    # 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 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
@@ -395,7 +378,7 @@ def main():
                 </div>
             </div>
         """.format(
-            (datetime.now() + timedelta(hours=6)).strftime('%Y-%m-%d %H:%M:%S'),
+            (datetime.now() + timedelta(hours=7)).strftime('%Y-%m-%d %H:%M:%S'),
             "status-online" if st.session_state.pipeline else "status-offline",
             "🟢" if st.session_state.pipeline else "🔴",
             "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน"