fix: removing replay feature

packages/excalidraw/components/TTDDialog/TextToDiagram.tsx

@@ -28,11 +28,7 @@ import { useChatManagement } from "./hooks/useChatManagement";
 import { TTDChatPanel } from "./components/TTDChatPanel";
 import { TTDPreviewPanel } from "./components/TTDPreviewPanel";
 
-import {
-  addMessages,
-  getLastAssistantMessage,
-  updateAssistantContent,
-} from "./utils/chat";
+import { addMessages, getLastAssistantMessage } from "./utils/chat";
 
 import type { ChatMessageType } from "../Chat";
 
@@ -113,34 +109,6 @@ const TextToDiagramContent = ({
     setChatHistory,
   ]);
 
-  // TODO:: just for testing
-  const onReplay = async () => {
-    const { default: mockChunks } = await import("./mock");
-
-    setChatHistory((prev) => {
-      return updateAssistantContent(prev, {
-        isGenerating: true,
-        content: "",
-      });
-    });
-    for (const chunk of mockChunks) {
-      setChatHistory((prev) => {
-        const lastAssistantMessage = getLastAssistantMessage(prev);
-        return updateAssistantContent(prev, {
-          content: lastAssistantMessage.content + chunk,
-        });
-      });
-      const delay = Math.floor(Math.random() * 5) + 1;
-      await new Promise((resolve) => setTimeout(resolve, delay));
-    }
-
-    setChatHistory((prev) =>
-      updateAssistantContent(prev, {
-        isGenerating: false,
-      }),
-    );
-  };
-
   const onViewAsMermaid = () => {
     if (typeof lastAssistantMessage?.content === "string") {
       saveMermaidDataToStorage(lastAssistantMessage.content);
@@ -284,7 +252,6 @@ const TextToDiagramContent = ({
           error={error}
           loaded={mermaidToExcalidrawLib.loaded}
           onInsert={handleInsertToEditor}
-          onReplay={onReplay}
           isReplayDisabled={lastAssistantMessage?.isGenerating ?? false}
         />
       )}

packages/excalidraw/components/TTDDialog/components/TTDPreviewPanel.tsx

@@ -1,5 +1,3 @@
-import { isDevEnv } from "@excalidraw/common";
-
 import { t } from "../../../i18n";
 import { ArrowRightIcon } from "../../icons";
 
@@ -13,8 +11,6 @@ interface TTDPreviewPanelProps {
   error: Error | null;
   loaded: boolean;
   onInsert: () => void;
-  onReplay: () => void;
-  isReplayDisabled: boolean;
   hideErrorDetails?: boolean;
 }
 
@@ -23,27 +19,16 @@ export const TTDPreviewPanel = ({
   error,
   loaded,
   onInsert,
-  onReplay,
-  isReplayDisabled,
   hideErrorDetails,
 }: TTDPreviewPanelProps) => {
-  const actions: TTDPanelAction[] = [];
-
-  if (isDevEnv()) {
-    actions.push({
-      action: onReplay,
-      label: "Replay",
+  const actions: TTDPanelAction[] = [
+    {
+      action: onInsert,
+      label: t("chat.insert"),
+      icon: ArrowRightIcon,
       variant: "button",
-      disabled: isReplayDisabled,
-    });
-  }
-
-  actions.push({
-    action: onInsert,
-    label: t("chat.insert"),
-    icon: ArrowRightIcon,
-    variant: "button",
-  });
+    },
+  ];
 
   return (
     <TTDDialogPanel

packages/excalidraw/components/TTDDialog/utils/chat.test.ts

@@ -1,7 +1,3 @@
-import { vi } from "vitest";
-
-import type { ChatHistory } from "../../Chat";
-
 import {
   addMessages,
   getLastAssistantMessage,
@@ -10,6 +6,8 @@ import {
   updateAssistantContent,
 } from "./chat";
 
+import type { ChatHistory } from "../../Chat";
+
 describe("chat utils", () => {
   describe("updateAssistantContent", () => {
     it("should update the last assistant message with new payload", () => {