@marimo-team/islands 0.19.10-dev9 → 0.19.11-dev0

package/src/plugins/impl/anywidget/AnyWidgetPlugin.tsx

@@ -21,7 +21,6 @@ import { BINDING_MANAGER, WIDGET_DEF_REGISTRY } from "./widget-binding";
 interface Data {
   jsUrl: string;
   jsHash: string;
-  css?: string | null;
 }
 
 type AnyWidgetState = ModelState;
@@ -118,21 +117,18 @@ export const AnyWidgetPlugin = createPlugin<ModelIdRef>("marimo-anywidget")
     z.object({
       jsUrl: z.string(),
       jsHash: z.string(),
-      css: z.string().nullish(),
     }),
   )
   .withFunctions({})
   .renderer((props) => <AnyWidgetSlot {...props} />);
 
 const AnyWidgetSlot = (props: IPluginProps<ModelIdRef, Data>) => {
-  const { css, jsUrl, jsHash } = props.data;
+  const { jsUrl, jsHash } = props.data;
   const { model_id: modelId } = props.value;
   const host = props.host as HTMLElementNotDerivedFromRef;
 
   const { jsModule, error } = useAnyWidgetModule({ jsUrl, jsHash });
 
-  useMountCss(css, host);
-
   if (error) {
     return <ErrorBanner error={error} />;
   }
@@ -160,6 +156,7 @@ const AnyWidgetSlot = (props: IPluginProps<ModelIdRef, Data>) => {
       key={key}
       widget={jsModule.default}
       modelId={modelId}
+      host={host}
     />
   );
 };
@@ -207,11 +204,13 @@ function isAnyWidgetModule(mod: any): mod is { default: AnyWidget } {
 interface Props<T extends AnyWidgetState> {
   widget: AnyWidget<T>;
   modelId: WidgetModelId;
+  host: HTMLElementNotDerivedFromRef;
 }
 
 const LoadedSlot = <T extends AnyWidgetState>({
   widget,
   modelId,
+  host,
 }: Props<T> & { widget: AnyWidget<T> }) => {
   const htmlRef = useRef<HTMLDivElement>(null);
 
@@ -222,6 +221,9 @@ const LoadedSlot = <T extends AnyWidgetState>({
     Logger.error("Model not found for modelId", modelId);
   }
 
+  const css = model?.get("_css");
+  useMountCss(css, host);
+
   useEffect(() => {
     if (!htmlRef.current || !model) {
       return;

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

@@ -31,6 +31,12 @@ vi.mock("@/core/network/requests", () => ({
   }),
 }));
 
+// Mock isStaticNotebook — default to false (normal mode)
+const mockIsStatic = vi.fn().mockReturnValue(false);
+vi.mock("@/core/static/static-state", () => ({
+  isStaticNotebook: () => mockIsStatic(),
+}));
+
 // Helper to create a mock MarimoComm
 function createMockComm<T>() {
   return {
@@ -393,4 +399,51 @@ describe("ModelManager", () => {
 
     expect(BINDING_MANAGER.has(testId)).toBe(false);
   });
+
+  describe("static mode", () => {
+    beforeEach(() => {
+      mockIsStatic.mockReturnValue(true);
+    });
+
+    afterAll(() => {
+      mockIsStatic.mockReturnValue(false);
+    });
+
+    it("should create model with no-op comm in static mode", async () => {
+      await handleWidgetMessage(modelManager, {
+        model_id: testId,
+        message: {
+          method: "open",
+          state: { count: 42 },
+          buffer_paths: [],
+          buffers: [],
+        },
+      });
+
+      const model = await modelManager.get(testId);
+      expect(model.get("count")).toBe(42);
+
+      // save_changes should not call the real request client
+      model.set("count", 100);
+      model.save_changes();
+      expect(mockSendModelValue).not.toHaveBeenCalled();
+    });
+
+    it("should not throw on send in static mode", async () => {
+      await handleWidgetMessage(modelManager, {
+        model_id: testId,
+        message: {
+          method: "open",
+          state: { count: 0 },
+          buffer_paths: [],
+          buffers: [],
+        },
+      });
+
+      const model = await modelManager.get(testId);
+      // send() should silently no-op
+      await expect(model.send({ test: true })).resolves.toBeUndefined();
+      expect(mockSendModelValue).not.toHaveBeenCalled();
+    });
+  });
 });

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

@@ -35,6 +35,7 @@ export interface PlotProps {
   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;
@@ -48,6 +49,7 @@ const EVENT_NAMES = [
   "Relayouting",
   "Selected",
   "Deselect",
+  "Click",
   "SunburstClick",
   "TreemapClick",
 ] as const;

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

@@ -193,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;
@@ -224,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>[] {
@@ -238,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/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]);