# Use an official Python runtime as a parent image
FROM python:3.8

# Set the working directory in the container
WORKDIR $HOME/app
# Install Python dependencies
COPY ./requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r requirements.txt

# Install system dependencies
RUN apt-get update && apt-get install -y \
    build-essential \
    cmake \
    curl \
    ca-certificates \
    gcc \
    git \
    git-lfs \
    wget \
    libpq-dev \
    libsndfile1-dev \
    libgl1 \
    unzip \
    libjpeg-dev \
    libpng-dev \
    libgomp1 \
    && rm -rf /var/lib/apt/lists/* \
    && apt-get clean

WORKDIR $HOME/app
#COPY --chown=user . $HOME/app
# Create .streamlit/config.toml file
RUN mkdir -p /app/.streamlit && \
    echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > /app/.streamlit/config.toml

# Clone the repository
RUN git clone https://github.com/nguyen-brat/text-remove.git $HOME/app

# Set the working directory to the cloned repo
WORKDIR $HOME/app/text-remove
#COPY --chown=user . $HOME/app/text-remove
RUN mkdir -p $HOME/app/text-remove/.streamlit && \
    echo "[server]\nenableXsrfProtection = false\nenableCORS = false" > $HOME/app/text-remove/.streamlit/config.toml

# Set up Craft
WORKDIR $HOME/app/text-remove/CRAFT-pytorch
#COPY --chown=user . $HOME/app/text-remove/CRAFT-pytorch
RUN wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1Jk4eGD7crsqCCg9C9VjCLkMN3ze8kutZ' -O "craft_mlt_25k.pth"
RUN pip install gdown
RUN gdown 1XSaFwBkOaFOdtk4Ane3DFyJGPRw6v5bO

# Set up lama
WORKDIR $HOME/app/text-remove/lama
#COPY --chown=user . $HOME/app/text-remove/lama
RUN curl -LJO https://huggingface.co/smartywu/big-lama/resolve/main/big-lama.zip
RUN unzip big-lama.zip

#RUN chmod 777 $HOME/app/text-remove/target_test
#RUN chmod 777 $HOME/app/text-remove/test_folder

# RUN useradd -m -u 1000 user
# USER user
# ENV HOME /home/user
# ENV PATH $HOME/.local/bin:$PATH

# Set the working directory back to the root of the project
WORKDIR $HOME/app/text-remove

# Make port 7860 available to the world outside this container
EXPOSE 7860

ENV PYTHONPATH=$HOME/app/text-remove

# Run app.py when the container launches
ENTRYPOINT ["streamlit", "run"]
CMD ["app.py", "--server.port=7860", "--server.headless=true", "--server.enableCORS=false", "--server.enableXsrfProtection=false", "--server.fileWatcherType=none"]

#CMD streamlit run app.py \
#    ---server.port=7860 \
#    --server.headless true \
#    --server.enableCORS false \
#    --server.enableXsrfProtection false \
#    --server.fileWatcherType none