Update components/CustomNode.tsx

components/CustomNode.tsx

@@ -10,7 +10,8 @@ import {
   Mic, Speaker, Radio, Cuboid, Target, Scan,
   AudioLines, FileAudio, Hexagon, Component,
   ScanText, Mountain, Move, Radar, Map, Orbit, Wind,
-  Film, Video, FastForward, Timer, Clapperboard, Merge
+  Film, Video, FastForward, Timer, Clapperboard, Merge,
+  Share2, Atom, Dna, Gamepad2, Sparkles
 } from 'lucide-react';
 import { NodeData, LayerType } from '../types';
 
@@ -65,6 +66,7 @@ const getIcon = (type: LayerType) => {
     case LayerType.WAV2VEC2_ENC: return <FileAudio className="w-4 h-4" />;
     case LayerType.VOCODER: return <Speaker className="w-4 h-4" />;
     case LayerType.AUDIO_EMBEDDING: return <Mic className="w-4 h-4" />;
+    case LayerType.SINC_CONV: return <Activity className="w-4 h-4" />;
 
     // Detection / YOLO
     case LayerType.C2F_BLOCK:
@@ -79,7 +81,8 @@ const getIcon = (type: LayerType) => {
     case LayerType.GAUSSIAN_SPLAT:
     case LayerType.TRIPLANE_ENC: return <Cuboid className="w-4 h-4" />;
     case LayerType.POINTNET_BLOCK:
-    case LayerType.POINT_TRANSFORMER: return <Scan className="w-4 h-4" />;
+    case LayerType.POINT_TRANSFORMER: 
+    case LayerType.MESH_CONV: return <Scan className="w-4 h-4" />;
 
     // OCR
     case LayerType.TPS_TRANSFORM: 
@@ -104,6 +107,39 @@ const getIcon = (type: LayerType) => {
     case LayerType.NON_LOCAL_BLOCK: return <Clapperboard className="w-4 h-4" />;
     case LayerType.MULTIMODAL_FUSION: return <Merge className="w-4 h-4" />;
 
+    // Graph
+    case LayerType.GCN_CONV:
+    case LayerType.GRAPH_SAGE:
+    case LayerType.GAT_CONV:
+    case LayerType.GIN_CONV: return <Share2 className="w-4 h-4" />;
+
+    // Physics
+    case LayerType.NEURAL_ODE:
+    case LayerType.HAMILTONIAN_NN:
+    case LayerType.PINN_LINEAR: return <Atom className="w-4 h-4" />;
+    case LayerType.PROTEIN_FOLDING: return <Dna className="w-4 h-4" />;
+
+    // RL
+    case LayerType.DUELING_HEAD:
+    case LayerType.PPO_HEAD:
+    case LayerType.SAC_HEAD: return <Gamepad2 className="w-4 h-4" />;
+
+    // Spiking
+    case LayerType.LIF_NEURON:
+    case LayerType.SPIKING_LAYER: return <Zap className="w-4 h-4" />;
+
+    // Advanced / Niche
+    case LayerType.CAPSULE:
+    case LayerType.HYPER_NET:
+    case LayerType.MAMBA_BLOCK:
+    case LayerType.RWKV_BLOCK:
+    case LayerType.HOPFIELD:
+    case LayerType.NORMALIZING_FLOW:
+    case LayerType.DNC_MEMORY:
+    case LayerType.ARCFACE:
+    case LayerType.ECHO_STATE: return <Sparkles className="w-4 h-4" />;
+
+
     case LayerType.LSTM:
     case LayerType.GRU: return <GitBranch className="w-4 h-4" />;
     case LayerType.EMBEDDING: return <Database className="w-4 h-4" />;
@@ -188,14 +224,16 @@ const getColor = (type: LayerType) => {
     case LayerType.VOCODER:
     case LayerType.RVC_ENCODER:
     case LayerType.WAV2VEC2_ENC:
-    case LayerType.AUDIO_EMBEDDING: return 'border-indigo-500 shadow-indigo-500/20';
+    case LayerType.AUDIO_EMBEDDING:
+    case LayerType.SINC_CONV: return 'border-indigo-500 shadow-indigo-500/20';
 
     // 3D - Teal theme
     case LayerType.NERF_BLOCK:
     case LayerType.POINTNET_BLOCK:
     case LayerType.POINT_TRANSFORMER:
     case LayerType.TRIPLANE_ENC:
-    case LayerType.GAUSSIAN_SPLAT: return 'border-teal-500 shadow-teal-500/20';
+    case LayerType.GAUSSIAN_SPLAT:
+    case LayerType.MESH_CONV: return 'border-teal-500 shadow-teal-500/20';
 
     // OCR - Yellow/Lime
     case LayerType.TPS_TRANSFORM: 
@@ -220,6 +258,38 @@ const getColor = (type: LayerType) => {
     case LayerType.NON_LOCAL_BLOCK:
     case LayerType.MULTIMODAL_FUSION: return 'border-fuchsia-500 shadow-fuchsia-500/20';
 
+    // Graph - Green
+    case LayerType.GCN_CONV:
+    case LayerType.GRAPH_SAGE:
+    case LayerType.GAT_CONV:
+    case LayerType.GIN_CONV: return 'border-green-500 shadow-green-500/20';
+
+    // Physics - Blue/Gray
+    case LayerType.NEURAL_ODE:
+    case LayerType.PINN_LINEAR:
+    case LayerType.HAMILTONIAN_NN:
+    case LayerType.PROTEIN_FOLDING: return 'border-blue-400 shadow-blue-400/20';
+
+    // Spiking - Electric Yellow
+    case LayerType.LIF_NEURON:
+    case LayerType.SPIKING_LAYER: return 'border-yellow-400 shadow-yellow-400/20';
+
+    // RL - Purple/Red
+    case LayerType.DUELING_HEAD:
+    case LayerType.PPO_HEAD:
+    case LayerType.SAC_HEAD: return 'border-purple-600 shadow-purple-600/20';
+
+    // Advanced - Multi/Rainbow feel (White glow)
+    case LayerType.CAPSULE:
+    case LayerType.HYPER_NET:
+    case LayerType.MAMBA_BLOCK:
+    case LayerType.RWKV_BLOCK:
+    case LayerType.HOPFIELD:
+    case LayerType.NORMALIZING_FLOW:
+    case LayerType.DNC_MEMORY:
+    case LayerType.ARCFACE:
+    case LayerType.ECHO_STATE: return 'border-white shadow-white/30';
+
     case LayerType.CONCAT:
     case LayerType.ADD: return 'border-pink-500 shadow-pink-500/20';
     
@@ -307,4 +377,4 @@ const CustomNode = ({ data, selected }: NodeProps<NodeData>) => {
   );
 };
 
-export default memo(CustomNode);
+export default memo(CustomNode);