Update app.py

app.py

@@ -12,9 +12,12 @@ from calendar_rag import (
     PipelineConfig
 )
 
-all_history = []
+# Define repository path
+os.environ["GITHUB_REPO_PATH"] = r"D:\Last SWU\swu-chat-bot-project"
 
-export GITHUB_REPO_PATH = r"D:\Last SWU\swu-chat-bot-project"
+# Initialize global state
+if 'all_history' not in st.session_state:
+    st.session_state.all_history = []
 
 # Custom CSS for enhanced styling
 def load_custom_css():
@@ -136,17 +139,19 @@ def save_qa_history(history_entry):
         history_file = Path("qa_history.json")
         
         # Initialize or load existing history
+        history_data = []
         if history_file.exists():
-            with open(history_file, "r", encoding="utf-8") as f:
-                try:
+            try:
+                with open(history_file, "r", encoding="utf-8") as f:
                     history_data = json.load(f)
-                except json.JSONDecodeError:
-                    history_data = []
-        else:
-            history_data = []
+            except json.JSONDecodeError:
+                pass
         
-        # Append new entry
-        history_data.append(history_entry)
+        # Append new entry if it's a list of entries
+        if isinstance(history_entry, list):
+            history_data.extend(history_entry)
+        else:
+            history_data.append(history_entry)
         
         # Save updated history locally
         with open(history_file, "w", encoding="utf-8") as f:
@@ -154,12 +159,11 @@ def save_qa_history(history_entry):
         
         # Try to push to GitHub if configured
         try:
-            # Get repository path from environment variable or streamlit secrets
             repo_path = os.getenv('GITHUB_REPO_PATH') or st.secrets.get('GITHUB_REPO_PATH')
             
             if repo_path and Path(repo_path).exists():
                 repo = Repo(repo_path)
-                repo.git.add("qa_history.json")
+                repo.git.add(str(history_file.absolute()))
                 if repo.is_dirty():
                     repo.index.commit("Update QA history")
                     origin = repo.remote(name="origin")
@@ -171,7 +175,6 @@ def save_qa_history(history_entry):
                 
         except Exception as git_error:
             print(f"GitHub sync failed (continuing without GitHub push): {str(git_error)}")
-            # Continue execution - GitHub sync is optional
             
         return True
             
@@ -180,34 +183,44 @@ def save_qa_history(history_entry):
         return False
 
 def add_to_qa_history(query: str, answer: str):
-    """Add new QA pair to history and store it in all_history"""
+    """Add new QA pair to history and store it in session state"""
     history_entry = {
         "timestamp": (datetime.now() + timedelta(hours=7)).isoformat(),
         "query": query,
         "answer": answer
     }
     
-    # Append the new history entry to the all_history list
-    all_history.append(history_entry)
-
+    # Append to session state history
+    st.session_state.all_history.append(history_entry)
     return history_entry
 
 def clear_qa_history():
-    """Clear QA history file"""
+    """Clear QA history and sync with all storages"""
     try:
-        # Write empty list to history file
+        # Clear session state
+        st.session_state.all_history = []
+        
+        # Clear local file
         with open("qa_history.json", "w", encoding="utf-8") as f:
             json.dump([], f, ensure_ascii=False, indent=2)
         
+        # Push to GitHub
+        if st.session_state.github_sync_enabled:
+            save_qa_history([])
+        
         # 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"
-        )
+        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 hf_error:
+            print(f"Hugging Face sync failed: {str(hf_error)}")
+            
     except Exception as e:
         st.error(f"Error clearing QA history: {str(e)}")
         
@@ -217,9 +230,12 @@ def add_to_history(role: str, message: str):
     
     # If this is an assistant response, save the QA pair
     if role == "assistant" and len(st.session_state.chat_history) >= 2:
-        # Get the corresponding user query (previous message)
         user_query = st.session_state.chat_history[-2][1]
-        add_to_qa_history(user_query, message)
+        history_entry = add_to_qa_history(user_query, message)
+        
+        # Save to storage if GitHub sync is enabled
+        if st.session_state.github_sync_enabled:
+            save_qa_history(history_entry)
     
 def display_chat_history():
     """Display chat history with enhanced styling"""