Update app.py

app.py

@@ -124,10 +124,13 @@ 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 GitHub"""
+    """
+    Save QA history entry to local JSON file and push to GitHub with improved error handling
+    and configurable paths.
+    """
     try:
+        # Save to local JSON file first
         history_file = Path("qa_history.json")
         
         # Initialize or load existing history
@@ -143,29 +146,36 @@ def save_qa_history(history_entry):
         # Append new entry
         history_data.append(history_entry)
         
-        # Save updated history
+        # Save updated history locally
         with open(history_file, "w", encoding="utf-8") as f:
             json.dump(history_data, f, ensure_ascii=False, indent=2)
         
-        # GitHub repository path
-        repo_path = Path(r"D:/Last SWU/swu-chat-bot-project")  # Update this path as per your system
-        if not repo_path.exists():
-            raise FileNotFoundError(f"Repository path does not exist: {repo_path}")
-        
-        # Initialize Git repository
-        repo = Repo(repo_path)
-        
-        # Add file to Git, commit, and push
-        repo.git.add("qa_history.json")
-        if repo.is_dirty():
-            repo.index.commit("Update QA history")
-            origin = repo.remote(name="origin")
-            origin.push()
-        else:
-            print("No changes to commit.")
-    
+        # 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")
+                if repo.is_dirty():
+                    repo.index.commit("Update QA history")
+                    origin = repo.remote(name="origin")
+                    origin.push()
+                else:
+                    print("No changes to commit.")
+            else:
+                print("GitHub sync skipped - repository path not configured or doesn't exist")
+                
+        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
+            
     except Exception as e:
-        st.error(f"Error saving QA history to GitHub: {str(e)}")
+        print(f"Error saving QA history: {str(e)}")
+        return False
 
 def add_to_qa_history(query: str, answer: str):
     """Add new QA pair to history and store it in all_history"""
@@ -227,6 +237,40 @@ def display_chat_history():
                 </div>
             """, unsafe_allow_html=True)
 
+def initialize_github_sync():
+    """
+    Initialize GitHub repository connection and validate configuration.
+    Returns True if GitHub sync is properly configured, False otherwise.
+    """
+    try:
+        repo_path = os.getenv('GITHUB_REPO_PATH') or st.secrets.get('GITHUB_REPO_PATH')
+        
+        if not repo_path:
+            print("GitHub repository path not configured")
+            return False
+            
+        repo_dir = Path(repo_path)
+        if not repo_dir.exists():
+            print(f"Repository directory does not exist: {repo_path}")
+            return False
+            
+        # Validate Git repository
+        try:
+            repo = Repo(repo_dir)
+            # Check if remote 'origin' exists
+            if 'origin' not in [remote.name for remote in repo.remotes]:
+                print("Remote 'origin' not configured in repository")
+                return False
+        except Exception as e:
+            print(f"Error validating Git repository: {str(e)}")
+            return False
+            
+        return True
+        
+    except Exception as e:
+        print(f"Error initializing GitHub sync: {str(e)}")
+        return False
+        
 def main():
     # Page config
     st.set_page_config(
@@ -245,6 +289,9 @@ def main():
     
     if 'chat_history' not in st.session_state:
         st.session_state.chat_history = []
+    
+    if 'github_sync_enabled' not in st.session_state:
+        st.session_state.github_sync_enabled = initialize_github_sync()
 
     # Load QA history at startup
     if 'qa_history_loaded' not in st.session_state:
@@ -255,6 +302,7 @@ def main():
     st.markdown("""
         <div style="text-align: center; padding: 2rem 0;">
             <h1>🎓 ระบบค้นหาข้อมูลปฏิทินการศึกษา</h1>
+            <p style="font-size: 1.2rem; color: #666;">บัณฑิตวิทยาลัย มหาวิทยาลัยศรีนครินทรวิโรฒ</p>
         </div>
     """, unsafe_allow_html=True)
 
@@ -263,6 +311,7 @@ def main():
         with st.spinner("กำลังเริ่มต้นระบบ..."):
             st.session_state.pipeline = initialize_pipeline()
 
+    # Layout setup
     chat_col, info_col = st.columns([7, 3])
     
     with chat_col:
@@ -313,7 +362,16 @@ def main():
                 try:
                     with st.spinner("🔍 กำลังค้นหาคำตอบ..."):
                         result = st.session_state.pipeline.process_query(query)
-                        add_to_history("assistant", result["answer"])
+                        answer = result["answer"]
+                        add_to_history("assistant", answer)
+                        
+                        # Save QA pair to history
+                        history_entry = add_to_qa_history(query, answer)
+                        if st.session_state.github_sync_enabled:
+                            if save_qa_history(history_entry):
+                                print("Successfully saved QA history to GitHub")
+                            else:
+                                print("Failed to save QA history to GitHub")
                         
                         with st.expander("📚 แสดงข้อมูลอ้างอิง", expanded=False):
                             for i, doc in enumerate(result["documents"], 1):
@@ -337,7 +395,6 @@ def main():
             
             # Clear history
             if clear_history:
-                time.sleep(20)
                 st.session_state.chat_history = []
                 save_qa_history([])  # Clear saved history
                 st.rerun()
@@ -373,17 +430,21 @@ def main():
                     <span class="status-indicator {}">
                         {} {}
                     </span></p>
+                    <p><strong>💾 สถานะ GitHub Sync:</strong><br>
+                    <span class="status-indicator {}">
+                        {} {}
+                    </span></p>
                 </div>
             </div>
         """.format(
-            (datetime.now() + timedelta(hours=6)).strftime('%Y-%m-%d %H:%M:%S'),
+            (datetime.now() + timedelta(hours=7)).strftime('%Y-%m-%d %H:%M:%S'),
             "status-online" if st.session_state.pipeline else "status-offline",
             "🟢" if st.session_state.pipeline else "🔴",
-            "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน"
+            "พร้อมใช้งาน" if st.session_state.pipeline else "ไม่พร้อมใช้งาน",
+            "status-online" if st.session_state.github_sync_enabled else "status-offline",
+            "🟢" if st.session_state.github_sync_enabled else "🔴",
+            "เชื่อมต่อแล้ว" if st.session_state.github_sync_enabled else "ไม่ได้เชื่อมต่อ"
         ), unsafe_allow_html=True)
-    
-    time.sleep(20)
-    save_qa_history(all_history)
 
 if __name__ == "__main__":
     main()