add run-norewrite button

ui/src/components/nodes/Code.tsx

@@ -38,6 +38,7 @@ import ContentCopyIcon from "@mui/icons-material/ContentCopy";
 import ContentCutIcon from "@mui/icons-material/ContentCut";
 import Grid from "@mui/material/Grid";
 import PlayCircleOutlineIcon from "@mui/icons-material/PlayCircleOutline";
+import PlayDisabledIcon from "@mui/icons-material/PlayDisabled";
 import DeleteIcon from "@mui/icons-material/Delete";
 import ViewComfyIcon from "@mui/icons-material/ViewComfy";
 import { CopyToClipboard } from "react-copy-to-clipboard";
@@ -228,6 +229,7 @@ function FloatingToolbar({ id }) {
   const devMode = useStore(store, (state) => state.devMode);
   // const pod = useStore(store, (state) => state.pods[id]);
   const wsRun = useStore(store, (state) => state.wsRun);
+  const wsRunNoRewrite = useStore(store, (state) => state.wsRunNoRewrite);
   const clearResults = useStore(store, (s) => s.clearResults);
   // right, bottom
   const [layout, setLayout] = useState("bottom");
@@ -278,6 +280,19 @@ function FloatingToolbar({ id }) {
           </IconButton>
         </Tooltip>
       )}
+      {!isGuest && (
+        <Tooltip title="Run without rewrite">
+          <IconButton
+            size="small"
+            onClick={() => {
+              clearResults(id);
+              wsRunNoRewrite(id);
+            }}
+          >
+            <PlayDisabledIcon fontSize="inherit" />
+          </IconButton>
+        </Tooltip>
+      )}
       <CopyToClipboard
         text="dummy"
         options={{ debug: true, format: "text/plain", onCopy } as any}

ui/src/lib/store/runtimeSlice.tsx

@@ -163,7 +163,8 @@ export interface RuntimeSlice {
   parseAllPods: () => void;
   resolvePod: (id) => void;
   resolveAllPods: () => void;
-  wsRun: (id) => void;
+  wsRun: (id: string) => void;
+  wsRunNoRewrite: (id: string) => void;
   wsInterruptKernel: ({ lang }) => void;
   clearResults: (id) => void;
   clearAllResults: () => void;
@@ -265,6 +266,33 @@ export const createRuntimeSlice: StateCreator<MyState, [], [], RuntimeSlice> = (
     get().resolvePod(id);
     // rewrite the code
     const newcode = rewriteCode(id, get);
+
+    // Run the code in remote kernel.
+    get().setRunning(id);
+    let pod = get().pods[id];
+    get().socket?.send(
+      JSON.stringify({
+        type: "runCode",
+        payload: {
+          lang: pod.lang,
+          code: newcode,
+          raw: true,
+          podId: pod.id,
+          sessionId: get().sessionId,
+        },
+      })
+    );
+  },
+  wsRunNoRewrite: async (id) => {
+    if (!get().socket) {
+      get().addError({
+        type: "error",
+        msg: "Runtime not connected",
+      });
+      return;
+    }
+    const newcode = get().pods[id].content;
+
     // Run the code in remote kernel.
     get().setRunning(id);
     let pod = get().pods[id];