Update app.py for new hub version and download to local

app.py

@@ -18,7 +18,7 @@ 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", token=token)
+    file = hf_hub_download(REPO_ID, repo_type="dataset", filename="data_info.jsonl", token=token)
     with open(file) as f:
         # data_info = json.load(f)
         data_info = [json.loads(line) for line in f]
@@ -27,6 +27,9 @@ def get_patient_ids():
     df["disease_site"] = df["patient_id"].str.extract(r"^(.*?)_")
     return df
 
+@st.cache_data
+def _list_all_repo_files():
+    return list_repo_files(repo_id=REPO_ID, repo_type="dataset")
 
 @st.cache_data
 def load_all_metadata(disease_site):
@@ -40,7 +43,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", token=token)
+        planner_file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/PlannerBeams.json", token=token)
         with open(planner_file) as f:
             planner_data = json.load(f)
             planner_beam_ids = planner_data.get("IDs", [])
@@ -54,25 +57,24 @@ 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", token=token)
+    file = hf_hub_download(REPO_ID, repo_type="dataset", filename=f"data/{patient_id}/StructureSet_MetaData.json", token=token)
     with open(file) as f:
         return json.load(f)
 
 @st.cache_data
 def load_beam_metadata(patient_id):
-    beam_meta_paths = []
-
-    files = list_repo_files(repo_id=REPO_ID, repo_type="dataset")
+    files = _list_all_repo_files()
     beam_meta_paths = [
         f for f in files
         if f.startswith(f"data/{patient_id}/Beams/Beam_") and f.endswith("_MetaData.json")
     ]
-    # for bid in beam_ids:
-    #     beam_meta_paths.append(f"data/{patient_id}/Beams/Beam_{bid}_MetaData.json")
 
     beam_meta = []
     for path in beam_meta_paths:
-        file = hf_hub_download(REPO_ID, repo_type="dataset", filename=path, local_dir="./temp", token=token)
+        file = hf_hub_download(REPO_ID,
+                               repo_type="dataset",
+                               filename=path,
+                               token=token)   # no local_dir
         with open(file) as f:
             beam_meta.append(json.load(f))
     return beam_meta