Update app.py

app.py

@@ -1,7 +1,12 @@
 import streamlit as st
 import json
 import os
-from calendar_rag import create_default_config, CalendarRAG
+from datetime import datetime
+from calendar_rag import (
+    create_default_config,
+    AcademicCalendarRAG,
+    PipelineConfig
+)
 
 # Page config
 st.set_page_config(
@@ -10,9 +15,12 @@ st.set_page_config(
     layout="wide"
 )
 
-# Initialize session state
+# Initialize session state for pipeline and chat history
 if 'pipeline' not in st.session_state:
     st.session_state.pipeline = None
+    
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []
 
 def initialize_pipeline():
     """Initialize RAG pipeline with configurations"""
@@ -20,10 +28,14 @@ def initialize_pipeline():
         # Get OpenAI API key from environment variable or Streamlit secrets
         openai_api_key = os.getenv('OPENAI_API_KEY') or st.secrets['OPENAI_API_KEY']
         
-        # Create config and initialize pipeline
+        # Create config
         config = create_default_config(openai_api_key)
         config.localization.enable_thai_normalization = True
-        pipeline = CalendarRAG(config)
+        config.retriever.top_k = 5
+        config.model.temperature = 0.3
+        
+        # Initialize pipeline
+        pipeline = AcademicCalendarRAG(config)
         
         # Load calendar data
         with open("calendar.json", "r", encoding="utf-8") as f:
@@ -36,6 +48,21 @@ def initialize_pipeline():
         st.error(f"Error initializing pipeline: {str(e)}")
         return None
 
+def display_chat_history():
+    """Display chat history with user and assistant messages"""
+    for i, (role, message) in enumerate(st.session_state.chat_history):
+        if role == "user":
+            st.markdown(f"**🧑 คำถาม:**")
+            st.markdown(f"> {message}")
+        else:
+            st.markdown(f"**🤖 คำตอบ:**")
+            st.markdown(message)
+            st.markdown("---")
+
+def add_to_history(role: str, message: str):
+    """Add message to chat history"""
+    st.session_state.chat_history.append((role, message))
+
 def main():
     # Header
     st.title("🎓 ระบบค้นหาข้อมูลปฏิทินการศึกษา")
@@ -46,59 +73,87 @@ def main():
         with st.spinner("กำลังเริ่มต้นระบบ..."):
             st.session_state.pipeline = initialize_pipeline()
     
-    # Main query interface
-    query = st.text_input(
-        "โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:",
-        placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?"
-    )
+    # Create two columns for chat layout
+    chat_col, info_col = st.columns([2, 1])
+    
+    with chat_col:
+        # Display chat history
+        display_chat_history()
+        
+        # Main query interface
+        query = st.text_input(
+            "โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:",
+            placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
+            key="query_input"
+        )
 
-    # Add query button
-    if st.button("ค้นหา", type="primary"):
-        if not query:
-            st.warning("กรุณาระบุคำถาม")
-            return
-            
-        if st.session_state.pipeline is None:
-            st.error("ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
-            return
-            
-        try:
-            with st.spinner("กำลังค้นหาคำตอบ..."):
-                # Process query
-                result = st.session_state.pipeline.process_query(query)
-                
-                # Display answer
-                st.markdown("### 📝 คำตอบ")
-                st.write(result["answer"])
+        # Add query button and clear chat button in the same line
+        col1, col2 = st.columns([1, 4])
+        with col1:
+            send_query = st.button("ส่งคำถาม", type="primary", use_container_width=True)
+        with col2:
+            if st.button("ล้างประวัติการสนทนา", type="secondary", use_container_width=True):
+                st.session_state.chat_history = []
+                st.rerun()
+
+        # Process query when button is clicked
+        if send_query and query:
+            if st.session_state.pipeline is None:
+                st.error("ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
+                return
                 
-                # Display relevant documents in expander
-                with st.expander("แสดงข้อมูลอ้างอิง"):
-                    for i, doc in enumerate(result["documents"], 1):
-                        st.markdown(f"**เอกสารที่ {i}:**")
-                        st.text(doc.content)
-                        st.markdown("---")
+            # Add user query to history
+            add_to_history("user", query)
                 
-                # Display query understanding
-                with st.expander("รายละเอียดการวิเคราะห์คำถาม"):
-                    st.json(result["query_info"])
+            try:
+                with st.spinner("กำลังค้นหาคำตอบ..."):
+                    # Process query
+                    result = st.session_state.pipeline.process_query(query)
+                    
+                    # Add assistant response to history
+                    add_to_history("assistant", result["answer"])
+                    
+                    # Create containers for additional information
+                    with st.expander("📚 แสดงข้อมูลอ้างอิง"):
+                        for i, doc in enumerate(result["documents"], 1):
+                            st.markdown(f"**เอกสารที่ {i}:**")
+                            st.text(doc.content)
+                            st.markdown("---")
+                    
+                    with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม"):
+                        st.json(result["query_info"])
+                    
+                    # Clear the input field
+                    st.session_state.query_input = ""
                     
-        except Exception as e:
-            st.error(f"เกิดข้อผิดพลาด: {str(e)}")
+                    # Rerun to update chat display
+                    st.rerun()
+                    
+            except Exception as e:
+                st.error(f"เกิดข้อผิดพลาด: {str(e)}")
+        
+        elif send_query and not query:
+            st.warning("กรุณาระบุคำถาม")
     
-    # Add sidebar with information
-    with st.sidebar:
-        st.markdown("### 📚 เกี่ยวกับระบบ")
+    with info_col:
+        # System information
+        st.markdown("### ℹ️ เกี่ยวกับระบบ")
         st.markdown("""
         ระบบนี้ใช้เทคโนโลยี RAG (Retrieval-Augmented Generation) 
         ในการค้นหาและตอบคำถามเกี่ยวกับปฏิทินการศึกษา
         
         สามารถสอบถามข้อมูลเกี่ยวกับ:
-        - กำหนดการต่างๆ
-        - วันสำคัญ
-        - การลงทะเบียน
-        - การสอบ
-        - วันหยุด
+        - กำหนดการต่างๆ ในปฏิทินการศึกษา
+        - วันสำคัญและกิจกรรม
+        - การลงทะเบียนเรียน
+        - กำหนดการสอบ
+        - วันหยุดการศึกษา
         """)
+        
+        # System status
+        st.markdown("### 🔄 สถานะระบบ")
+        st.markdown(f"**เวลาปัจจุบัน:** {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
+        st.markdown(f"**สถานะระบบ:** {'🟢 พร้อมใช้งาน' if st.session_state.pipeline else '🔴 ไม่พร้อมใช้งาน'}")
 
 if __name__ == "__main__":
     main()