FROM python:3.11-slim-buster

ENV DEBIAN_FRONTEND=noninteractive
ENV NUMBA_DISABLE_CACHE=1
WORKDIR /app

RUN apt-get update && apt-get upgrade -y
RUN apt-get install libgl1-mesa-glx ffmpeg -y

RUN mkdir -p /.cache/huggingface/hub && chmod -R 777 /.cache/huggingface/hub
RUN mkdir -p /.config/matplotlib && chmod -R 777 /.config/matplotlib
RUN mkdir -p ./GPT2 && chmod -R 777 ./GPT2
RUN mkdir -p /nltk_data && chmod -R 777 /nltk_data
RUN mkdir -p ./TranslationModel \
    ./CodeGenModel \
    ./ImageGenModel \
    ./MusicGenModel \
    ./SentimentModel \
    ./SummarizationModel \
    ./STTModel \
    ./TTSModel \
    ./ImageTo3DModel \
    ./TextToVideoModel \
    ./GFPGAN \
    ./RestoreFormer \
    ./CodeFormer \
    ./RealESRGAN \
    ./checkpoints && \
    chmod -R 777 ./TranslationModel \
    ./CodeGenModel \
    ./ImageGenModel \
    ./MusicGenModel \
    ./SentimentModel \
    ./SummarizationModel \
    ./STTModel \
    ./TTSModel \
    ./ImageTo3DModel \
    ./TextToVideoModel \
    ./GFPGAN \
    ./RestoreFormer \
    ./CodeFormer \
    ./RealESRGAN \
    ./checkpoints

RUN pip install --no-cache-dir accelerate faker retry asyncio basicsr beautifulsoup4 bs4 opencv-python deep-translator duckduckgo-search fastapi flask flask-cors facexlib ffmpeg-python gfpgan imageio imageio-ffmpeg langdetect librosa nltk numpy Pillow pydub pytorch-lightning PyYAML retry safetensors scikit-learn scipy scikit-image soundfile torch torchaudio torchvision tqdm wget yacs numba

COPY . .

EXPOSE 7860

CMD ["python", "main.py"]