Update Dockerfile

Dockerfile

@@ -1,11 +1,11 @@
-# --- STAGE 1: Build Rust Backend ---
-FROM rust:1.84-bookworm AS backend_build
+# --- STAGE 1: Build Rust Backend (Using Nightly) ---
+FROM rustlang/rust:nightly-bookworm AS backend_build
 WORKDIR /app
 RUN apt-get update && apt-get install -y pkg-config libssl-dev
 COPY Cargo.toml ./
-# Build dependencies only (caching)
+# Build dependencies
 RUN mkdir src && echo "fn main() {}" > src/main.rs && cargo build --release
-# Build actual source
+# Build source
 COPY src ./src
 RUN touch src/main.rs && cargo build --release
 
@@ -19,7 +19,6 @@ RUN npm run build
 
 # --- STAGE 3: Final Runtime (Nix + Agent) ---
 FROM debian:bookworm-slim
-# Install system tools and browser libs
 RUN apt-get update && apt-get install -y \
     curl xz-utils wget ca-certificates procps \
     libssl3 libasound2 libatk1.0-0 libc6 libcairo2 libcups2 \
@@ -31,30 +30,19 @@ RUN apt-get update && apt-get install -y \
     libxrandr2 libxrender1 libxss1 libxtst6 \
     && rm -rf /var/lib/apt/lists/*
 
-# Setup non-root user for Hugging Face
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user
 WORKDIR /home/user
 
-# Install Nix in single-user mode
+# Install Nix
 RUN curl -L https://nixos.org/nix/install | sh -s -- --no-daemon
 ENV PATH="/home/user/.nix-profile/bin:${PATH}"
 ENV NIX_PATH=nixpkgs=https://github.com/NixOS/nixpkgs/archive/nixos-unstable.tar.gz
 
-# Copy the Rust binary from STAGE 1
 COPY --from=backend_build /app/target/release/polymorphic-agent .
-
-# Copy the Frontend dist from STAGE 2
-# Axum will serve this folder
 COPY --from=frontend_build /app/frontend/dist ./frontend/dist
-
-# Create dynamic UI folder for the agent to write to
 RUN mkdir -p ./frontend/src/dynamic
 
-# Set environment variables
-ENV PORT=7860
 EXPOSE 7860
-
-# Start the agent
 CMD ["./polymorphic-agent"]