Update components/AIBuilderModal.tsx

components/AIBuilderModal.tsx

@@ -1,3 +1,5 @@
+
+
 import React, { useState, useEffect, useRef } from 'react';
 import { Sparkles, X, MessageSquare, Play, RefreshCw, AlertCircle, HardHat, Search, Wand2, CheckCircle2 } from 'lucide-react';
 import { generateGraphWithAgents, AgentStatus } from '../services/geminiService';
@@ -7,7 +9,7 @@ import { NodeData } from '../types';
 interface AIBuilderModalProps {
   isOpen: boolean;
   onClose: () => void;
-  onApply: (nodes: any[], edges: any[]) => void;
+  onApply: (nodes: any[], edges: any[], logMsg?: string) => void;
   currentNodes: Node<NodeData>[];
 }
 
@@ -43,15 +45,20 @@ const AIBuilderModal: React.FC<AIBuilderModalProps> = ({ isOpen, onClose, onAppl
       
       if (result && result.nodes && result.edges) {
         // Ensure nodes have correct style defaults if missing
-        const processedNodes = result.nodes.map((n: any) => ({
-          ...n,
-          type: 'custom', // enforce custom type
-          data: {
-            ...n.data,
-            // ensure label exists if AI forgot it
-            label: n.data.label || n.data.type 
-          }
-        }));
+        const processedNodes = result.nodes.map((n: any) => {
+            const data = n.data || {};
+            const type = data.type || n.type || 'Identity'; // Fallback
+            return {
+              ...n,
+              type: 'custom',
+              data: {
+                ...data,
+                type: type,
+                label: data.label || n.label || type, // Safe access
+                params: data.params || {}
+              }
+            };
+        });
         
         // Ensure edges have styling
         const processedEdges = result.edges.map((e: any) => ({
@@ -62,7 +69,7 @@ const AIBuilderModal: React.FC<AIBuilderModalProps> = ({ isOpen, onClose, onAppl
 
         // Small delay to let user see "Complete" state
         setTimeout(() => {
-             onApply(processedNodes, processedEdges);
+             onApply(processedNodes, processedEdges, "AI Architect generated new graph.");
              onClose();
              setIsBuilding(false);
         }, 1000);