Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 import json
 import os
-from datetime import *
-import subprocess  # Added missing import
-from datetime import datetime
+from datetime import datetime, timedelta
+import subprocess
 from huggingface_hub import HfApi
+from pathlib import Path
 from calendar_rag import (
     create_default_config,
     AcademicCalendarRAG,
@@ -17,43 +17,37 @@ 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 */
+                color: #000000;
                 background-color: white;
             }
             
             /* Headers styling */
             h1 {
-                color: #000000; /* Black headers */
-                font-size: 2.8rem !important; /* Larger font for headers */
+                color: #000000;
+                font-size: 2.8rem !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: 3px solid #1E3A8A;
             }
             
             h3, h4 {
-                color: #000000; /* Black sub-headers */
+                color: #000000;
                 font-weight: 600 !important;
-                font-size: 1.6rem !important; /* Increase font size for sub-headers */
+                font-size: 1.6rem !important;
                 margin-top: 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 */
-            }
-            
             /* Chat message styling */
             .chat-message {
                 padding: 1.5rem;
@@ -61,9 +55,9 @@ def load_custom_css():
                 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 */
+                line-height: 1.6 !important;
                 font-family: "Arial", sans-serif !important;
-                color: #000000 !important; /* Black text in chat */
+                color: #000000 !important;
             }
             
             .user-message {
@@ -74,50 +68,13 @@ def load_custom_css():
                 background-color: #EFF6FF !important;
             }
             
-            /* Input field styling */
-            .stTextInput input {
-                border-radius: 8px;
-                border: 2px solid #E5E7EB;
-                padding: 0.75rem;
-                font-size: 1.1rem;
-                font-family: "Arial", sans-serif !important;
-                color: #000000; /* Black input text */
-            }
-            
-            .stTextInput input:focus {
-                border-color: #1E3A8A;
-                box-shadow: 0 0 0 2px rgba(30, 58, 138, 0.1);
-            }
-            
-            /* 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 */
-                transition: all 0.2s ease;
-            }
-            
-            .stButton button:hover {
-                transform: translateY(-1px);
-                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-            }
-            
-            /* List styling */
-            ul {
-                font-size: 1.2rem !important;
-                margin-left: 1rem;
-                color: #000000; /* Black text in lists */
-                list-style-type: disc;
-            }
-            
-            /* Status indicator styling */
+            /* Status indicators */
             .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 */
+                font-size: 1.2rem;
+                color: #000000;
             }
             
             .status-online {
@@ -127,32 +84,11 @@ def load_custom_css():
             
             .status-offline {
                 background-color: #FDE8E8;
-                color:rgb(255, 255, 255);
+                color: rgb(255, 255, 255);
             }
             </style>
     """, unsafe_allow_html=True)
 
-# Page config
-st.set_page_config(
-    page_title="Academic Calendar Assistant",
-    page_icon="📅",
-    layout="wide",
-    initial_sidebar_state="collapsed"
-)
-
-# Load custom CSS
-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 = []
-
 def initialize_pipeline():
     """Initialize RAG pipeline with configurations"""
     try:
@@ -173,43 +109,6 @@ def initialize_pipeline():
         st.error(f"Error initializing pipeline: {str(e)}")
         return None
 
-def add_feedback(query: str, answer: str, is_correct: bool):
-    """Add feedback entry to session state and save to JSON and GitHub"""
-    feedback_entry = {
-        "timestamp": (datetime.now() + timedelta(hours=7)).isoformat(),
-        "query": query,
-        "answer": answer,
-        "is_correct": is_correct
-    }
-    st.session_state.feedback_data.append(feedback_entry)
-    
-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 load_qa_history():
     """Load QA history from local JSON file"""
     try:
@@ -230,7 +129,8 @@ def save_qa_history(history_data):
             json.dump(history_data, f, ensure_ascii=False, indent=2)
         
         # Push to Hugging Face
-        api = HfApi()
+        hf_token = os.getenv('HF_TOKEN') or st.secrets['HF_TOKEN']
+        api = HfApi(token=hf_token)
         api.upload_file(
             path_or_fileobj="qa_history.json",
             path_in_repo="qa_history.json",
@@ -258,7 +158,6 @@ def add_to_qa_history(query: str, answer: str):
     save_qa_history(history_data)
     return history_data
 
-# Modify the main code to include these changes
 def add_to_history(role: str, message: str):
     """Add message to chat history and save if it's a complete QA pair"""
     st.session_state.chat_history.append((role, message))
@@ -269,20 +168,49 @@ def add_to_history(role: str, message: str):
         user_query = st.session_state.chat_history[-2][1]
         add_to_qa_history(user_query, message)
 
-# In the main function, load history at startup
-if 'qa_history_loaded' not in st.session_state:
-    st.session_state.qa_history_loaded = True
-    loaded_history = load_qa_history()
-    # You might want to display or process the loaded history here
-
-# Modify the clear history button to also clear the saved history
-if clear_history:
-    st.session_state.chat_history = []
-    save_qa_history([])  # Save empty history
-    st.rerun()
+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)
 
 def main():
