diff --git a/src/types/nodes.ts b/src/types/nodes.ts
index 85ec4acf..4737f0c1 100644
--- a/src/types/nodes.ts
+++ b/src/types/nodes.ts
@@ -102,7 +102,7 @@ export interface ImageInputNodeData extends BaseNodeData {
   isOptional?: boolean;
   derivedImage?: {
     sourceNodeId: string;
-    operation: "crop" | "multiAngle" | "splitGrid" | "highDefinition" | "outpainting" | "inpainting" | "videoFrame";
+    operation: "crop" | "multiAngle" | "splitGrid" | "highDefinition" | "outpainting" | "inpainting" | "videoFrame" | "annotate";
     task?: DerivedImageTask;
   };
 }
diff --git a/src/utils/derivedImageNodes.ts b/src/utils/derivedImageNodes.ts
index 1cda3f28..a5c31daf 100644
--- a/src/utils/derivedImageNodes.ts
+++ b/src/utils/derivedImageNodes.ts
@@ -30,6 +30,7 @@ export interface CreateDerivedImageNodeOptions {
   dimensions?: { width: number; height: number } | null;
   customTitle: string;
   operation: DerivedImageOperation;
+  size?: DerivedImageNodeSize;
   addNode: AddSmartImageNode;
   onConnect: (connection: Connection) => void;
   selectSingleNode?: (nodeId: string) => void;
@@ -59,6 +60,7 @@ export function createDerivedImageNode({
   dimensions = null,
   customTitle,
   operation,
+  size,
   addNode,
   onConnect,
   selectSingleNode,
@@ -75,7 +77,7 @@ export function createDerivedImageNode({
       sourceNodeId,
       operation,
     },
-  });
+  }, size);
 
   onConnect({
     source: sourceNodeId,
diff --git a/src/utils/smartMediaMode.ts b/src/utils/smartMediaMode.ts
index 13948db8..2828158a 100644
--- a/src/utils/smartMediaMode.ts
+++ b/src/utils/smartMediaMode.ts
@@ -7,8 +7,8 @@ export type SmartImageMode = "neutral" | "image" | "generate";
 export type SmartVideoMode = "neutral" | "video" | "generate";
 export type SmartAudioMode = "empty" | "audio" | "generate";
 
-export type SmartImageAutoTaskOperation = "crop" | "splitGrid" | "multiAngle" | "highDefinition" | "outpainting" | "inpainting" | "videoFrame";
-export const SMART_IMAGE_AUTO_TASK_OPERATIONS: SmartImageAutoTaskOperation[] = ["crop", "splitGrid", "multiAngle", "highDefinition", "outpainting", "inpainting", "videoFrame"];
+export type SmartImageAutoTaskOperation = "crop" | "splitGrid" | "multiAngle" | "highDefinition" | "outpainting" | "inpainting" | "videoFrame" | "annotate";
+export const SMART_IMAGE_AUTO_TASK_OPERATIONS: SmartImageAutoTaskOperation[] = ["crop", "splitGrid", "multiAngle", "highDefinition", "outpainting", "inpainting", "videoFrame", "annotate"];
 export const SMART_IMAGE_LOCAL_DERIVED_TASK_OPERATIONS: SmartImageAutoTaskOperation[] = ["crop", "splitGrid", "videoFrame"];
 export const SMART_IMAGE_AI_DERIVED_TASK_OPERATIONS: SmartImageAutoTaskOperation[] = ["multiAngle", "highDefinition", "outpainting", "inpainting"];
 export const SMART_IMAGE_AI_STATIC_DERIVED_TASK_OPERATIONS: SmartImageAutoTaskOperation[] = ["multiAngle", "outpainting", "inpainting"];