From 835a58c0e40efba761710ac44bedc9c63fd8d896 Mon Sep 17 00:00:00 2001
From: Darrel Pol <poldarrel.dev@gmail.com>
Date: Tue, 5 Nov 2024 21:09:02 -0600
Subject: [PATCH] mounting folders

---
 .github/workflows/on_push_deploy_to_staging.yaml | 6 ++++++
 Dockerfile                                       | 4 ++++
 2 files changed, 10 insertions(+)

diff --git a/.github/workflows/on_push_deploy_to_staging.yaml b/.github/workflows/on_push_deploy_to_staging.yaml
index c879664b6..22a019e30 100644
--- a/.github/workflows/on_push_deploy_to_staging.yaml
+++ b/.github/workflows/on_push_deploy_to_staging.yaml
@@ -6,8 +6,12 @@ on:
       - master
     paths:
       - ".github/workflows/on_push_deploy_to_staging.yaml"
+<<<<<<< Updated upstream
       # TODO: Decide if we want to add other file changes here. 
       # TODO: Decide how we handle updating this forked copy.
+=======
+      # TODO: Decide if we want to add other file changes here.
+>>>>>>> Stashed changes
 jobs:
   deploy:
     runs-on: ubuntu-latest
@@ -77,6 +81,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
             
diff --git a/Dockerfile b/Dockerfile
index 29fd3f838..c3a0fe948 100644
--- a/Dockerfile
+++ b/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