Deploy BioStack RLHF Medical Demo

Dockerfile

@@ -1,30 +1,35 @@
-FROM python:3.11-slim
-
-# Install Node.js for building React app
-RUN apt-get update && apt-get install -y \
-    nodejs \
-    npm \
-    && rm -rf /var/lib/apt/lists/*
+FROM node:20-slim AS frontend-builder
 
-# Set working directory
+# Set working directory for frontend
 WORKDIR /app
 
-# Copy requirements first for better caching
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy package files and install dependencies
+# Copy package files first for better caching
 COPY package*.json ./
 RUN npm install
 
-# Copy all files
+# Copy React source and build
 COPY . .
-
-# Build React app
 RUN npm run build
 
+# Production stage with Python
+FROM python:3.11-slim
+
+# Copy built React app from frontend-builder
+COPY --from=frontend-builder /app/build /app/build
+
+# Copy Python requirements and install
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy Python source code
+COPY server.py ./
+
 # Expose port
 EXPOSE 7860
 
-# Run both frontend and backend
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Run the application
 CMD ["python", "server.py"]

server.py

@@ -373,6 +373,12 @@ def reward_feedback(report: str, score: float) -> str:
 # ─────────────────────────────────────────────────────────────────────────────
 app = FastAPI(title="RLHF Medical Demo")
 
+
+# Mount static files for React app (must come before CORS and routes)
+from fastapi.staticfiles import StaticFiles
+app.mount("/", StaticFiles(directory="build", html=True), name="static")
+
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],