Update app.py

app.py

@@ -14,61 +14,56 @@ def load_custom_css():
             <style>
             /* General body styling */
             body {
-                font-family: "Arial", sans-serif !important; /* Clean and readable font */
-                color: #000000 !important; /* Black text */
+                font-family: "Arial", sans-serif !important;
+                color: #000000 !important;
                 background-color: white !important;
-                line-height: 1.7 !important; /* Increase line spacing */
+                line-height: 1.7 !important;
             }
             
             /* Main container styling */
             .main {
                 padding: 2rem;
-                color: #000000; /* Ensure all text is black */
-                background-color: white;
+                background-color: #F9FAFB;
             }
             
             /* Headers styling */
             h1 {
-                color: #000000; /* Black headers */
-                font-size: 2.8rem !important; /* Larger font for headers */
+                color: #1E40AF;
+                font-size: 2.5rem !important;
                 font-weight: 700 !important;
-                margin-bottom: 1.5rem !important;
                 text-align: center;
-                padding: 1rem 0;
-                border-bottom: 3px solid #1E3A8A; /* Keep a blue line for a clean design */
+                border-bottom: 2px solid #1E40AF;
+                padding-bottom: 0.5rem;
             }
             
             h3, h4 {
-                color: #000000; /* Black sub-headers */
+                color: #1F2937;
                 font-weight: 600 !important;
-                font-size: 1.6rem !important; /* Increase font size for sub-headers */
-                margin-top: 1.5rem !important;
+                font-size: 1.5rem !important;
             }
             
             /* Paragraphs and text */
             p, span, li {
-                font-size: 1.2rem !important; /* Larger font size for regular text */
-                color: #000000 !important; /* Black text */
+                font-size: 1.1rem !important;
+                color: #374151 !important;
             }
             
             /* Chat message styling */
             .chat-message {
-                padding: 1.5rem;
-                border-radius: 10px;
+                padding: 1rem;
+                border-radius: 8px;
                 margin: 1rem 0;
-                box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
                 font-size: 1.1rem !important;
-                line-height: 1.6 !important; /* Better readability */
-                font-family: "Arial", sans-serif !important;
-                color: #000000 !important; /* Black text in chat */
+                line-height: 1.6 !important;
+                color: #374151 !important;
             }
             
             .user-message {
-                background-color: #F3F4F6 !important;
+                background-color: #E5E7EB;
             }
             
             .assistant-message {
-                background-color: #EFF6FF !important;
+                background-color: #E0F2FE;
             }
             
             /* Input field styling */
@@ -77,26 +72,26 @@ def load_custom_css():
                 border: 2px solid #E5E7EB;
                 padding: 0.75rem;
                 font-size: 1.1rem;
-                font-family: "Arial", sans-serif !important;
-                color: #000000; /* Black input text */
+                color: #111827;
             }
             
             .stTextInput input:focus {
-                border-color: #1E3A8A;
-                box-shadow: 0 0 0 2px rgba(30, 58, 138, 0.1);
+                border-color: #1E40AF;
+                box-shadow: 0 0 0 2px rgba(30, 64, 175, 0.3);
             }
             
             /* Button styling */
             .stButton button {
                 border-radius: 8px;
                 font-weight: 600;
-                font-size: 1.2rem !important; /* Bigger buttons for better interaction */
-                color: #000000; /* Black text on buttons */
+                font-size: 1.2rem !important;
+                color: #ffffff;
+                background-color: #1E40AF;
                 transition: all 0.2s ease;
             }
             
             .stButton button:hover {
-                transform: translateY(-1px);
+                transform: translateY(-2px);
                 box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
             }
             
@@ -104,28 +99,9 @@ def load_custom_css():
             ul {
                 font-size: 1.2rem !important;
                 margin-left: 1rem;
-                color: #000000; /* Black text in lists */
+                color: #374151;
                 list-style-type: disc;
             }
