Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 from datetime import *
 import subprocess  # Added missing import
 from datetime import datetime
+from huggingface_hub import HfApi
 from calendar_rag import (
     create_default_config,
     AcademicCalendarRAG,
@@ -172,27 +173,6 @@ def initialize_pipeline():
         st.error(f"Error initializing pipeline: {str(e)}")
         return None
 
-def save_feedback_to_json():
-    """Save feedback data to a JSON file"""
-    try:
-        # Get current directory
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        file_path = os.path.join(current_dir, 'feedback_data.json')
-
-        # Save the file
-        with open(file_path, "w", encoding="utf-8") as f:
-            json.dump(
-                st.session_state.feedback_data,
-                f,
-                ensure_ascii=False,
-                indent=2
-            )
-        return file_path
-
-    except Exception as e:
-        st.error(f"Error saving feedback: {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 = {
@@ -203,27 +183,6 @@ def add_feedback(query: str, answer: str, is_correct: bool):
     }
     st.session_state.feedback_data.append(feedback_entry)
     
-    file_path = save_feedback_to_json()
-    if file_path:
-        if save_feedback_to_repo(file_path):
-            st.success("Feedback saved successfully")
-        else:
-            st.warning("Feedback saved locally but not pushed to repository")
-
-def evaluate_answer(query: str, answer: str):
-    """Display evaluation buttons for the answer"""
-    col1, col2, col3 = st.columns([1, 1, 4])
-
-    with col1:
-        if st.button("✅ Correct", key=f"correct_{len(st.session_state.chat_history)}"):
-            add_feedback(query, answer, True)
-            st.success("Feedback recorded")
-
-    with col2:
-        if st.button("❌ Incorrect", key=f"incorrect_{len(st.session_state.chat_history)}"):
-            add_feedback(query, answer, False)
-            st.error("Feedback recorded")
-
 def display_chat_history():
     """Display chat history with enhanced styling"""
     for i, (role, message) in enumerate(st.session_state.chat_history):
@@ -251,6 +210,77 @@ 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:
+        history_file = Path("qa_history.json")
+        if history_file.exists():
+            with open(history_file, "r", encoding="utf-8") as f:
+                return json.load(f)
+        return []
+    except Exception as e:
+        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"""
+    try:
+        # Save locally
+        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
+        api = HfApi()
+        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)}")
+
+def add_to_qa_history(query: str, answer: str):
+    """Add new QA pair to history"""
+    history_entry = {
+        "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
+
+# 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))
+    
+    # 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)
+
+# 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 main():
     # Header with animation
     st.markdown("""