Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import subprocess
 from streamlit_pdf_viewer import pdf_viewer
 from github import Github
+import shutil  # Make sure this is imported at the top
 
 # --- Configuration ---
 TOPICS_DIR = "topics"
@@ -76,36 +77,47 @@ def push_to_github(local_path, content, commit_message):
     except Exception as e:
         return False, f"GitHub Error: {str(e)}"
 
-# Place this with your other Helper Functions
+
+
 def pull_from_github():
     """
-    Downloads the latest 'topics' folder from GitHub 
-    and overwrites local files.
+    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"])
         
-        # Get all files in the topics directory from GitHub
+        # 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:
-                # It's a file, download and overwrite locally
-                if file_content.path.endswith(".tex"):
-                    # Ensure local directory exists
-                    os.makedirs(os.path.dirname(file_content.path), exist_ok=True)
-                    
-                    # Write content
-                    with open(file_content.path, "wb") as f:
-                        f.write(file_content.decoded_content)
-                    count += 1
+                # 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"Successfully pulled {count} files from GitHub!"
+        return True, f"Clean sync complete! Downloaded {count} files."
     except Exception as e:
         return False, str(e)