Update app.py

app.py

@@ -1,8 +1,8 @@
 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 (
@@ -11,6 +11,8 @@ from calendar_rag import (
     PipelineConfig
 )
 
+all_history = []
+
 # Custom CSS for enhanced styling
 def load_custom_css():
     st.markdown("""
@@ -121,9 +123,11 @@ def load_qa_history():
         st.error(f"Error loading QA history: {str(e)}")
         return []
 
+
 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
@@ -156,14 +160,16 @@ def save_qa_history(history_entry):
         st.error(f"Error saving QA history: {str(e)}")
 
 def add_to_qa_history(query: str, answer: str):
-    """Add new QA pair to history"""
+    """Add new QA pair to history and store it in all_history"""
     history_entry = {
         "timestamp": (datetime.now() + timedelta(hours=7)).isoformat(),
         "query": query,
         "answer": answer
     }
     
-    save_qa_history(history_entry)
+    # Append the new history entry to the all_history list
+    all_history.append(history_entry)
+
     return history_entry
 
 def clear_qa_history():
@@ -269,7 +275,7 @@ def main():
                 key="query_input"
             )
 
-            # Button layout inside main() function
+            # Button layout
             col1, col2, col3 = st.columns([1, 1, 4])
         
             with col1:
@@ -287,12 +293,13 @@ def main():
                     use_container_width=True,
                     key="clear_history_button"
                 )
+
             # Process query
             if send_query and query:
                 if st.session_state.pipeline is None:
                     st.error("❌ ไม่สามารถเชื่อมต่อกับระบบได้ กรุณาลองใหม่อีกครั้ง")
                     return
-                    
+
                 add_to_history("user", query)
                     
                 try:
@@ -319,12 +326,12 @@ def main():
             
             elif send_query and not query:
                 st.warning("⚠️ กรุณาระบุคำถาม")
-
-        # Handle clear history action
-        if clear_history:
-            st.session_state.chat_history = []
-            clear_qa_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
@@ -366,5 +373,14 @@ def main():
             "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน"
         ), unsafe_allow_html=True)
 
+        # Handle clear history action
+        if clear_history:
+            st.session_state.chat_history = []
+            clear_qa_history()
+            st.rerun()
+    
+    time.sleep(20)
+    save_qa_history(all_history)
+
 if __name__ == "__main__":
     main()