Update app.py

app.py

@@ -121,10 +121,25 @@ def load_qa_history():
         st.error(f"Error loading QA history: {str(e)}")
         return []
 
-def save_qa_history(history_data):
-    """Save QA history to local JSON file and push to Hugging Face"""
+def save_qa_history(history_entry):
+    """Save QA history entry to local JSON file and push to Hugging Face"""
     try:
-        # Save locally
+        history_file = Path("qa_history.json")
+        
+        # Initialize or load existing history
+        if history_file.exists():
+            with open(history_file, "r", encoding="utf-8") as f:
+                try:
+                    history_data = json.load(f)
+                except json.JSONDecodeError:
+                    history_data = []
+        else:
+            history_data = []
+        
+        # Append new entry
+        history_data.append(history_entry)
+        
+        # Save updated history
         with open("qa_history.json", "w", encoding="utf-8") as f:
             json.dump(history_data, f, ensure_ascii=False, indent=2)
         
@@ -143,21 +158,33 @@ def save_qa_history(history_data):
 def add_to_qa_history(query: str, answer: str):
     """Add new QA pair to history"""
     history_entry = {
-        "timestamp": (datetime.now() + timedelta(hours=6)).isoformat(),
+        "timestamp": (datetime.now() + timedelta(hours=7)).isoformat(),
         "query": query,
         "answer": answer
     }
     
-    # Load existing history
-    history_data = load_qa_history()
-    
-    # Add new entry
-    history_data.append(history_entry)
-    
-    # Save updated history
-    save_qa_history(history_data)
-    return history_data
+    save_qa_history(history_entry)
+    return history_entry
 
+def clear_qa_history():
+    """Clear QA history file"""
+    try:
+        # Write empty list to history file
+        with open("qa_history.json", "w", encoding="utf-8") as f:
+            json.dump([], f, ensure_ascii=False, indent=2)
+        
+        # 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 clearing QA history: {str(e)}")
+        
 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))
@@ -168,6 +195,12 @@ def add_to_history(role: str, message: str):
         user_query = st.session_state.chat_history[-2][1]
         add_to_qa_history(user_query, message)
 
+# Modify the clear history button handler in main():
+if clear_history:
+    st.session_state.chat_history = []
+    clear_qa_history()  # Clear saved history
+    st.rerun()
+    
 def display_chat_history():
     """Display chat history with enhanced styling"""
     for i, (role, message) in enumerate(st.session_state.chat_history):