Docker

.dockerignore

@@ -0,0 +1,3 @@
+venv
+.venv
+.idea

.gitignore

@@ -2,3 +2,4 @@
 .idea
 __pycache__
 /app/settings.py
+/settings.py

Dockerfile

@@ -0,0 +1,7 @@
+FROM python
+WORKDIR /app
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+COPY . .
+EXPOSE 8000
+CMD ["python", "main.py"]

app/aihorde_image_generator.py

@@ -6,8 +6,8 @@
 import requests
 from PIL import Image
 
-from app.settings import AI_HORDE_API_KEY
-from app.settings import SD_MODEL
+from settings import AI_HORDE_API_KEY
+from settings import SD_MODEL
 
 
 class AiHordeImageGenerator:

app/db.py

@@ -1,6 +1,6 @@
 from tortoise import Tortoise
 
-from app.settings import TORTOISE_ORM
+from settings import TORTOISE_ORM
 
 
 async def init():

app/image_generator.py

@@ -3,7 +3,7 @@
 
 from app.aihorde_image_generator import AiHordeImageGenerator
 from app.article_repository import ImageGenerated, ImageToCreate
-from app.settings import SD_MODEL
+from settings import SD_MODEL
 
 image_generator = AiHordeImageGenerator()
 

docker-compose.yml

@@ -0,0 +1,5 @@
+services:
+  web:
+    build: .
+    ports:
+      - "8000:8000"

main.py

@@ -1,3 +1,4 @@
+import logging
 import os
 from contextlib import asynccontextmanager
 
@@ -13,6 +14,8 @@
 from app.get_image import router as get_image_router
 from app.wait import router as wait_router
 
+logging.basicConfig(level=logging.DEBUG)
+
 
 @asynccontextmanager
 async def lifespan(_app: FastAPI):

pyproject.toml

@@ -1,4 +1,4 @@
 [tool.aerich]
-tortoise_orm = "app.settings.TORTOISE_ORM"
+tortoise_orm = "settings.TORTOISE_ORM"
 location = "./migrations"
 src_folder = "./."

settings.py.example

@@ -0,0 +1,22 @@
+TORTOISE_ORM = {
+    "connections": {
+        "default": {
+            "engine": "tortoise.backends.asyncpg",
+            "credentials": {
+                "database": ***
+                "host": ***
+                "user": ***
+                "password": ***
+            }
+        }
+    },
+    "apps": {
+        "models": {
+            "models": ["app.models", "aerich.models"],
+            "default_connection": "default",
+        }
+    },
+}
+
+AI_HORDE_API_KEY = ***
+SD_MODEL = "AlbedoBase XL (SDXL)"