Update services/geminiService.ts

services/geminiService.ts

@@ -1,14 +1,25 @@
-
-
-
 import { GoogleGenAI } from "@google/genai";
 import { Node, Edge } from 'reactflow';
 import { NodeData, LayerType } from '../types';
 import { LAYER_DEFINITIONS } from '../constants';
 
 // Key Management
+const getEnvKey = () => {
+  // 1. Try Vite standard env var (Matches Dockerfile VITE_ prefix)
+  // @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 '';
+};
+
 let userApiKey = typeof window !== 'undefined' ? localStorage.getItem('gemini_api_key') || '' : '';
-const defaultEnvKey = process.env.API_KEY || '';
+const defaultEnvKey = getEnvKey();
 
 export const setUserApiKey = (key: string) => {
   userApiKey = key;
@@ -629,4 +640,4 @@ export const fixArchitectureErrors = async (
   } catch (e) {
     throw new Error("Patcher agent failed to generate valid JSON.");
   }
-};
+};