Update services/geminiService.ts

services/geminiService.ts

@@ -664,7 +664,7 @@ export const generateCodeWithAgents = async (
     promptText: string,
     onStatusUpdate: (status: AgentStatus, log: string) => void,
     model: string = DEFAULT_MODEL
-): Promise<{ code: string, evaluation?: any, initialPatches?: any }> => {
+): Promise<{ code: string, evaluation?: any, initialPatches?: any, autoFixError?: string | null }> => {
   const ai = getAiClient();
   
   const irRepresentation = JSON.stringify({ nodes, edges });
@@ -768,15 +768,20 @@ Return ONLY the final Python code. No markdown backticks.`;
       const evaluation = await evaluateCode(promptText, irRepresentation, finalCode, model);
       
       let initialPatches = null;
+      let autoFixError = null;
       if (evaluation.score < 90 && evaluation.improvement_tasks.length > 0) {
-          onStatusUpdate('refiner', 'Applying automated improvements based on evaluation...');
-          const patchResult = await generateCodePatches(finalCode, evaluation.improvement_tasks.join("\n"), model);
-          finalCode = applyPatches(finalCode, patchResult.patches);
-          initialPatches = patchResult;
+          onStatusUpdate('patcher', 'Applying automated improvements based on evaluation...');
+          try {
+              const patchResult = await generateCodePatches(finalCode, evaluation.improvement_tasks.join("\n"), model);
+              finalCode = applyPatches(finalCode, patchResult.patches);
+              initialPatches = patchResult;
+          } catch(e: any) {
+              autoFixError = e.message || "Failed to generate or apply patches.";
+          }
       }
 
       onStatusUpdate('complete', 'Code generation complete!');
-      return { code: finalCode, evaluation, initialPatches };
+      return { code: finalCode, evaluation, initialPatches, autoFixError };
   } catch(e) {
       throw new Error("Polisher agent failed to generate code.");
   }