mishig HF Staff commited on
Commit
ee8cc9f
·
verified ·
1 Parent(s): 9ec09fe

Sync from GitHub via hub-sync

Browse files
src/app/[org]/[dataset]/[episode]/episode-viewer.tsx CHANGED
@@ -48,6 +48,7 @@ type ActiveTab =
48
  | "frames"
49
  | "insights"
50
  | "filtering"
 
51
  | "urdf";
52
 
53
  export default function EpisodeViewer({
@@ -548,6 +549,20 @@ function EpisodeViewerInner({
548
  <span className="absolute bottom-0 left-0 right-0 h-0.5 bg-orange-500" />
549
  )}
550
  </button>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
551
  {hasURDFSupport(datasetInfo.robot_type) &&
552
  datasetInfo.codebase_version >= "v3.0" && (
553
  <button
@@ -713,6 +728,38 @@ function EpisodeViewerInner({
713
  </Suspense>
714
  )}
715
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
716
  {activeTab === "urdf" && (
717
  <Suspense fallback={<Loading />}>
718
  <URDFViewer
 
48
  | "frames"
49
  | "insights"
50
  | "filtering"
51
+ | "doctor"
52
  | "urdf";
53
 
54
  export default function EpisodeViewer({
 
549
  <span className="absolute bottom-0 left-0 right-0 h-0.5 bg-orange-500" />
550
  )}
551
  </button>
552
+ <button
553
+ className={`px-6 py-2.5 text-sm font-medium transition-colors relative ${
554
+ activeTab === "doctor"
555
+ ? "text-orange-400"
556
+ : "text-slate-400 hover:text-slate-200"
557
+ }`}
558
+ onClick={() => handleTabChange("doctor")}
559
+ title="Dataset quality diagnostics (powered by lerobot-doctor)"
560
+ >
561
+ Doctor
562
+ {activeTab === "doctor" && (
563
+ <span className="absolute bottom-0 left-0 right-0 h-0.5 bg-orange-500" />
564
+ )}
565
+ </button>
566
  {hasURDFSupport(datasetInfo.robot_type) &&
567
  datasetInfo.codebase_version >= "v3.0" && (
568
  <button
 
728
  </Suspense>
729
  )}
730
 
731
+ {activeTab === "doctor" && (
732
+ <div className="flex flex-col h-full">
733
+ <div className="flex items-center justify-between px-1 pb-2 text-xs text-slate-400">
734
+ <span>
735
+ Dataset quality diagnostics &mdash; powered by{" "}
736
+ <a
737
+ href="https://github.com/jashshah999/lerobot-doctor"
738
+ target="_blank"
739
+ rel="noopener noreferrer"
740
+ className="underline hover:text-slate-200"
741
+ >
742
+ lerobot-doctor
743
+ </a>
744
+ </span>
745
+ <a
746
+ href={`https://jashshah999-lerobot-doctor.hf.space/?dataset=${org}/${dataset}`}
747
+ target="_blank"
748
+ rel="noopener noreferrer"
749
+ className="underline hover:text-slate-200"
750
+ >
751
+ Open in new tab
752
+ </a>
753
+ </div>
754
+ <iframe
755
+ src={`https://jashshah999-lerobot-doctor.hf.space/?dataset=${org}/${dataset}`}
756
+ title="lerobot-doctor"
757
+ className="flex-1 w-full rounded border border-slate-700 bg-white"
758
+ sandbox="allow-scripts allow-same-origin allow-popups allow-forms"
759
+ />
760
+ </div>
761
+ )}
762
+
763
  {activeTab === "urdf" && (
764
  <Suspense fallback={<Loading />}>
765
  <URDFViewer