Fix: Serve frontend from backend at root URL

Dockerfile

@@ -1,18 +1,29 @@
-FROM python:3.12-slim
+# Stage 1: Build Frontend
+FROM node:20 AS frontend-build
+WORKDIR /app/frontend
+COPY frontend/package*.json ./
+RUN npm install
+COPY frontend/ ./
+RUN npm run build
 
+# Stage 2: Backend & Serve
+FROM python:3.12-slim
 WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y build-essential && rm -rf /var/lib/apt/lists/*
 
-# Copy and install dependencies
+# Copy and install backend dependencies
 COPY backend/requirements.txt ./backend/
 RUN pip install --no-cache-dir -r backend/requirements.txt
 
-# Copy everything needed
+# Copy backend and middleware code
 COPY backend/ ./backend/
 COPY middleware/ ./middleware/
 
+# Copy the built frontend from Stage 1
+COPY --from=frontend-build /app/frontend/dist ./frontend/dist
+
 # Hugging Face requires port 7860
 EXPOSE 7860
 

backend/main.py

@@ -4,6 +4,9 @@ import pickle
 import sqlite3
 import pandas as pd
 from fastapi import FastAPI
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
+
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from dotenv import load_dotenv
@@ -255,3 +258,26 @@ def chat_with_bot(req: ChatRequest):
         print(f"Gemini API Error: {e}")
         return {"error": str(e)}
 
+
+# --- SERVE FRONTEND ---
+FRONTEND_PATH = os.path.join(BASE_DIR, "..", "frontend", "dist")
+
+if os.path.exists(FRONTEND_PATH):
+    # Mount the static files (CSS, JS, images)
+    app.mount("/static", StaticFiles(directory=FRONTEND_PATH), name="static")
+
+    # Catch-all route to serve the React index.html for any non-API route
+    @app.get("/{full_path:path}")
+    async def serve_frontend(full_path: str):
+        # If the path looks like an API call, let it 404 or be handled by other routes
+        if full_path.startswith("api/"):
+            return {"detail": "Not Found"}
+        
+        index_file = os.path.join(FRONTEND_PATH, "index.html")
+        if os.path.exists(index_file):
+            return FileResponse(index_file)
+        return {"detail": "Frontend build not found"}
+else:
+    @app.get("/")
+    def read_root():
+        return {"message": "Wafer Defect API is running. Frontend build folder not found."}