Update services/geminiService.ts

services/geminiService.ts

@@ -25,7 +25,7 @@ const getAiClient = () => {
 
 const MODEL_NAME = 'gemini-2.5-flash';
 
-export type AgentStatus = 'idle' | 'architect' | 'critic' | 'refiner' | 'complete' | 'error';
+export type AgentStatus = 'idle' | 'architect' | 'critic' | 'refiner' | 'debugger' | 'patcher' | 'complete' | 'error';
 
 /**
  * Internal helper to build the raw specification string.
@@ -476,4 +476,101 @@ export const generateCodeWithAgents = async (
   } catch(e) {
       throw new Error("Polisher agent failed.");
   }
-};
+};
+
+/**
+ * Fix Architecture Errors with Agents (Debugger -> Architect -> Patcher)
+ */
+export const fixArchitectureErrors = async (
+  nodes: Node<NodeData>[],
+  edges: Edge[],
+  errorMsg: string,
+  onStatusUpdate: (status: AgentStatus, log: string) => void
+): Promise<{ nodes: any[], edges: any[] } | null> => {
+  const ai = getAiClient();
+  const graphJson = JSON.stringify({
+    nodes: nodes.map(n => ({ id: n.id, type: n.data.type, label: n.data.label, params: n.data.params, position: n.position })),
+    edges: edges.map(e => ({ source: e.source, target: e.target }))
+  });
+
+  // --- Step 1: Debugger ---
+  onStatusUpdate('debugger', 'Debugger Agent is analyzing the error trace...');
+  
+  const debuggerPrompt = `
+    Role: Senior Systems Debugger.
+    Task: Analyze the architecture graph and the reported error to pinpoint the root cause.
+    
+    Graph: ${graphJson}
+    Error Message: "${errorMsg}"
+    
+    Output a technical analysis of exactly what is wrong (e.g. "Node A connects to Node B but shapes [X] and [Y] are incompatible").
+  `;
+
+  let debugAnalysis = "";
+  try {
+    const response = await ai.models.generateContent({
+      model: MODEL_NAME,
+      contents: debuggerPrompt,
+    });
+    debugAnalysis = response.text.trim();
+  } catch (e) {
+    throw new Error("Debugger agent failed.");
+  }
+
+  // --- Step 2: Architect ---
+  onStatusUpdate('architect', 'Architect Agent is planning the fix...');
+  
+  const architectPrompt = `
+    Role: Solution Architect.
+    Task: Propose a specific fix for the identified issue.
+    
+    Issue Analysis: ${debugAnalysis}
+    
+    Instructions:
+    - Determine if nodes need to be added (e.g. Flatten, Reshape), removed, or reconnected.
+    - Determine if parameters need changing.
+    
+    Output the plan in clear steps.
+  `;
+
+  let fixPlan = "";
+  try {
+    const response = await ai.models.generateContent({
+      model: MODEL_NAME,
+      contents: architectPrompt,
+    });
+    fixPlan = response.text.trim();
+  } catch (e) {
+    fixPlan = "Apply necessary structural corrections.";
+  }
+
+  // --- Step 3: Patcher ---
+  onStatusUpdate('patcher', 'Patcher Agent is applying the fix to the graph...');
+  
+  const patcherPrompt = `
+    Role: DevOps Engineer.
+    Task: Apply the fix to the graph JSON.
+    
+    Current Graph: ${graphJson}
+    Fix Plan: ${fixPlan}
+    
+    Requirements:
+    1. Return the complete, valid JSON with "nodes" and "edges".
+    2. Maintain existing node positions where possible, offset new nodes if added.
+    3. Ensure all LayerTypes are valid.
+    4. Return ONLY raw JSON.
+  `;
+
+  try {
+    const response = await ai.models.generateContent({
+      model: MODEL_NAME,
+      contents: patcherPrompt,
+      config: { responseMimeType: "application/json" }
+    });
+    const finalJson = JSON.parse(response.text.trim());
+    onStatusUpdate('complete', 'Fix applied successfully!');
+    return finalJson;
+  } catch (e) {
+    throw new Error("Patcher agent failed to generate valid JSON.");
+  }
+};