improve embeddings cta take 2

voxel51 · Oct 29, 2024 · 9ef86c9 · 9ef86c9
1 parent ee792ee
commit 9ef86c9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git 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
@@ -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() {
           </Grid>
           <Grid item />
           {!hasComputeVisualization && <OSSContent />}
-          {hasComputeVisualization && <TeamsContent />}
+          {hasComputeVisualization && <ComputeVisContent uri={computeVisUri} />}
         </Grid>
       </Box>
       {hasComputeVisualization && (
         <Box
           sx={{
             position: "absolute",
-            bottom: "100px",
+            bottom: "50px",
             display: "flex",
             flexDirection: "column",
             alignItems: "center",
@@ -120,13 +120,13 @@ function OSSContent() {
   );
 }
 
-function TeamsContent() {
+function ComputeVisContent({ uri }) {
   const theme = useTheme();
   const secondaryBodyColor = theme.text.secondary;
   return (
     <>
       <Grid item>
-        <ComputeVisualizationButton />
+        <ComputeVisualizationButton uri={uri} />
       </Grid>
     </>
   );