Update constants.ts

constants.ts

@@ -26,7 +26,7 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
       { name: 'in_features', type: 'number', label: 'In Features (Opt)', default: 0, description: "0 = Auto-infer" },
       { name: 'out_features', type: 'number', label: 'Output Features', default: 128 },
       { name: 'bias', type: 'boolean', label: 'Use Bias', default: true },
-      { name: 'activation', type: 'select', label: 'Fused Activation', default: 'None', options: ['None', 'ReLU', 'GELU', 'Sigmoid'] }
+      { name: 'activation', type: 'select', label: 'Fused Activation', default: 'None', options: ['None', 'ReLU', 'GELU', 'Sigmoid', 'Softplus'] }
     ]
   },
   [LayerType.OUTPUT]: {
@@ -62,6 +62,119 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
     ]
   },
 
+  // --- PREPROCESSING (TF/KERAS STYLE) ---
+  [LayerType.RESCALING]: {
+    type: LayerType.RESCALING,
+    label: 'Rescaling',
+    description: 'Scale inputs (x * scale + offset)',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'scale', type: 'number', label: 'Scale', default: 0.00392, description: 'e.g. 1/255' },
+      { name: 'offset', type: 'number', label: 'Offset', default: 0.0 }
+    ]
+  },
+  [LayerType.RESIZING]: {
+    type: LayerType.RESIZING,
+    label: 'Resizing',
+    description: 'Resize image batch',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'height', type: 'number', label: 'Height', default: 224 },
+      { name: 'width', type: 'number', label: 'Width', default: 224 },
+      { name: 'interpolation', type: 'select', label: 'Interpolation', default: 'bilinear', options: ['bilinear', 'nearest', 'bicubic'] }
+    ]
+  },
+  [LayerType.CENTER_CROP]: {
+    type: LayerType.CENTER_CROP,
+    label: 'Center Crop',
+    description: 'Crops the central portion',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'height', type: 'number', label: 'Height', default: 224 },
+      { name: 'width', type: 'number', label: 'Width', default: 224 }
+    ]
+  },
+  [LayerType.NORMALIZATION_LAYER]: {
+    type: LayerType.NORMALIZATION_LAYER,
+    label: 'Normalization (Stat)',
+    description: 'Normalize with mean/variance (Keras style)',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'mean', type: 'string', label: 'Mean', default: '0.0' },
+      { name: 'variance', type: 'string', label: 'Variance', default: '1.0' },
+      { name: 'axis', type: 'number', label: 'Axis', default: -1 }
+    ]
+  },
+  [LayerType.RANDOM_FLIP]: {
+    type: LayerType.RANDOM_FLIP,
+    label: 'Random Flip',
+    description: 'Augmentation: Flips image',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'mode', type: 'select', label: 'Mode', default: 'horizontal', options: ['horizontal', 'vertical', 'horizontal_and_vertical'] },
+      { name: 'seed', type: 'number', label: 'Seed', default: 42 }
+    ]
+  },
+  [LayerType.RANDOM_ROTATION]: {
+    type: LayerType.RANDOM_ROTATION,
+    label: 'Random Rotation',
+    description: 'Augmentation: Rotates image',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'factor', type: 'number', label: 'Factor (0-1)', default: 0.2 },
+      { name: 'fill_mode', type: 'select', label: 'Fill Mode', default: 'reflect', options: ['reflect', 'wrap', 'constant', 'nearest'] }
+    ]
+  },
+  [LayerType.RANDOM_ZOOM]: {
+    type: LayerType.RANDOM_ZOOM,
+    label: 'Random Zoom',
+    description: 'Augmentation: Zooms image',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'height_factor', type: 'number', label: 'Height Factor', default: 0.2 },
+      { name: 'width_factor', type: 'number', label: 'Width Factor', default: 0.2 }
+    ]
+  },
+  [LayerType.RANDOM_CONTRAST]: {
+    type: LayerType.RANDOM_CONTRAST,
+    label: 'Random Contrast',
+    description: 'Augmentation: Adjusts contrast',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'factor', type: 'number', label: 'Factor', default: 0.1 }
+    ]
+  },
+  [LayerType.TEXT_VECTORIZATION]: {
+    type: LayerType.TEXT_VECTORIZATION,
+    label: 'Text Vectorization',
+    description: 'Map text to integer sequences',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'max_tokens', type: 'number', label: 'Max Tokens', default: 10000 },
+      { name: 'output_sequence_length', type: 'number', label: 'Seq Length', default: 256 },
+      { name: 'output_mode', type: 'select', label: 'Mode', default: 'int', options: ['int', 'binary', 'count', 'tf_idf'] }
+    ]
+  },
+  [LayerType.DISCRETIZATION]: {
+    type: LayerType.DISCRETIZATION,
+    label: 'Discretization',
+    description: 'Bucketizes continuous features',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'num_bins', type: 'number', label: 'Num Bins', default: 10 }
+    ]
+  },
+  [LayerType.CATEGORY_ENCODING]: {
+    type: LayerType.CATEGORY_ENCODING,
+    label: 'Category Encoding',
+    description: 'One-hot or Multi-hot encoding',
+    category: 'Preprocessing',
+    parameters: [
+      { name: 'num_tokens', type: 'number', label: 'Num Tokens', default: 10 },
+      { name: 'output_mode', type: 'select', label: 'Mode', default: 'one_hot', options: ['one_hot', 'multi_hot', 'count'] }
+    ]
+  },
+
   // --- GRAPH NEURAL NETWORKS ---
   [LayerType.GCN_CONV]: {
     type: LayerType.GCN_CONV,
@@ -745,6 +858,50 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
     category: 'Merge',
     parameters: []
   },
