diff --git a/apps/opik-frontend/src/components/pages/PromptPage/PromptTab/EditPromptDialog.tsx b/apps/opik-frontend/src/components/pages/PromptPage/PromptTab/EditPromptDialog.tsx
index 4ba38a3828..9f2befdda9 100644
--- a/apps/opik-frontend/src/components/pages/PromptPage/PromptTab/EditPromptDialog.tsx
+++ b/apps/opik-frontend/src/components/pages/PromptPage/PromptTab/EditPromptDialog.tsx
@@ -11,7 +11,14 @@ import {
import { Textarea } from "@/components/ui/textarea";
import { Label } from "@/components/ui/label";
import { Button } from "@/components/ui/button";
+import { ToggleGroup, ToggleGroupItem } from "@/components/ui/toggle-group";
import useCreatePromptVersionMutation from "@/api/prompts/useCreatePromptVersionMutation";
+import TextDiff from "@/components/shared/CodeDiff/TextDiff";
+
+enum PROMPT_PREVIEW_MODE {
+ write = "write",
+ diff = "diff",
+}
type EditPromptDialogProps = {
open: boolean;
@@ -29,6 +36,9 @@ const EditPromptDialog: React.FunctionComponent