FROM python:3.10-slim

# Install system dependencies
RUN apt-get update
RUN apt-get install -y libgl1 libglib2.0-0 wget

# Install Python dependencies
RUN pip install ultralytics==8.2.8 flask==3.1.1 torch==2.3.0 torchvision==0.18.0 uvicorn fastapi python-multipart

RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user
WORKDIR $HOME/app

RUN wget https://huggingface.co/ogkalu/comic-speech-bubble-detector-yolov8m/resolve/main/comic-speech-bubble-detector.pt -O model.pt

RUN mkdir output
COPY --chown=user app.py $HOME/app

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]