Update Dockerfile

Dockerfile

@@ -6,6 +6,8 @@ RUN cd ui && npm install --ignore-scripts
 FROM node:20-bullseye AS build
 ARG HF_OAUTH_CLIENT_ID=bfa7db67-cf78-4a4b-8232-69d855ea836b
 ARG NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=bfa7db67-cf78-4a4b-8232-69d855ea836b
+ARG HF_OAUTH_REDIRECT_URI=https://multimodalart-ai-toolkit.hf.space/auth/callback
+ARG NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI=https://multimodalart-ai-toolkit.hf.space/auth/callback
 WORKDIR /app
 COPY . .
 COPY --from=deps /app/ui/node_modules ./ui/node_modules
@@ -13,12 +15,16 @@ WORKDIR /app/ui
 ENV NODE_ENV=production \
     NEXT_PUBLIC_DB_MODE=browser \
     HF_OAUTH_CLIENT_ID=${HF_OAUTH_CLIENT_ID} \
-    NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=${NEXT_PUBLIC_HF_OAUTH_CLIENT_ID}
+    NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=${NEXT_PUBLIC_HF_OAUTH_CLIENT_ID} \
+    HF_OAUTH_REDIRECT_URI=${HF_OAUTH_REDIRECT_URI} \
+    NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI=${NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI}
 RUN npx prisma generate && npm run build
 
 FROM node:20-bullseye-slim AS runner
 ARG HF_OAUTH_CLIENT_ID=bfa7db67-cf78-4a4b-8232-69d855ea836b
 ARG NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=bfa7db67-cf78-4a4b-8232-69d855ea836b
+ARG HF_OAUTH_REDIRECT_URI=https://multimodalart-ai-toolkit.hf.space/auth/callback
+ARG NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI=https://multimodalart-ai-toolkit.hf.space/auth/callback
 WORKDIR /app/ui
 
 ENV NODE_ENV=production \
@@ -29,7 +35,12 @@ ENV NODE_ENV=production \
     TRAINING_FOLDER=/data/output \
     HF_JOBS_DEFAULT_HARDWARE=a100-large \
     HF_OAUTH_CLIENT_ID=${HF_OAUTH_CLIENT_ID} \
-    NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=${NEXT_PUBLIC_HF_OAUTH_CLIENT_ID}
+    NEXT_PUBLIC_HF_OAUTH_CLIENT_ID=${NEXT_PUBLIC_HF_OAUTH_CLIENT_ID} \
+    HF_OAUTH_REDIRECT_URI=${HF_OAUTH_REDIRECT_URI} \
+    NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI=${NEXT_PUBLIC_HF_OAUTH_REDIRECT_URI}
+
+# Expect HF_OAUTH_CLIENT_SECRET to be provided at runtime.
+ENV HF_OAUTH_CLIENT_SECRET=""
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
       python3 python3-pip git curl ca-certificates bash tini \
@@ -45,4 +56,4 @@ COPY --from=deps /app/ui/node_modules ./node_modules
 
 EXPOSE 7860
 ENTRYPOINT ["tini", "--"]
-CMD ["npm", "run", "start", "--", "--hostname", "0.0.0.0", "--port", "7860"]
+CMD ["npm", "run", "start", "--", "--hostname", "0.0.0.0", "--port", "7860"]