xinjjj commited on
Commit
ce60326
·
1 Parent(s): 1303338
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -639,6 +639,7 @@ def download_urdf(asset_dir: str, req: gr.Request) -> str | None:
639
  if not asset_dir or not os.path.isdir(asset_dir):
640
  gr.Warning("Please select an asset first.")
641
  return None
 
642
  ensure_hf_files([f"{_rel_dir(asset_dir)}/**"])
643
  zip_path, uid = _fmt_zip_path(asset_dir, "urdf", req)
644
  _zip_items(zip_path, [(asset_dir, "")])
@@ -652,6 +653,7 @@ def download_usd(asset_dir: str, req: gr.Request) -> str | None:
652
  gr.Warning("Please select an asset first.")
653
  return None
654
 
 
655
  rel = _rel_dir(asset_dir)
656
  stem = _find_urdf_stem(asset_dir)
657
  if stem is not None:
@@ -683,6 +685,7 @@ def download_mjcf(asset_dir: str, req: gr.Request) -> str | None:
683
  gr.Warning("Please select an asset first.")
684
  return None
685
 
 
686
  ensure_hf_files([f"{_rel_dir(asset_dir)}/mjcf/**"])
687
  mjcf_dir = os.path.join(asset_dir, "mjcf")
688
  if not os.path.isdir(mjcf_dir):
 
639
  if not asset_dir or not os.path.isdir(asset_dir):
640
  gr.Warning("Please select an asset first.")
641
  return None
642
+ gr.Info("⏳ Preparing URDF asset for download, please wait...")
643
  ensure_hf_files([f"{_rel_dir(asset_dir)}/**"])
644
  zip_path, uid = _fmt_zip_path(asset_dir, "urdf", req)
645
  _zip_items(zip_path, [(asset_dir, "")])
 
653
  gr.Warning("Please select an asset first.")
654
  return None
655
 
656
+ gr.Info("⏳ Preparing USD asset for download, please wait...")
657
  rel = _rel_dir(asset_dir)
658
  stem = _find_urdf_stem(asset_dir)
659
  if stem is not None:
 
685
  gr.Warning("Please select an asset first.")
686
  return None
687
 
688
+ gr.Info("⏳ Preparing MJCF asset for download, please wait...")
689
  ensure_hf_files([f"{_rel_dir(asset_dir)}/mjcf/**"])
690
  mjcf_dir = os.path.join(asset_dir, "mjcf")
691
  if not os.path.isdir(mjcf_dir):