Create docker-compose for ui and ai

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.9
+
+RUN mkdir /app
+WORKDIR /app
+
+RUN apt update
+
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 8000
+
+ENTRYPOINT ["uvicorn", "main:app", "--reload", "--host", "0.0.0.0"]

docker-compose.yml

@@ -0,0 +1,28 @@
+version: '3.3'
+
+services:
+  stability-ai:
+    container_name: sd
+    ports:
+      - '5000:5000'
+    image: 'r8.im/stability-ai/stable-diffusion@sha256:06eb78b36068500c616a7f33c15e6fa40404f8e14b5bfad57ebe0c7fe0f6bdf1'
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+
+  stable-diffusion-ui:
+    container_name: sd-ui
+    ports:
+      - '8000:8000'
+    build:
+      context: .
+      dockerfile: Dockerfile
+    volumes:
+      - .:/app
+    depends_on:
+      - stability-ai
+
+networks:
+  default:

index.html

@@ -92,6 +92,7 @@ function setStatus(statusType, msg, msgType) {
 
 function playSound() {
     const audio = new Audio('/ding.mp3')
+    audio.volume = 0.2
     audio.play()
 }
 
@@ -171,7 +172,7 @@ async function makeImage() {
         console.log('request error', e)
         setStatus('request', 'error', 'error')
     }
-    
+
     btn.innerHTML = 'Make Image'
     btn.disabled = false;
 

main.py

@@ -4,7 +4,7 @@ from pydantic import BaseModel
 
 import requests
 
-LOCAL_SERVER_URL = 'http://localhost:5000'
+LOCAL_SERVER_URL = 'http://stability-ai:5000'
 PREDICT_URL = LOCAL_SERVER_URL + '/predictions'
 
 app = FastAPI()

requirements.txt

@@ -0,0 +1,3 @@
+requests
+fastapi==0.80.0
+uvicorn==0.18.2