-            
-            /* Status indicator styling */
-            .status-indicator {
-                padding: 0.5rem 1rem;
-                border-radius: 6px;
-                font-weight: 500;
-                font-size: 1.2rem; /* Larger for better clarity */
-                color: #000000; /* Black text for status */
-            }
-            
-            .status-online {
-                background-color: #DEF7EC;
-                color: #03543F;
-            }
-            
-            .status-offline {
-                background-color: #FDE8E8;
-                color: #9B1C1C;
-            }
             </style>
     """, unsafe_allow_html=True)
 
@@ -134,7 +110,6 @@ st.set_page_config(
     page_title="Academic Calendar Assistant",
     page_icon="📅",
     layout="wide",
-    initial_sidebar_state="collapsed"
 )
 
 # Load custom CSS
@@ -143,10 +118,10 @@ load_custom_css()
 # Initialize session state
 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 = []
-    
+
 if 'feedback_data' not in st.session_state:
     st.session_state.feedback_data = []
 
@@ -170,211 +145,70 @@ def initialize_pipeline():
         st.error(f"Error initializing pipeline: {str(e)}")
         return None
 
-def save_feedback_to_json():
-    """Save feedback data to a JSON file"""
-    try:
-        # Get current directory
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        file_path = os.path.join(current_dir, 'feedback_data.json')
-        
-        # Log the attempt
-        st.write(f"Attempting to save feedback to: {file_path}")
-        
-        # Create directory if it doesn't exist
-        os.makedirs(current_dir, exist_ok=True)
-        
-        # Save the file
-        with open(file_path, "w", encoding="utf-8") as f:
-            json.dump(
-                st.session_state.feedback_data,
-                f,
-                ensure_ascii=False,
-                indent=2
-            )
-            
-        # Verify file was created
-        if os.path.exists(file_path):
-            st.write(f"✅ Feedback saved successfully to {file_path}")
-            st.write(f"Current feedback data: {st.session_state.feedback_data}")
-            return True
-        else:
-            st.error("File was not created despite no errors")
-            return False
-            
-    except Exception as e:
-        st.error(f"Error saving feedback: {str(e)}")
-        st.write(f"Current directory: {current_dir}")
-        st.write(f"Current feedback data: {st.session_state.feedback_data}")
-        return False
-
-def add_feedback(query: str, answer: str, is_correct: bool):
-    """Add feedback entry to session state and save to JSON"""
-    feedback_entry = {
-        "timestamp": datetime.now().isoformat(),
-        "query": query,
-        "answer": answer,
-        "is_correct": is_correct
-    }
-    st.session_state.feedback_data.append(feedback_entry)
-    save_feedback_to_json()
-
-def evaluate_answer(query: str, answer: str):
-    """Display evaluation buttons for the answer"""
-    col1, col2, col3 = st.columns([1, 1, 4])
-    
-    with col1:
-        if st.button("✅ ถูกต้อง", type="primary", key=f"correct_{len(st.session_state.chat_history)}"):
-            add_feedback(query, answer, True)
-            st.success("บันทึกผลการประเมินแล้ว")
-            return True
-            
-    with col2:
-        if st.button("❌ ไม่ถูกต้อง", type="secondary", key=f"incorrect_{len(st.session_state.chat_history)}"):
-            add_feedback(query, answer, False)
-            st.error("บันทึกผลการประเมินแล้ว")
-            return True
-            
-    return False
-
-def display_chat_history():
-    """Display chat history with enhanced styling"""
-    for i, (role, message) in enumerate(st.session_state.chat_history):
-        if role == "user":
-            st.markdown(f"""
-                <div class="chat-message user-message">
-                    <strong>🧑 คำถาม:</strong><br>
-                    {message}
-                </div>
-            """, unsafe_allow_html=True)
-        else:
-            st.markdown(f"""
-                <div class="chat-message assistant-message">
-                    <strong>🤖 คำตอบ:</strong><br>
-                    {message}
-                </div>
-            """, unsafe_allow_html=True)
-            
-            # Add evaluation buttons after each assistant response
-            if i == len(st.session_state.chat_history) - 1:  # Only for the latest answer
-                user_query = st.session_state.chat_history[i-1][1]  # Get the corresponding user query
-                evaluate_answer(user_query, message)
-
 def add_to_history(role: str, message: str):
     """Add message to chat history"""
     st.session_state.chat_history.append((role, message))
 
+def display_chat_history():
+    """Display chat history with enhanced styling"""
+    for role, message in st.session_state.chat_history:
+        style_class = "user-message" if role == "user" else "assistant-message"
+        st.markdown(f"""
+            <div class="chat-message {style_class}">
+                <strong>{'🧑 คำถาม:' if role == 'user' else '🤖 คำตอบ:'}</strong><br>
+                {message}
+            </div>
+        """, unsafe_allow_html=True)
+
 def main():
-    # Header with animation
+    # Header
     st.markdown("""
