diff --git a/app/packages/embeddings/src/ComputeVisualizationButton.tsx b/app/packages/embeddings/src/ComputeVisualizationButton.tsx
index 4c945307c8..1170bc5312 100644
--- a/app/packages/embeddings/src/ComputeVisualizationButton.tsx
+++ b/app/packages/embeddings/src/ComputeVisualizationButton.tsx
@@ -4,7 +4,7 @@ import { usePanelId } from "@fiftyone/spaces";
import AddIcon from "@mui/icons-material/Add";
import { useTheme } from "@fiftyone/components";
-export default function ComputeVisualizationButton() {
+export default function ComputeVisualizationButton({ uri }) {
const triggerEvent = usePanelEvent();
const theme = useTheme();
const panelId = usePanelId();
@@ -19,7 +19,7 @@ export default function ComputeVisualizationButton() {
onClick={() => {
triggerEvent(panelId, {
params: { delegate: true },
- operator: "@voxel51/brain/compute_visualization",
+ operator: uri,
prompt: true,
});
}}
diff --git a/app/packages/embeddings/src/EmptyEmbeddings.tsx b/app/packages/embeddings/src/EmptyEmbeddings.tsx
index d7c5bfd931..5bd2b119c2 100644
--- a/app/packages/embeddings/src/EmptyEmbeddings.tsx
+++ b/app/packages/embeddings/src/EmptyEmbeddings.tsx
@@ -20,7 +20,7 @@ const useFirstExistingUri = (uris: string[]) => {
export default function EmptyEmbeddings() {
const theme = useTheme();
const secondaryBodyColor = theme.text.secondary;
- const { firstExistingUri: computeVisUri, exists: hasComputeVisualization } =
+ let { firstExistingUri: computeVisUri, exists: hasComputeVisualization } =
useFirstExistingUri([
"@voxel51/brain/compute_visualization",
"@voxel51/operators/compute_visualization",
@@ -33,7 +33,7 @@ export default function EmptyEmbeddings() {
display: "flex",
alignItems: "center",
justifyContent: "center",
- backgroundColor: "#262626",
+ backgroundColor: theme.background.body,
margin: 2,
}}
>
@@ -68,14 +68,14 @@ export default function EmptyEmbeddings() {
{!hasComputeVisualization && }
- {hasComputeVisualization && }
+ {hasComputeVisualization && }
{hasComputeVisualization && (
-
+
>
);