sneaker_tycoon_shoe_maker

A batch stable diffusion generator

For the purpose of this project, there was a need to generate a large number of shoe combinations. This project is a batch generator that generates a large number of shoe combinations and stores them in a database.

Architecture

The project is made up of 4 parts

• Frontend (svelte/typescript)
• Pocketbase (job database)
• Windmill (job scheduler)
• Stable Diffusion (image generator)

sequenceDiagram
    Front->>Pocketbase: Send in batch generated prompts as jobs 
    Wmill-->>Pocketbase: Get all pending jobs
    Wmill-->>StableDiffusion: Ask for an image generation
    StableDiffusion-->>Wmill: return an image
    Wmill-->>Pocketbase: update job with image
    Pocketbase-->>Front: Fetch and display finished jobs
    Front-->>Pocketbase: Update a job score or delete

Loading

Database schema

classDiagram
    class Color{
        id
        name
        machine_name
    }
    class Material{
        id
        name
        machine_name
    }
        class Form{
        id
        name
        image
        machine_name
    }
        class Jobs{
            id
            prompt
            init_text
            status
            image
            form
            rank
            json
    }
        class settings {
            id
            width
            height
            init_prompt
            ending_prompt
        }
        
        Jobs <|-- Form
        

Loading

Usage

Needs to access the automatic111 stable diffusion api setup instruction

Env Copy the .env.example file to .env and fill in the values

Run docker compose up