Update app.py

app.py

@@ -4,61 +4,11 @@ import subprocess
 from streamlit_pdf_viewer import pdf_viewer
 from github import Github
 import shutil  # Make sure this is imported at the top
-
-
-def pull_from_github():
-    """
-    1. Wipes the local 'topics' folder.
-    2. Downloads a fresh copy from GitHub.
-    """
-    try:
-        # 1. WIPE LOCAL FOLDER CLEAN
-        if os.path.exists(TOPICS_DIR):
-            shutil.rmtree(TOPICS_DIR)  # Deletes the folder and everything inside
-        
-        # Re-create the empty folder
-        os.makedirs(TOPICS_DIR, exist_ok=True)
-
-        # 2. CONNECT TO GITHUB
-        g = Github(st.secrets["github_token"])
-        repo = g.get_repo(st.secrets["github_repo"])
-        
-        # 3. DOWNLOAD EVERYTHING
-        contents = repo.get_contents(TOPICS_DIR, ref=st.secrets["github_branch"])
-        count = 0
-        
-        while contents:
-            file_content = contents.pop(0)
-            if file_content.type == "dir":
-                contents.extend(repo.get_contents(file_content.path, ref=st.secrets["github_branch"]))
-            else:
-                # Calculate where to save it locally
-                # We strip the TOPICS_DIR prefix from the GitHub path to ensure correct nesting
-                local_path = file_content.path 
-                
-                # Ensure the subfolder exists
-                os.makedirs(os.path.dirname(local_path), exist_ok=True)
-                
-                # Write the file
-                with open(local_path, "wb") as f:
-                    f.write(file_content.decoded_content)
-                count += 1
-                    
-        return True, f"Clean sync complete! Downloaded {count} files."
-    except Exception as e:
-        return False, str(e)
-        
-# Auto-load files if the container restarted
-if not os.path.exists(TOPICS_DIR) or not os.listdir(TOPICS_DIR):
-    print("System restart detected. Auto-pulling files from GitHub...")
-    pull_from_github()
-
 # --- Configuration ---
 TOPICS_DIR = "topics"
 TEMP_DIR = "temp_build"
 os.makedirs(TEMP_DIR, exist_ok=True)
 
-
 st.set_page_config(page_title="Physics Topics", layout="wide")
 
 # ==========================================
@@ -126,7 +76,47 @@ def push_to_github(local_path, content, commit_message):
     except Exception as e:
         return False, f"GitHub Error: {str(e)}"
 
+def pull_from_github():
+    """
+    1. Wipes the local 'topics' folder.
+    2. Downloads a fresh copy from GitHub.
+    """
+    try:
+        # 1. WIPE LOCAL FOLDER CLEAN
+        if os.path.exists(TOPICS_DIR):
+            shutil.rmtree(TOPICS_DIR)  # Deletes the folder and everything inside
+        
+        # Re-create the empty folder
+        os.makedirs(TOPICS_DIR, exist_ok=True)
 
+        # 2. CONNECT TO GITHUB
+        g = Github(st.secrets["github_token"])
+        repo = g.get_repo(st.secrets["github_repo"])
+        
+        # 3. DOWNLOAD EVERYTHING
+        contents = repo.get_contents(TOPICS_DIR, ref=st.secrets["github_branch"])
+        count = 0
+        
+        while contents:
+            file_content = contents.pop(0)
+            if file_content.type == "dir":
+                contents.extend(repo.get_contents(file_content.path, ref=st.secrets["github_branch"]))
+            else:
+                # Calculate where to save it locally
+                # We strip the TOPICS_DIR prefix from the GitHub path to ensure correct nesting
+                local_path = file_content.path 
+                
+                # Ensure the subfolder exists
+                os.makedirs(os.path.dirname(local_path), exist_ok=True)
+                
+                # Write the file
+                with open(local_path, "wb") as f:
+                    f.write(file_content.decoded_content)
+                count += 1
+                    
+        return True, f"Clean sync complete! Downloaded {count} files."
+    except Exception as e:
+        return False, str(e)
 
 
 
@@ -178,7 +168,7 @@ def compile_latex(file_path):
 st.sidebar.title(f"Physics Archive ({current_role.title()})")
 
 # ADMIN ONLY: Sync Button
-if current_role == 'admin':
+if current_role == 'admin' or current_role == 'viewer':
     if st.sidebar.button("🔄 Pull from GitHub"):
         with st.spinner("Downloading latest files..."):
             success, msg = pull_from_github()