Update app.py

app.py

@@ -9,10 +9,13 @@ from huggingface_hub import list_repo_files
 # Replace this with your actual Hugging Face repo ID
 REPO_ID = "PortPy-Project/PortPy_Dataset"
 
+# Load from private repo using token
+token = os.getenv("HF_TOKEN")
+
 @st.cache_data
 def get_patient_ids():
     # Extract disease site from patient ID prefix (e.g., Lung_Patient_1)
-    file = hf_hub_download(REPO_ID, repo_type="dataset", filename="data_info.jsonl", local_dir="./temp")
+    file = hf_hub_download(REPO_ID, repo_type="dataset", filename="data_info.jsonl", local_dir="./temp", use_auth_token=token)
     with open(file) as f:
         # data_info = json.load(f)
         data_info = [json.loads(line) for line in f]
@@ -34,7 +37,7 @@ def load_all_metadata(disease_site):
         structs = load_structure_metadata(patient_id)
         # Load beam metadata for the patient
         beams = load_beam_metadata(patient_id)
-        planner_file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/PlannerBeams.json", local_dir="./temp")
+        planner_file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/PlannerBeams.json", local_dir="./temp", use_auth_token=token)
         with open(planner_file) as f:
             planner_data = json.load(f)
             planner_beam_ids = planner_data.get("IDs", [])
@@ -48,7 +51,7 @@ def load_all_metadata(disease_site):
 
 @st.cache_data
 def load_structure_metadata(patient_id):
-    file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/StructureSet_MetaData.json", local_dir="./temp")
+    file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/StructureSet_MetaData.json", local_dir="./temp", use_auth_token=token)
     with open(file) as f:
         return json.load(f)
 
@@ -66,7 +69,7 @@ def load_beam_metadata(patient_id):
 
     beam_meta = []
     for path in beam_meta_paths:
-        file = hf_hub_download(REPO_ID, repo_type="dataset", filename=path, local_dir="./temp")
+        file = hf_hub_download(REPO_ID, repo_type="dataset", filename=path, local_dir="./temp", use_auth_token=token)
         with open(file) as f:
             beam_meta.append(json.load(f))
     return beam_meta
@@ -152,7 +155,8 @@ def download_data(repo_id, patient_ids, beam_ids=None, planner_beam_ids=True, ma
                         repo_id=repo_id,
                         repo_type="dataset",
                         filename=hf_path,
-                        local_dir=local_dir
+                        local_dir=local_dir,
+                        use_auth_token=token
                     )
                     downloaded_files.append(local_path)
                     break
@@ -182,7 +186,8 @@ def download_data(repo_id, patient_ids, beam_ids=None, planner_beam_ids=True, ma
                                 repo_id=repo_id,
                                 repo_type="dataset",
                                 filename=hf_path,
-                                local_dir=local_dir
+                                local_dir=local_dir,
+                                use_auth_token=token
                             )
                             downloaded_files.append(local_path)
                             break