FROM python:3.11-slim as backend-builder

WORKDIR /app
COPY backend/requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

FROM node:20-slim as frontend-builder
WORKDIR /app/frontend
COPY frontend/package.json ./
COPY frontend/package-lock.json* ./
RUN npm install
COPY frontend/ ./
RUN npm run build

FROM python:3.11-slim
WORKDIR /app

# Copy backend dependencies
COPY --from=backend-builder /usr/local/lib/python3.11/site-packages/ /usr/local/lib/python3.11/site-packages/
COPY --from=backend-builder /usr/local/bin/ /usr/local/bin/

# Install extra system dependencies for PyMuPDF (if needed)
RUN apt-get update && apt-get install -y --no-install-recommends \
    mupdf mupdf-tools \
    && rm -rf /var/lib/apt/lists/*

# Copy backend code
COPY backend/ /app/backend/

# Copy built frontend
COPY --from=frontend-builder /app/frontend/dist /app/frontend/dist

# Setup FastAPI to serve static files
RUN mkdir -p /app/static
RUN cp -r /app/frontend/dist/* /app/static/

# Environment variables
ENV PYTHONUNBUFFERED=1
ENV PORT=7860
ENV UPLOAD_DIR=/app/data/uploads
ENV CHROMA_DB_PATH=/app/data/chroma_db

# Create data directories with permissions for Hugging Face Spaces
RUN mkdir -p /app/data/uploads /app/data/chroma_db && \
    chmod -R 777 /app/data

# EXPOSE port for Hugging Face
EXPOSE 7860

# Override main.py to mount static files
RUN echo '\nfrom fastapi.staticfiles import StaticFiles\nfrom fastapi.responses import FileResponse\nimport os\n\n# Mount frontend static files\napp.mount("/assets", StaticFiles(directory="static/assets"), name="assets")\n\n@app.get("/{full_path:path}")\nasync def serve_frontend(full_path: str):\n    # If the path points to an actual file in static, serve it\n    static_file_path = os.path.join("static", full_path)\n    if os.path.isfile(static_file_path):\n        return FileResponse(static_file_path)\n    # Otherwise, fallback to index.html for React Router\n    return FileResponse("static/index.html")\n' >> /app/backend/main.py

# Set PYTHONPATH so Python can find modules like config in /app/backend
ENV PYTHONPATH=/app/backend
WORKDIR /app

# Run Uvicorn
CMD ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "7860"]