FROM python:3.10

WORKDIR /app

COPY requirements.txt .
RUN pip install -r requirements.txt

RUN mkdir /.cache && chmod -R 777 /.cache
RUN mkdir .chroma && chmod -R 777 .chroma

COPY . .

RUN chmod -R 777 /app

EXPOSE 7860

CMD ["python", "server.py", "--cpu", "--share", "--secure", "--enable-modules=summarize,classify,silero-tts,edge-tts,chromadb", "--classification-model=joeddav/distilbert-base-uncased-go-emotions-student", "--summarization-model=slauw87/bart_summarisation"]