Merge pull request #31 from Moonlite-Media/mounting-required-folders

.github/workflows/on_push_deploy_to_staging.yaml

@@ -6,7 +6,7 @@ on:
       - master
     paths:
       - ".github/workflows/on_push_deploy_to_staging.yaml"
-      # TODO: Decide if we want to add other file changes here. 
+      # TODO: Decide if we want to add other file changes here.
       # TODO: Decide how we handle updating this forked copy.
 jobs:
   deploy:
@@ -77,6 +77,8 @@ jobs:
             
             # Run the new Docker container
             docker run --gpus all -d -p 7861:7861 --name moonlite-sd \
+              -v /home/ec2-user/apps/stable-diffusion-webui/outputs/img2img-images:/outputs/img2img-images \
+              -v /home/ec2-user/apps/media_root:/home/ec2-user/apps/media_root \
               --env-file /home/${{ secrets.STAGING_EC2_USER }}/.env \
               ${{ secrets.STAGING_ECR_URI }}:latest
             

.github/workflows/on_tag_deploy_to_production.yaml

@@ -74,6 +74,8 @@ jobs:
             
             # Run the new Docker container
             docker run --gpus all -d -p 7861:7861 --name moonlite-sd \
+              -v /home/ec2-user/apps/stable-diffusion-webui/outputs/img2img-images:/outputs/img2img-images \
+              -v /home/ec2-user/apps/media_root:/home/ec2-user/apps/media_root \
               --env-file /home/${{ secrets.PROD_EC2_USER }}/.env \
               ${{ secrets.PROD_ECR_URI }}:latest
             

Dockerfile

@@ -8,6 +8,10 @@ ENV WORKDIR /app
 RUN mkdir -p $WORKDIR
 WORKDIR $WORKDIR
 
+# Create the outputs/img2img-images folder
+RUN mkdir -p /outputs/img2img-images
+RUN mkdir -p /home/ec2-user/apps/media_root
+
 RUN apt-get update
 
 # Install Python, pip, and other necessary dependencies