-        <div style="text-align: center; padding: 2rem 0;">
-            <h1>🎓 ระบบค้นหาข้อมูลปฏิทินการศึกษา</h1>
+        <div style="text-align: center;">
+            <h1>🎓 Academic Calendar Assistant</h1>
         </div>
     """, unsafe_allow_html=True)
 
     # Initialize pipeline if needed
     if st.session_state.pipeline is None:
-        with st.spinner("กำลังเริ่มต้นระบบ..."):
+        with st.spinner("Initializing system..."):
             st.session_state.pipeline = initialize_pipeline()
-    
-    # Create two columns with better proportions
-    chat_col, info_col = st.columns([7, 3])
-    
-    with chat_col:
-        # Add a subtle container for the chat interface
-        with st.container():
-            # Display chat history
-            display_chat_history()
-            
-            # Main query interface with enhanced styling
-            st.markdown("<br>", unsafe_allow_html=True)
-            query = st.text_input(
-                "โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:",
-                placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
-                key="query_input"
-            )
 
-            # Button layout with enhanced styling
-            col1, col2, col3 = st.columns([1, 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()
+    # Two-column layout
+    chat_col, info_col = st.columns([2, 1])
 
-            # Process query
-            if send_query and query:
-                if st.session_state.pipeline is None:
-                    st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
-                    return
-                    
+    with chat_col:
+        # Chat interface
+        display_chat_history()
+        query = st.text_input("Enter your question about the academic calendar:", placeholder="e.g., What is the last day of exams for Term 1, 2023?")
+        
+        if st.button("Send"):
+            if not query:
+                st.warning("Please enter a question.")
+            else:
                 add_to_history("user", query)
-                    
                 try:
-                    with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
+                    with st.spinner("Searching for answers..."):
                         result = st.session_state.pipeline.process_query(query)
                         add_to_history("assistant", result["answer"])
-                        
-                        # Enhanced expandable sections
-                        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()
-                        
+                        st.experimental_rerun()
                 except Exception as e:
-                    st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
-            
-            elif send_query and not query:
-                st.warning("⚠️ กรุณาระบุคำถาม")
-    
+                    st.error(f"Error: {str(e)}")
+
     with info_col:
-        # Enhanced system information section
+        # System info
         st.markdown("""
-            <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px; margin-bottom: 2rem;">
-                <h3>ℹ️ เกี่ยวกับระบบ</h3>
-                <p style="color: #4B5563;">
-                    ระบบนี้ใช้เทคโนโลยี <strong>RAG (Retrieval-Augmented Generation)</strong> 
-                    ในการค้นหาและตอบคำถามเกี่ยวกับปฏิทินการศึกษา
-                </p>
-                <h4 style="color: #1E3A8A; margin-top: 1rem;">สามารถสอบถามข้อมูลเกี่ยวกับ:</h4>
-                <ul style="list-style-type: none; padding-left: 0;">
-                    <li>📅 กำหนดการต่างๆ ในปฏิทินการศึกษา</li>
-                    <li>🎯 วันสำคัญและกิจกรรม</li>
-                    <li>📝 การลงทะเบียนเรียน</li>
-                    <li>📚 กำหนดการสอบ</li>
-                    <li>🏖️ วันหยุดการศึกษา</li>
+            <div style="background-color: #E5E7EB; padding: 1rem; border-radius: 8px;">
+                <h3>About the System</h3>
+                <p>This system uses <strong>Retrieval-Augmented Generation (RAG)</strong> to answer questions about academic calendars.</p>
+                <ul>
+                    <li>📅 Event schedules</li>
+                    <li>🎯 Important dates</li>
+                    <li>📝 Registration deadlines</li>
+                    <li>📚 Exam schedules</li>
+                    <li>🏖️ Holidays</li>
                 </ul>
             </div>
         """, unsafe_allow_html=True)
-        
-        # Enhanced system status section
-        st.markdown("""
-            <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px;">
-                <h3>🔄 สถานะระบบ</h3>
-                <div style="margin-top: 1rem;">
-                    <p><strong>⏰ เวลาปัจจุบัน:</strong><br>
-                    {}</p>
-                    <p><strong>📡 สถานะระบบ:</strong><br>
-                    <span class="status-indicator {}">
-                        {} {}
-                    </span></p>
-                </div>
-            </div>
-        """.format(
-            datetime.now().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 "ไม่พร้อมใช้งาน"
-        ), unsafe_allow_html=True)
 
 if __name__ == "__main__":
-    main()
+    main()