Update app.py

app.py

@@ -14,13 +14,20 @@ from calendar_rag import (
     PipelineConfig
 )
 
+def get_repo_path():
+    """
+    Get repository path based on the operating system
+    """
+    if platform.system() == 'Windows':
+        return r"D:\Last SWU\swu-chat-bot-project"
+    else:
+        # For Linux/Mac, use the current directory
+        return os.path.join(os.getcwd(), "swu-chat-bot-project")
+        
 # Define repository path
-os.environ["GITHUB_REPO_PATH"] = "D:\Last SWU\swu-chat-bot-project"
+REPO_PATH = get_repo_path()
 GITHUB_REPO_URL = "https://github.com/jirasaksaimekJijo/swu-chat-bot-project.git"
 
-# 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():
@@ -330,29 +337,63 @@ def display_chat_history():
                 </div>
             """, unsafe_allow_html=True)
 
+def ensure_repo_exists():
+    """
+    Ensure the repository exists and is properly initialized
+    """
+    try:
+        repo_path = Path(REPO_PATH)
+        
+        # Create directory if it doesn't exist
+        repo_path.mkdir(parents=True, exist_ok=True)
+        
+        # Initialize repository if .git doesn't exist
+        if not (repo_path / '.git').exists():
+            try:
+                # Initialize new repository
+                repo = Repo.init(str(repo_path))
+                
+                # Add remote
+                try:
+                    origin = repo.remote('origin')
+                except ValueError:
+                    origin = repo.create_remote('origin', GITHUB_REPO_URL)
+                
+                # Set up Git configuration
+                with repo.config_writer() as config:
+                    config.set_value("user", "name", "jirasaksaimekJijo")
+                    config.set_value("user", "email", "[email protected]")
+                
+                print(f"Repository initialized at {repo_path}")
+                return True
+                
+            except Exception as init_error:
+                print(f"Failed to initialize repository: {str(init_error)}")
+                return False
+        else:
+            print(f"Repository already exists at {repo_path}")
+            return True
+            
+    except Exception as e:
+        print(f"Failed to ensure repository exists: {str(e)}")
+        return False
+        
 def initialize_github_sync():
     """
     Initialize GitHub repository connection with proper path verification
     """
     try:
-        # Use the exact path where we know the repository exists
-        repo_path = Path(r"D:\Last SWU\swu-chat-bot-project")
-        print(f"Initializing repo at: {repo_path}")
-        
-        # Verify repository exists
-        if not repo_path.exists():
-            print(f"Repository directory not found at: {repo_path}")
+        # Ensure repository exists
+        if not ensure_repo_exists():
             return False
             
+        repo_path = Path(REPO_PATH)
+        print(f"Initializing repo at: {repo_path}")
+        
         try:
             # Initialize repository
             repo = Repo(str(repo_path))
             
-            # Verify .git directory
-            if not (repo_path / '.git').exists():
-                print(".git directory not found")
-                return False
-                
             # Check remote configuration
             try:
                 origin = repo.remote('origin')
@@ -386,6 +427,10 @@ def initialize_github_sync():
         print(f"Initialization error: {str(e)}")
         return False
 
+# Initialize global state
+if 'all_history' not in st.session_state:
+    st.session_state.all_history = []
+    
 # Add this function to test the connection
 def test_github_connection():
     """
@@ -504,8 +549,8 @@ def main():
         initial_sidebar_state="collapsed"
     )
 
-    # Initialize paths first
-    paths_ok = initialize_paths()
+    # Ensure repository exists and initialize paths
+    repo_exists = ensure_repo_exists()
     
     # Load custom CSS
     load_custom_css()
@@ -518,7 +563,7 @@ def main():
         st.session_state.chat_history = []
     
     if 'github_sync_enabled' not in st.session_state:
-        st.session_state.github_sync_enabled = paths_ok and initialize_github_sync()
+        st.session_state.github_sync_enabled = repo_exists and initialize_github_sync()
 
     # Load QA history at startup
     if 'qa_history_loaded' not in st.session_state:
@@ -549,7 +594,7 @@ def main():
             # Query input section
             st.markdown("""
                 <label for="query_input" style="font-size: 1.2rem; font-weight: 600; margin-bottom: 1rem; display: block;">
-                    <span style="color: #ffffff; border-left: 4px solid #ffffff; padding-left: 0.8rem;">
+                    <span style="color: #000000; border-left: 4px solid #1E3A8A; padding-left: 0.8rem;">
                         โปรดระบุคำถามเกี่ยวกับปฏิทินการศึกษา:
                     </span>
                 </label>
@@ -625,42 +670,43 @@ def main():
             # Clear history
             if clear_history:
                 st.session_state.chat_history = []
-                save_qa_history([])  # Clear saved history
+                clear_qa_history()  # This function handles both local and remote clearing
                 st.rerun()
-
-    # Add this to your main() function in the info_col section
-    with st.expander("🔧 GitHub Connection Status", expanded=True):
-        col1, col2 = st.columns([1, 2])
-        
-        with col1:
-            if st.button("Test Connection"):
-                with st.spinner("Testing connection..."):
-                    if test_github_connection():
-                        st.success("✅ GitHub connection successful!")
-                        st.session_state.github_sync_enabled = True
-                    else:
-                        st.error("❌ GitHub connection failed!")
-                        st.session_state.github_sync_enabled = False
-        
-        with col2:
-            # Display current status
-            status = "🟢 Connected" if st.session_state.github_sync_enabled else "🔴 Disconnected"
-            st.info(f"Current Status: {status}")
-            
-        # Show repository details
-        if st.checkbox("Show Repository Details"):
-            repo_path = Path(r"D:\Last SWU\swu-chat-bot-project")
-            repo = Repo(str(repo_path))
-            
-            st.write("Repository Information:")
-            st.code(f"""
-    Path: {repo_path}
-    Remote URL: {list(repo.remote('origin').urls)[0]}
-    Current Branch: {repo.active_branch.name}
-    Git Status: {repo.git.status()}
-            """)
     
     with info_col:
+        # GitHub Sync Status
+        with st.expander("🔧 GitHub Connection Status", expanded=True):
+            col1, col2 = st.columns([1, 2])
+            
+            with col1:
+                if st.button("Test Connection"):
+                    with st.spinner("Testing connection..."):
+                        if test_github_connection():
+                            st.success("✅ GitHub connection successful!")
+                            st.session_state.github_sync_enabled = True
+                        else:
+                            st.error("❌ GitHub connection failed!")
+                            st.session_state.github_sync_enabled = False
+            
+            with col2:
+                # Display current status
+                status = "🟢 Connected" if st.session_state.github_sync_enabled else "🔴 Disconnected"
+                st.info(f"Current Status: {status}")
+                
+            # Show repository details
+            if st.checkbox("Show Repository Details"):
+                try:
+                    repo = Repo(str(Path(REPO_PATH)))
+                    st.write("Repository Information:")
+                    st.code(f"""
+        Path: {REPO_PATH}
+        Remote URL: {list(repo.remote('origin').urls)[0]}
+        Current Branch: {repo.active_branch.name}
+        Git Status: {repo.git.status()}
+                    """)
+                except Exception as e:
+                    st.error(f"Error accessing repository: {str(e)}")
+
         # System information
         st.markdown("""
             <div style="background-color: #F9FAFB; padding: 1.5rem; border-radius: 12px; margin-bottom: 2rem;">