prompt multi-select, and other things

machado2 · Jul 4, 2024 · a9a4f2c · a9a4f2c
1 parent 8c02448
commit a9a4f2c
Show file tree

Hide file tree

Showing 9 changed files with 291 additions and 103 deletions.
diff --git a/backend/app/populate_db_models.py b/backend/app/populate_db_models.py
@@ -137,7 +137,8 @@ async def populate_prompts():
 
 
 async def populate_db_models():
+    await populate_prompts()
     await populate_models()
     await populate_styles()
     await populate_categories()
-    await populate_prompts()
+
diff --git a/backend/app/settings.py b/backend/app/settings.py
@@ -14,6 +14,7 @@
 DATABASE_HOST = os.getenv('DATABASE_HOST')
 DATABASE_PORT = os.getenv('DATABASE_PORT')
 DATABASE_PASSWORD = os.getenv('DATABASE_PASSWORD')
+PROMPT_KIND = os.getenv('PROMPT_KIND')
 
 TORTOISE_ORM = {
     "connections": {

diff --git a/backend/app/things_to_draw.py b/backend/app/things_to_draw.py
@@ -8,5 +8,55 @@
     "A monster shaped like a pillow with soft, fluffy fur and eyes that glow a gentle blue, lounging in a child's bedroom.",
     "A spaceship piloted by anthropomorphic cats, navigating an asteroid field with ease while sipping alien beverages.",
     "A steampunk dragon with intricate metallic wings and a furnace-like chest, breathing fire mixed with sparks and gears.",
-    "A mythical creature with the body of a lion, the wings of an eagle, and the tail of a serpent, perched on a rock in a mystical landscape."
+    "A mythical creature with the body of a lion, the wings of an eagle, and the tail of a serpent, perched on a rock in a mystical landscape.",
+    "Sunset over a mountain",
+    "City skyline at night",
+    "Forest in autumn",
+    "Beach with palm trees",
+    "Desert landscape",
+    "Snow-covered village",
+    "Underwater coral reef",
+    "Space with planets and stars",
+    "Medieval castle",
+    "Futuristic city",
+    "Tropical rainforest",
+    "Farm with a barn and animals",
+    "Old European street",
+    "Japanese garden",
+    "Safari with wild animals",
+    "Volcano erupting",
+    "Waterfall in a jungle",
+    "Lighthouse on a cliff",
+    "Winter wonderland",
+    "Ancient ruins",
+    "Carnival with rides",
+    "Fairytale forest",
+    "Garden with flowers",
+    "Zoo with various animals",
+    "Island with a lighthouse",
+    "Ski resort",
+    "Cave interior",
+    "Aquarium with marine life",
+    "Space station",
+    "Pirate ship on the sea",
+    "Western desert town",
+    "Gothic cathedral",
+    "Vineyard in Tuscany",
+    "Zen temple",
+    "Hot air balloon festival",
+    "Grand Canyon",
+    "River with boats",
+    "Fantasy kingdom",
+    "Medieval market",
+    "Amusement park",
+    "Haunted house",
+    "Steampunk city",
+    "Mushroom forest",
+    "Busy airport",
+    "Rocket launch site",
+    "Circus tent",
+    "Ice cave",
+    "Surfing beach",
+    "Train station",
+    "Watermill in the countryside"
 ]
diff --git a/backend/generate-images.py b/backend/generate-images.py
@@ -5,6 +5,7 @@
 from app.db import init as init_db
 from app.exceptions import ImageGenerationException
 from app.models import Style, Prompt, StylePromptImage
+from app.settings import PROMPT_KIND
 
 image_generator = AiHordeImageGenerator()
 
@@ -49,7 +50,7 @@ async def get_missing_images() -> list[QueueItem]:
     from itertools import product
 
     styles = await Style.all()
-    prompts = await Prompt.all()
+    prompts = await Prompt.filter(kind=PROMPT_KIND).all()
     style_prompt_ids = set((spi.style_id, spi.prompt_id) for spi in await StylePromptImage.all())
     missing = [QueueItem(style, prompt) for style, prompt in product(styles, prompts) if
                (style.id, prompt.id) not in style_prompt_ids]

diff --git a/ui/package-lock.json b/ui/package-lock.json
diff --git a/ui/package.json b/ui/package.json
@@ -19,6 +19,7 @@
     "react-dom": "^18.3.1",
     "react-query": "^3.39.3",
     "react-scripts": "5.0.1",
+    "react-select": "^5.8.0",
     "typescript": "^4.9.5",
     "web-vitals": "^2.1.4"
   },