+  [LayerType.SUBTRACT]: {
+    type: LayerType.SUBTRACT,
+    label: 'Subtract',
+    description: 'Element-wise subtraction',
+    category: 'Merge',
+    parameters: []
+  },
+  [LayerType.MULTIPLY]: {
+    type: LayerType.MULTIPLY,
+    label: 'Multiply',
+    description: 'Element-wise multiplication',
+    category: 'Merge',
+    parameters: []
+  },
+  [LayerType.AVERAGE]: {
+    type: LayerType.AVERAGE,
+    label: 'Average',
+    description: 'Average of inputs',
+    category: 'Merge',
+    parameters: []
+  },
+  [LayerType.MAXIMUM]: {
+    type: LayerType.MAXIMUM,
+    label: 'Maximum',
+    description: 'Element-wise maximum',
+    category: 'Merge',
+    parameters: []
+  },
+  [LayerType.MINIMUM]: {
+    type: LayerType.MINIMUM,
+    label: 'Minimum',
+    description: 'Element-wise minimum',
+    category: 'Merge',
+    parameters: []
+  },
+  [LayerType.DOT]: {
+    type: LayerType.DOT,
+    label: 'Dot Product',
+    description: 'Dot product of two tensors',
+    category: 'Merge',
+    parameters: [
+      { name: 'axes', type: 'number', label: 'Axis', default: 1 }
+    ]
+  },
 
   // --- GENAI / ADVANCED ---
   [LayerType.RMSNORM]: {
@@ -885,6 +1042,29 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
       { name: 'padding', type: 'number', label: 'Padding', default: 1 }
     ]
   },
+  [LayerType.SEPARABLE_CONV2D]: {
+    type: LayerType.SEPARABLE_CONV2D,
+    label: 'Separable Conv2D',
+    description: 'Depthwise Separable Conv (TF/Keras style)',
+    category: 'Convolution',
+    parameters: [
+      { name: 'out_channels', type: 'number', label: 'Filters', default: 64 },
+      { name: 'kernel_size', type: 'number', label: 'Kernel Size', default: 3 },
+      { name: 'depth_multiplier', type: 'number', label: 'Depth Mult', default: 1 },
+      { name: 'stride', type: 'number', label: 'Stride', default: 1 }
+    ]
+  },
+  [LayerType.DEPTHWISE_CONV2D]: {
+    type: LayerType.DEPTHWISE_CONV2D,
+    label: 'Depthwise Conv2D',
+    description: 'Convolution per channel independently',
+    category: 'Convolution',
+    parameters: [
+      { name: 'kernel_size', type: 'number', label: 'Kernel Size', default: 3 },
+      { name: 'depth_multiplier', type: 'number', label: 'Depth Mult', default: 1 },
+      { name: 'stride', type: 'number', label: 'Stride', default: 1 }
+    ]
+  },
   [LayerType.CONV_TRANSPOSE2D]: {
     type: LayerType.CONV_TRANSPOSE2D,
     label: 'ConvTranspose2D',
@@ -1067,6 +1247,20 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
     category: 'Utility',
     parameters: []
   },
