FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime

WORKDIR /app

# Install dependencies
COPY requirements-base.txt requirements-cuda.txt ./
RUN pip install --no-cache-dir --timeout 300 -r requirements-cuda.txt

# App code and assets
COPY api.py frontend.py run_app_wo_docker.sh ./
COPY img/ ./img/

RUN chmod +x run_app_wo_docker.sh

# Hugging Face Spaces port 7860
EXPOSE 7860

# Backend (FastAPI) runs on 8000; frontend (Streamlit) on 7860
CMD ["./run_app_wo_docker.sh"]