Update app.py

app.py

@@ -22,20 +22,21 @@ def _list_collection_files(pr_number: str) -> Tuple[str, ...]:
     """
     prefix = f"pr-{pr_number}"
     try:
+        # List all files in the repo and filter by prefix
         entries = API.list_repo_tree(
             repo_id=DATASET_ID,
             repo_type="dataset",
-            path=prefix,
             recursive=True,
         )
     except HfHubHTTPError as error:
-        print(f"Failed to list repo tree for {prefix}: {error}")
+        print(f"Failed to list repo tree: {error}")
         return tuple()
 
     files = []
     for entry in entries:
         entry_type = getattr(entry, "type", None)
-        if entry_type == "file" and entry.path.endswith("collection_summary.json"):
+        # Filter by prefix and look for collection_summary.json files
+        if entry_type == "file" and entry.path.startswith(prefix) and entry.path.endswith("collection_summary.json"):
             files.append(entry.path)
     return tuple(files)
 
@@ -174,4 +175,4 @@ with gr.Blocks() as demo:
     refresh_btn.click(refresh_dataset, outputs=status)
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()