+  [LayerType.SOFTPLUS]: {
+    type: LayerType.SOFTPLUS,
+    label: 'Softplus',
+    description: 'Smooth approximation to ReLU',
+    category: 'Utility',
+    parameters: []
+  },
+  [LayerType.SOFTSIGN]: {
+    type: LayerType.SOFTSIGN,
+    label: 'Softsign',
+    description: 'Softsign Activation',
+    category: 'Utility',
+    parameters: []
+  },
   [LayerType.DROPOUT]: {
     type: LayerType.DROPOUT,
     label: 'Dropout',
@@ -1076,6 +1270,15 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
       { name: 'p', type: 'number', label: 'Probability', default: 0.5 }
     ]
   },
+  [LayerType.SPATIAL_DROPOUT]: {
+    type: LayerType.SPATIAL_DROPOUT,
+    label: 'Spatial Dropout',
+    description: 'Drops entire 2D feature maps',
+    category: 'Utility',
+    parameters: [
+      { name: 'p', type: 'number', label: 'Probability', default: 0.2 }
+    ]
+  },
   [LayerType.DROPPATH]: {
     type: LayerType.DROPPATH,
     label: 'DropPath',
@@ -1132,6 +1335,15 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
         { name: 'definition_code', type: 'text', label: 'Python Code (Def)', default: '', description: "Full class definition if strictly custom." }
     ]
   },
+  [LayerType.LAMBDA]: {
+    type: LayerType.LAMBDA,
+    label: 'Lambda (Expr)',
+    description: 'Evaluate custom expression (e.g. x * 2)',
+    category: 'Utility',
+    parameters: [
+        { name: 'expression', type: 'string', label: 'Expression', default: 'x', description: "Python expression using 'x'" }
+    ]
+  },
   [LayerType.IDENTITY]: {
     type: LayerType.IDENTITY,
     label: 'Identity',
@@ -1281,6 +1493,33 @@ export const INITIAL_EDGES = [
 ];
 
 export const TEMPLATES: Record<string, GraphTemplate> = {
+  'mobilenet_block': {
+    id: 'mobilenet_block',
+    name: 'MobileNet Block',
+    description: 'Efficient Separable Conv (TF/Keras style).',
+    nodes: [
+      { id: 'in', type: 'custom', position: {x: 250, y: 0}, data: {label: 'Input', type: LayerType.INPUT, params: {modality: 'Image'}} },
+      // Expansion
+      { id: 'pw1', type: 'custom', position: {x: 250, y: 100}, data: {label: '1x1 Conv (Expand)', type: LayerType.CONV2D, params: {out_channels: 144, kernel_size: 1}} },
+      { id: 'bn1', type: 'custom', position: {x: 250, y: 200}, data: {label: 'BatchNorm', type: LayerType.BATCHNORM, params: {}} },
+      { id: 'rel1', type: 'custom', position: {x: 250, y: 300}, data: {label: 'ReLU6', type: LayerType.RELU, params: {}} }, // Typically ReLU6
+      // Depthwise
+      { id: 'dw', type: 'custom', position: {x: 250, y: 400}, data: {label: 'Depthwise Conv', type: LayerType.DEPTHWISE_CONV2D, params: {kernel_size: 3, stride: 1}} },
+      { id: 'bn2', type: 'custom', position: {x: 250, y: 500}, data: {label: 'BatchNorm', type: LayerType.BATCHNORM, params: {}} },
+      { id: 'rel2', type: 'custom', position: {x: 250, y: 600}, data: {label: 'ReLU6', type: LayerType.RELU, params: {}} },
+      // Pointwise
+      { id: 'pw2', type: 'custom', position: {x: 250, y: 700}, data: {label: '1x1 Conv (Project)', type: LayerType.CONV2D, params: {out_channels: 24, kernel_size: 1}} },
+      { id: 'bn3', type: 'custom', position: {x: 250, y: 800}, data: {label: 'BatchNorm', type: LayerType.BATCHNORM, params: {}} },
+      // Residual
+      { id: 'add', type: 'custom', position: {x: 250, y: 900}, data: {label: 'Add Residual', type: LayerType.ADD, params: {}} },
+    ],
+    edges: [
+      { id: '1', source: 'in', target: 'pw1' }, { id: '2', source: 'pw1', target: 'bn1' }, { id: '3', source: 'bn1', target: 'rel1' },
+      { id: '4', source: 'rel1', target: 'dw' }, { id: '5', source: 'dw', target: 'bn2' }, { id: '6', source: 'bn2', target: 'rel2' },
+      { id: '7', source: 'rel2', target: 'pw2' }, { id: '8', source: 'pw2', target: 'bn3' },
+      { id: '9', source: 'bn3', target: 'add' }, { id: '10', source: 'in', target: 'add' }
+    ]
+  },
   'rvc_voice': {
     id: 'rvc_voice',
     name: 'Voice Cloning (RVC)',