@marimo-team/frontend 0.19.10-dev6 → 0.19.10

package/src/plugins/impl/anywidget/model.ts

@@ -5,6 +5,7 @@ import type { AnyModel } from "@anywidget/types";
 import { debounce } from "lodash-es";
 import type { NotificationMessageData } from "@/core/kernel/messages";
 import { getRequestClient } from "@/core/network/requests";
+import { isStaticNotebook } from "@/core/static/static-state";
 import {
   decodeFromWire,
   serializeBuffersToBase64,
@@ -343,12 +344,14 @@ export async function handleWidgetMessage(
         return;
       }
 
-      modelManager.create(
-        modelId,
-        (signal) =>
-          new Model(
-            stateWithBuffers,
-            {
+      modelManager.create(modelId, (signal) => {
+        // In static exports there is no kernel, so comm calls are no-ops.
+        const comm: MarimoComm<ModelState> = isStaticNotebook()
+          ? {
+              sendUpdate: async () => undefined,
+              sendCustomMessage: async () => undefined,
+            }
+          : {
               async sendUpdate(changeData) {
                 if (signal.aborted) {
                   Logger.debug(
@@ -377,10 +380,10 @@ export async function handleWidgetMessage(
                   buffers: buffers.map(dataViewToBase64),
                 });
               },
-            },
-            signal,
-          ),
-      );
+            };
+
+        return new Model(stateWithBuffers, comm, signal);
+      });
       return;
     }
 

package/src/plugins/impl/chat/ChatPlugin.tsx

@@ -47,6 +47,7 @@ export const ChatPlugin = createPlugin<{ messages: UIMessage[] }>(
       maxHeight: z.number().optional(),
       config: configSchema,
       allowAttachments: z.union([z.boolean(), z.string().array()]),
+      disabled: z.boolean().default(false),
     }),
   )
   .withFunctions<PluginFunctions>({
@@ -76,6 +77,7 @@ export const ChatPlugin = createPlugin<{ messages: UIMessage[] }>(
           showConfigurationControls={props.data.showConfigurationControls}
           maxHeight={props.data.maxHeight}
           allowAttachments={props.data.allowAttachments}
+          disabled={props.data.disabled}
           config={props.data.config}
           get_chat_history={props.functions.get_chat_history}
           delete_chat_history={props.functions.delete_chat_history}

package/src/plugins/impl/chat/chat-ui.tsx

@@ -67,6 +67,7 @@ interface Props extends PluginFunctions {
   showConfigurationControls: boolean;
   maxHeight: number | undefined;
   allowAttachments: boolean | string[];
+  disabled: boolean;
   value: UIMessage[];
   setValue: (messages: UIMessage[]) => void;
   host: HTMLElement;
@@ -450,6 +451,9 @@ export const Chatbot: React.FC<Props> = (props) => {
       <form
         onSubmit={async (evt) => {
           evt.preventDefault();
+          if (props.disabled) {
+            return;
+          }
 
           const fileParts = files
             ? await convertToFileUIPart(files)
@@ -462,7 +466,12 @@ export const Chatbot: React.FC<Props> = (props) => {
           resetInput();
         }}
         ref={formRef}
-        className="flex w-full border-t border-(--slate-6) px-2 py-1 items-center"
+        // biome-ignore lint/a11y/useSemanticElements: inert is used to disable the entire form
+        inert={props.disabled || undefined}
+        className={cn(
+          "flex w-full border-t border-(--slate-6) px-2 py-1 items-center",
+          props.disabled && "opacity-50 cursor-not-allowed",
+        )}
       >
         {props.showConfigurationControls && (
           <ConfigPopup config={config} onChange={setConfig} />

package/src/plugins/impl/data-frames/DataFramePlugin.tsx

@@ -48,6 +48,7 @@ type TableData<T> = T[] | CsvURL;
 interface Data {
   label?: string | null;
   columns: ColumnDataTypes;
+  dataframeName?: string;
   pageSize: number;
   showDownload: boolean;
   lazy: boolean;
@@ -93,6 +94,7 @@ export const DataFramePlugin = createPlugin<S>("marimo-dataframe")
       label: z.string().nullish(),
       pageSize: z.number().default(5),
       showDownload: z.boolean().default(true),
+      dataframeName: z.string().optional(),
       columns: z
         .array(z.tuple([z.string().or(z.number()), z.string(), z.string()]))
         .transform((value) => {
@@ -176,6 +178,7 @@ const EMPTY: Transformations = {
 export const DataFrameComponent = memo(
   ({
     columns,
+    dataframeName,
     pageSize,
     showDownload,
     lazy,
@@ -326,6 +329,7 @@ export const DataFrameComponent = memo(
           fieldTypes={field_types}
           rowHeaders={row_headers || Arrays.EMPTY}
           showDownload={showDownload}
+          downloadFileName={dataframeName}
           download_as={download_as}
           enableSearch={false}
           showFilters={false}

package/src/plugins/impl/plotly/Plot.tsx

@@ -0,0 +1,173 @@
+/* Copyright 2026 Marimo. All rights reserved. */
+
+import type * as PlotlyTypes from "plotly.js";
+// Import the pre-built dist bundle, not the source entry point.
+// The source entry point requires Node.js polyfills (e.g. `buffer/`)
+// that are unavailable in the browser/bundler environment.
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-expect-error — no type declarations for dist path, we use PlotlyTypes above
+import Plotly from "plotly.js/dist/plotly";
+import { useEffect, useRef } from "react";
+
+// Plotly attaches `on` and `removeListener` to the DOM element at runtime.
+// The @types/plotly.js PlotlyHTMLElement type includes `on` and `removeAllListeners`
+// but not the per-handler `removeListener`. We extend the type to include it.
+interface PlotlyElement extends PlotlyTypes.PlotlyHTMLElement {
+  removeListener(event: string, handler: (...args: never[]) => void): void;
+}
+
+export interface Figure {
+  data: PlotlyTypes.Data[];
+  layout: Partial<PlotlyTypes.Layout>;
+  frames: PlotlyTypes.Frame[] | null;
+}
+
+export interface PlotProps {
+  data: PlotlyTypes.Data[];
+  layout: Partial<PlotlyTypes.Layout>;
+  frames?: PlotlyTypes.Frame[];
+  config?: Partial<PlotlyTypes.Config>;
+  className?: string;
+  style?: React.CSSProperties;
+  useResizeHandler?: boolean;
+  divId?: string;
+  onRelayout?: (event: PlotlyTypes.PlotRelayoutEvent) => void;
+  onRelayouting?: (event: PlotlyTypes.PlotRelayoutEvent) => void;
+  onSelected?: (event: PlotlyTypes.PlotSelectionEvent) => void;
+  onDeselect?: () => void;
+  onClick?: (event: PlotlyTypes.PlotMouseEvent) => void;
+  onSunburstClick?: (event: PlotlyTypes.PlotMouseEvent) => void;
+  onTreemapClick?: (event: PlotlyTypes.PlotMouseEvent) => void;
+  onError?: (err: Error) => void;
+}
+
+// Plotly event name convention:
+//   - events are attached as `'plotly_' + name.toLowerCase()`
+//   - react props are `'on' + name`
+const EVENT_NAMES = [
+  "Relayout",
+  "Relayouting",
+  "Selected",
+  "Deselect",
+  "Click",
+  "SunburstClick",
+  "TreemapClick",
+] as const;
+
+type EventName = (typeof EVENT_NAMES)[number];
+
+function propName(event: EventName): keyof PlotProps {
+  return `on${event}` as keyof PlotProps;
+}
+
+function plotlyEventName(event: EventName): string {
+  return `plotly_${event.toLowerCase()}`;
+}
+
+export const Plot = (props: PlotProps) => {
+  const containerRef = useRef<HTMLDivElement>(null);
+
+  const {
+    data,
+    layout,
+    config,
+    frames,
+    className,
+    style,
+    useResizeHandler,
+    divId,
+    onError,
+  } = props;
+
+  // Render / update the plot
+  useEffect(() => {
+    const el = containerRef.current;
+    if (!el) {
+      return;
+    }
+
+    Plotly.react(el, data, layout, config)
+      .then(() => {
+        if (frames && frames.length > 0) {
+          return Plotly.addFrames(el as unknown as PlotlyTypes.Root, frames);
+        }
+      })
+      .catch((error: Error) => {
+        onError?.(error);
+      });
+  }, [data, layout, config, frames, onError]);
+
+  // Sync event handlers
+  useEffect(
+    () => {
+      const el = containerRef.current;
+      if (!el) {
+        return;
+      }
+
+      const plotlyEl = el as unknown as PlotlyElement;
+
+      // eslint-disable-next-line @typescript-eslint/ban-types -- Plotly's event API uses generic function references
+      const attached: {
+        plotlyName: string;
+        handler: (...args: never[]) => void;
+      }[] = [];
+
+      for (const name of EVENT_NAMES) {
+        const handler = props[propName(name)];
+        if (typeof handler === "function") {
+          const plotlyName = plotlyEventName(name);
+          plotlyEl.on(plotlyName as never, handler as never);
+          attached.push({
+            plotlyName,
+            handler: handler as (...args: never[]) => void,
+          });
+        }
+      }
+
+      return () => {
+        for (const { plotlyName, handler } of attached) {
+          plotlyEl.removeListener(plotlyName, handler as never);
+        }
+      };
+      // Re-sync whenever any event handler prop changes
+      // eslint-disable-next-line react-hooks/exhaustive-deps
+    },
+    EVENT_NAMES.map((name) => props[propName(name)]),
+  );
+
+  // Window resize handler
+  useEffect(() => {
+    if (!useResizeHandler) {
+      return;
+    }
+
+    const el = containerRef.current;
+    if (!el) {
+      return;
+    }
+
+    const handler = () => {
+      Plotly.Plots.resize(el as unknown as PlotlyTypes.Root);
+    };
+
+    window.addEventListener("resize", handler);
+    return () => {
+      window.removeEventListener("resize", handler);
+    };
+  }, [useResizeHandler]);
+
+  // Cleanup on unmount
+  useEffect(() => {
+    const el = containerRef.current;
+    return () => {
+      if (el) {
+        Plotly.purge(el as unknown as PlotlyTypes.Root);
+      }
+    };
+  }, []);
+
+  return (
+    <div id={divId} className={className} style={style} ref={containerRef} />
+  );
+};

package/src/plugins/impl/plotly/PlotlyPlugin.tsx

@@ -1,9 +1,10 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import type { Figure, PlotParams } from "react-plotly.js";
+import type * as Plotly from "plotly.js";
 import { z } from "zod";
 import type { IPlugin, IPluginProps, Setter } from "@/plugins/types";
 import { Logger } from "@/utils/Logger";
+import type { Figure } from "./Plot";
 
 import "./plotly.css";
 import "./mapbox.css";
@@ -70,13 +71,8 @@ interface PlotlyPluginProps extends Data {
   host: HTMLElement;
 }
 
-// For whatever reason, the version of vite-rolldown that we are one is not exporting this default export correctly.
-export const LazyPlot = lazy(() =>
-  import("react-plotly.js").then((module) => {
-    return module.default as unknown as {
-      default: React.ComponentType<PlotParams>;
-    };
-  }),
+const LazyPlot = lazy(() =>
+  import("./Plot").then((mod) => ({ default: mod.Plot })),
 );
 
 const SUNBURST_DATA_KEYS: (keyof Plotly.SunburstPlotDatum)[] = [
@@ -176,7 +172,6 @@ export const PlotlyComponent = memo(
             };
           });
         })}
-        // @ts-expect-error We patched this prop here so it doesn't exist in the types
         onTreemapClick={useEvent((evt: Readonly<Plotly.PlotMouseEvent>) => {
           if (!evt) {
             return;
@@ -198,6 +193,24 @@ export const PlotlyComponent = memo(
           }));
         })}
         config={plotlyConfig}
+        onClick={useEvent((evt: Readonly<Plotly.PlotMouseEvent>) => {
+          if (!evt) {
+            return;
+          }
+          // Only handle clicks for chart types where box/lasso selection
+          // (onSelected) doesn't work, such as heatmaps.
+          const isHeatmap = evt.points.some(
+            (point) => point.data?.type === "heatmap",
+          );
+          if (!isHeatmap) {
+            return;
+          }
+          setValue((prev) => ({
+            ...prev,
+            points: extractPoints(evt.points),
+            indices: evt.points.map((point) => point.pointIndex),
+          }));
+        })}
         onSelected={useEvent((evt: Readonly<Plotly.PlotSelectionEvent>) => {
           if (!evt) {
             return;
@@ -215,7 +228,7 @@ export const PlotlyComponent = memo(
         className="w-full"
         useResizeHandler={true}
         frames={figure.frames ?? undefined}
-        onError={useEvent((err) => {
+        onError={useEvent((err: Error) => {
           Logger.error("PlotlyPlugin: ", err);
         })}
       />
@@ -229,6 +242,17 @@ PlotlyComponent.displayName = "PlotlyComponent";
  * instead of the ones that Plotly uses internally,
  * by using the hovertemplate.
  */
+const STANDARD_POINT_KEYS: string[] = [
+  "x",
+  "y",
+  "z",
+  "lat",
+  "lon",
+  "curveNumber",
+  "pointNumber",
+  "pointIndex",
+];
+
 function extractPoints(
   points: Plotly.PlotDatum[],
 ): Record<AxisName, AxisDatum>[] {
@@ -243,6 +267,13 @@ function extractPoints(
     const hovertemplate = Array.isArray(point.data.hovertemplate)
       ? point.data.hovertemplate[0]
       : point.data.hovertemplate;
+
+    // For chart types with standard point keys (e.g. heatmaps),
+    // or when there's no hovertemplate, pick keys directly from the point.
+    if (!hovertemplate || point.data?.type === "heatmap") {
+      return pick(point, STANDARD_POINT_KEYS);
+    }
+
     // Update or create a parser
     parser = parser
       ? parser.update(hovertemplate)

package/src/plugins/impl/plotly/__tests__/parse-from-template.test.ts

@@ -1,4 +1,5 @@
 /* Copyright 2026 Marimo. All rights reserved. */
+import type * as Plotly from "plotly.js";
 import { describe, expect, it } from "vitest";
 import { createParser } from "../parse-from-template";
 

package/src/plugins/impl/plotly/__tests__/usePlotlyLayout.test.ts

@@ -1,7 +1,8 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
-import type { Figure } from "react-plotly.js";
+import type * as Plotly from "plotly.js";
 import { describe, expect, it } from "vitest";
+import type { Figure } from "../Plot";
 import {
   computeLayoutOnFigureChange,
   computeLayoutUpdate,

package/src/plugins/impl/plotly/parse-from-template.ts

@@ -1,6 +1,7 @@
 /* Copyright 2026 Marimo. All rights reserved. */
 
 import { get } from "lodash-es";
+import type * as Plotly from "plotly.js";
 
 export interface PlotlyTemplateParser {
   /**

package/src/plugins/impl/plotly/usePlotlyLayout.ts

@@ -2,9 +2,10 @@
 
 import { usePrevious } from "@uidotdev/usehooks";
 import { isEqual, pick } from "lodash-es";
+import type * as Plotly from "plotly.js";
 import { useEffect, useState } from "react";
-import type { Figure } from "react-plotly.js";
 import { Objects } from "@/utils/objects";
+import type { Figure } from "./Plot";
 
 /**
  * Keys that are preserved across figure updates when set by user interaction.

package/src/theme/ThemeProvider.tsx

@@ -10,8 +10,10 @@ export const ThemeProvider: React.FC<PropsWithChildren> = memo(
     const { theme } = useTheme();
     useLayoutEffect(() => {
       document.body.classList.add(theme, `${theme}-theme`);
+      document.body.dataset.theme = theme;
       return () => {
         document.body.classList.remove(theme, `${theme}-theme`);
+        delete document.body.dataset.theme;
       };
     }, [theme]);