Add web player for audio objects (#3098)

stanford-crfm · Oct 27, 2024 · 2a16c51 · 2a16c51
1 parent a618a01
commit 2a16c51
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/helm-frontend/src/components/MediaObjectDisplay.tsx b/helm-frontend/src/components/MediaObjectDisplay.tsx
@@ -21,6 +21,20 @@ export default function MediaObjectDisplay({ mediaObject }: Props) {
         <br />
       </div>
     );
+  } else if (mediaObject.content_type.includes("audio")) {
+    if (mediaObject.location === undefined) {
+      return null;
+    }
+    const url = getBenchmarkEndpoint(
+      mediaObject.location
+        .replace("benchmark_output/", "")
+        .replace("prod_env/", "../"),
+    );
+    return (
+      <div>
+        <audio controls src={url}></audio>
+      </div>
+    );
   } else {
     if (
       mediaObject.text &&

diff --git a/helm-frontend/src/utils/getBenchmarkEndpoint.ts b/helm-frontend/src/utils/getBenchmarkEndpoint.ts
@@ -1,4 +1,7 @@
 export default function getBenchmarkEndpoint(path: string): string {
+  if (path.startsWith("http://") || path.startsWith("https://")) {
+    return path;
+  }
   return `${window.BENCHMARK_OUTPUT_BASE_URL.replace(/\/$/, "")}/${path.replace(
     /^\//,
     "",