Update Dockerfile

Dockerfile

@@ -1,16 +1,37 @@
-# Stage 1: Build React/Vite app
-FROM node:18-alpine AS builder
-WORKDIR /app
-COPY package*.json ./
-RUN npm install
-COPY . .
-ARG GEMINI_API_KEY
-ENV GEMINI_API_KEY=$GEMINI_API_KEY
-RUN npm run build
-
-# Stage 2: Serve with Python
-FROM python:3.11-slim
-WORKDIR /app
-COPY --from=builder /app/dist .
-EXPOSE 7860
-CMD ["python3", "-m", "http.server", "7860", "--bind", "0.0.0.0"]
+# =========================================
+# Stage 1 — Build Vite App
+# =========================================
+FROM node:18-alpine AS builder
+
+WORKDIR /app
+
+# Install deps
+COPY package*.json ./
+RUN npm install
+
+# Copy project files
+COPY . .
+
+# --- IMPORTANT ---
+# Vite only exposes vars starting with VITE_
+ARG VITE_GEMINI_API_KEY
+ARG VITE_OTHER_KEY
+
+# Make them available during build
+ENV VITE_GEMINI_API_KEY=$VITE_GEMINI_API_KEY
+ENV VITE_OTHER_KEY=$VITE_OTHER_KEY
+
+# Build app with env injected
+RUN npm run build
+
+# =========================================
+# Stage 2 — Serve static files with Python
+# =========================================
+FROM python:3.11-slim
+
+WORKDIR /app
+COPY --from=builder /app/dist .
+
+EXPOSE 7860
+
+CMD ["python3", "-m", "http.server", "7860", "--bind", "0.0.0.0"]