Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
 import json
 import os
-import time
 from datetime import datetime, timedelta
-import subprocess
 from huggingface_hub import HfApi
 from pathlib import Path
 from calendar_rag import (
@@ -126,6 +124,7 @@ def load_qa_history():
 def save_qa_history(history_entry):
     """Save QA history entry to local JSON file and push to Hugging Face"""
     try:
+        
         history_file = Path("qa_history.json")
         
         # Initialize or load existing history
@@ -145,19 +144,15 @@ def save_qa_history(history_entry):
         with open("qa_history.json", "w", encoding="utf-8") as f:
             json.dump(history_data, f, ensure_ascii=False, indent=2)
         
-        # Push to Hugging Face without triggering a reload
-        try:
-            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",
-                repo_id="JirasakJo/Questions_Graduate_Studies_Calendar_2024",
-                repo_type="space"
-            )
-        except Exception as e:
-            st.warning(f"Warning: Could not sync history to Hugging Face: {str(e)}", icon="⚠️")
-            
+        # Push to Hugging Face
+        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",
+            repo_id="JirasakJo/Questions_Graduate_Studies_Calendar_2024",
+            repo_type="space"
+        )
     except Exception as e:
         st.error(f"Error saving QA history: {str(e)}")
 
@@ -219,30 +214,6 @@ def display_chat_history():
                 </div>
             """, unsafe_allow_html=True)
 
-
-def process_query(query: str, pipeline):
-    """Process a query and update the chat history"""
-    try:
-        result = pipeline.process_query(query)
-        
-        # Add to session state history
-        st.session_state.chat_history.append(("user", query))
-        st.session_state.chat_history.append(("assistant", result["answer"]))
-        
-        # Save to QA history without reloading
-        history_entry = {
-            "timestamp": (datetime.now() + timedelta(hours=7)).isoformat(),
-            "query": query,
-            "answer": result["answer"]
-        }
-        save_qa_history(history_entry)
-        
-        return result
-        
-    except Exception as e:
-        st.error(f"Error processing query: {str(e)}")
-        return None
-        
 def main():
     # Page config
     st.set_page_config(
@@ -261,9 +232,18 @@ def main():
     
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
-    
-    if 'current_result' not in st.session_state:
-        st.session_state.current_result = None
+
+    # 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>
+        </div>
+    """, unsafe_allow_html=True)
 
     # Initialize pipeline
     if st.session_state.pipeline is None:
@@ -284,7 +264,6 @@ def main():
                 </label>
             """, unsafe_allow_html=True)
             
-            # Query input with key
             query = st.text_input(
                 "",
                 placeholder="เช่น: วันสุดท้ายของการสอบปากเปล่าในภาคเรียนที่ 1/2567 คือวันที่เท่าไร?",
@@ -303,49 +282,51 @@ def main():
                 )
         
             with col2:
-                if st.button(
+                clear_history = st.button(
                     "🗑️ ล้างประวัติ",
                     type="secondary",
                     use_container_width=True,
                     key="clear_history_button"
-                ):
-                    st.session_state.chat_history = []
-                    st.session_state.current_result = None
-                    clear_qa_history()
-                    st.rerun()
+                )
 
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
-                else:
+                    return
+
+                add_to_history("user", query)
+                    
+                try:
                     with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
-                        result = process_query(query, st.session_state.pipeline)
-                        if result:
-                            st.session_state.current_result = result
-                            
-                            # Display reference documents
-                            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)
-                            
-                            # Display query analysis
-                            with st.expander("🔍 รายละเอียดการวิเคราะห์คำถาม", expanded=False):
-                                st.json(result["query_info"])
+                        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"])
+                        
+                        st.rerun()
+                        
+                except Exception as e:
+                    st.error(f"❌ เกิดข้อผิดพลาด: {str(e)}")
             
             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()
+            # Clear history
+            if clear_history:
+                st.session_state.chat_history = []
+                save_qa_history([])  # Clear saved history
+                st.rerun()
     
     with info_col:
         # System information