Dockerfile

Dockerfile

@@ -1,42 +1,61 @@
 FROM nvidia/cuda:12.2.0-base-ubuntu22.04
-ENV DEBIAN_FRONTEND noninteractive
-
-RUN apt-get update -y && apt-get upgrade -y \ 
- && apt-get install -y libgl1 libgl1-mesa-glx libglib2.0-0 wget git git-lfs python3-pip python-is-python3  snapd curl ca-certificates sudo zip unzip htop bzip2 libx11-6 build-essential libsndfile-dev software-properties-common \
- 
 
-RUN sudo add-apt-repository ppa:savoury1/ffmpeg4 && apt-get update -y && apt-get install -y ffmpeg
+ENV DEBIAN_FRONTEND noninteractive
 
+# Update and install necessary packages
+RUN apt-get update -y && apt-get upgrade -y \
+    && apt-get install -y \
+    libgl1 \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
+    wget \
+    git \
+    git-lfs \
+    python3-pip \
+    python-is-python3 \
+    snapd \
+    curl \
+    ca-certificates \
+    sudo \
+    zip \
+    unzip \
+    htop \
+    bzip2 \
+    libx11-6 \
+    build-essential \
+    libsndfile-dev \
+    software-properties-common \
+    && apt-get clean
+
+# Add FFmpeg PPA and install FFmpeg
+RUN add-apt-repository ppa:savoury1/ffmpeg4 -y \
+    && apt-get update -y \
+    && apt-get install -y ffmpeg \
+    && apt-get clean
+
+# Upgrade pip
 RUN pip3 install --upgrade pip
 
-RUN git clone https://github.com/hellovivian/generative-disco.git && cd generative-disco && pip  install --ignore-installed stable_diffusion_videos
+# Clone the repository and install Python dependencies
+RUN git clone https://github.com/hellovivian/generative-disco.git \
+    && cd generative-disco \
+    && pip install --ignore-installed stable_diffusion_videos 
 
-RUN adduser --disabled-password --gecos '' user && adduser user sudo
-#RUN mkdir /content && chown -R user:user /content
-#WORKDIR /content
-RUN chown -R user:user /generative-disco/static/
+# Add user and give permissions
+RUN adduser --disabled-password --gecos '' user \
+    && adduser user sudo \
+    && chown -R user:user /generative-disco/static/
 
+# Switch to the new user
 USER user
-CMD /bin/bash
-
-RUN echo ffmpeg -version
-
-#RUN echo $PATH
-#RUN export PATH="$HOME/user/.local/bin:$PATH"; echo $PATH
-#RUN echo $PATH
-
-#RUN  pip install numba youtube-dl librosa flask openai moviepy Pillow accelerate \
-# && mkdir ./.cache 
 
+# Set working directory and copy contents
+WORKDIR /generative-disco
+COPY . .
 
+# Expose the port
 EXPOSE 7860
-WORKDIR generative-disco
-COPY . ./generative-disco
-
-
-CMD  [ "python",  "../generative-disco/stable_diffusion_videos/app.py"]
-
-
-#CMD ["python", "../generative-disco/flask_app.py", "--port", "7860", "--host", "0.0.0.0"]
 
+# Run the application
+CMD ["python", "flask_app.py"]