-    # Header with animation
+    # Page config
+    st.set_page_config(
+        page_title="Academic Calendar Assistant",
+        page_icon="📅",
+        layout="wide",
+        initial_sidebar_state="collapsed"
+    )
+
+    # Load custom CSS
+    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 = []
+
+    # Load QA history at startup
+    if 'qa_history_loaded' not in st.session_state:
+        st.session_state.qa_history_loaded = True
+        load_qa_history()
+
+    # Header
     st.markdown("""
         <div style="text-align: center; padding: 2rem 0;">
             <h1>🎓 ระบบค้นหาข้อมูลปฏิทินการศึกษา</h1>
@@ -297,26 +225,24 @@ def main():
     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()
             
-            # Add the styled label using st.markdown
+            # Query input section
             st.markdown("""
                 <label for="query_input" style="font-size: 1.2rem; font-weight: bold; color: black;">
                     <span style="color: white; background-color: yellow; padding: 0 0.2rem;">โปรดระบุคำถามเกี่ยวกับ ปฏิทินการศึกษา:</span>
                 </label>
             """, unsafe_allow_html=True)
             
-            # The input box without a label
             query = st.text_input(
                 "",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
                 key="query_input"
             )
 
-            # Button layout with enhanced styling
+            # Button layout
             col1, col2, col3 = st.columns([1, 1, 4])
 
             with col1:
@@ -326,14 +252,6 @@ def main():
                     use_container_width=True,
                     key="send_query_button"
                 )
-                st.markdown("""
-                    <style>
-                    #send_query_button button {
-                        background-color: #28a745 !important; /* Green background */
-                        color: white !important;
-                    }
-                    </style>
-                """, unsafe_allow_html=True)
 
             with col2:
                 clear_history = st.button(
@@ -342,15 +260,6 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
-                st.markdown("""
-                    <style>
-                    #clear_history_button button {
-                        background-color: #dc3545 !important; /* Red background */
-                        color: white !important;
-                    }
-                    </style>
-                """, unsafe_allow_html=True)
-
 
             # Process query
             if send_query and query:
@@ -365,7 +274,6 @@ def main():
                         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"""
@@ -385,9 +293,15 @@ def main():
             
             elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
+
+            # Clear history
+            if clear_history:
+                st.session_state.chat_history = []
+                save_qa_history([])  # Clear saved history
+                st.rerun()
     
     with info_col:
-        # Enhanced system information section
+        # System information
         st.markdown("""
             <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px; margin-bottom: 2rem;">
                 <h3 style="color: #1E3A8A;">ℹ️ เกี่ยวกับระบบ</h3>
@@ -406,7 +320,7 @@ def main():
             </div>
         """, unsafe_allow_html=True)
         
-        # Enhanced system status section
+        # System status
         st.markdown("""
             <div style="background-color: #f9fafb; padding: 1.5rem; border-radius: 12px;">
                 <h3 style="color: #1E3A8A;">🔄 สถานะระบบ</h3>