Update services/geminiService.ts

services/geminiService.ts

@@ -5,24 +5,13 @@ import { LAYER_DEFINITIONS } from '../constants';
 
 // Key Management
 const getEnvKey = () => {
-  // STRICT BYOK (Bring Your Own Key) POLICY for Production:
-  // We ignore environment variables in production to ensure the host's key isn't leaked/used by public visitors.
+  // In Development: Returns key from .env file if available
+  // In Production (Docker): Returns undefined/empty because Dockerfile does not inject the ARG
   // @ts-ignore
-  if (typeof import.meta !== 'undefined' && import.meta.env && import.meta.env.PROD) {
-    console.log("Production mode detected: Enforcing BYOK. Environment API keys are ignored.");
-    return '';
-  }
-
-  // 1. Try Vite standard env var (Matches Dockerfile VITE_ prefix) - Development Only
-  // @ts-ignore - import.meta is standard in Vite but TS might complain without config
   if (typeof import.meta !== 'undefined' && import.meta.env && import.meta.env.VITE_GEMINI_API_KEY) {
     // @ts-ignore
     return import.meta.env.VITE_GEMINI_API_KEY;
   }
-  // 2. Fallback to process.env (Safety check to prevent "process is not defined" crash)
-  if (typeof process !== 'undefined' && process.env && process.env.API_KEY) {
-    return process.env.API_KEY;
-  }
   return '';
 };
 
@@ -146,7 +135,7 @@ export const generateRefinedPrompt = async (nodes: Node<NodeData>[], edges: Edge
         return response.text.trim();
     } catch (error) {
         console.error("Prompt refinement failed:", error);
-        return buildRawPrompt(nodes, edges) + "\n\n(AI Refinement failed, showing raw spec)";
+        throw error; // Re-throw so caller can handle auth errors
     }
 };
 
@@ -185,7 +174,7 @@ export const validateArchitecture = async (nodes: Node<NodeData>[], edges: Edge[
     });
     return response.text;
   } catch (error) {
-    return "Error validating architecture.";
+    throw error;
   }
 }
 
@@ -230,7 +219,7 @@ export const getArchitectureSuggestions = async (nodes: Node<NodeData>[], edges:
     });
     return response.text;
   } catch (error) {
-    return "Error generating suggestions.";
+    throw error;
   }
 }
 
@@ -285,7 +274,7 @@ export const implementArchitectureSuggestions = async (
     const result = JSON.parse(response.text.trim());
     return sanitizeGraph(result);
   } catch (error) {
-    throw new Error("Failed to implement suggestions.");
+    throw new Error("Failed to implement suggestions: " + (error instanceof Error ? error.message : String(error)));
   }
 };
 
@@ -396,7 +385,7 @@ export const generateGraphWithAgents = async (
     });
     draftJsonStr = response.text.trim();
   } catch (e) {
-      throw new Error("Architect agent failed to generate draft.");
+      throw e;
   }
 
   // --- Step 2: Critic ---
@@ -501,7 +490,7 @@ export const generateCodeWithAgents = async (
       });
       draftCode = response.text.trim().replace(/```python/g, '').replace(/```/g, '');
   } catch(e) {
-      throw new Error("Coder agent failed.");
+      throw e;
   }
 
   // --- Step 2: Reviewer (Critic) ---
@@ -549,7 +538,7 @@ export const generateCodeWithAgents = async (
       onStatusUpdate('complete', 'Code generation complete!');
       return finalCode;
   } catch(e) {
-      throw new Error("Polisher agent failed.");
+      throw new Error("Polisher agent failed to generate code.");
   }
 };
 
@@ -589,7 +578,7 @@ export const fixArchitectureErrors = async (
     });
     debugAnalysis = response.text.trim();
   } catch (e) {
-    throw new Error("Debugger agent failed.");
+    throw e;
   }
 
   // --- Step 2: Architect ---
@@ -648,4 +637,4 @@ export const fixArchitectureErrors = async (
   } catch (e) {
     throw new Error("Patcher agent failed to generate valid JSON.");
   }
-};
+};