React Express MongoDB Stable DiffusionAug 2023

AI Avatar

Stable Diffusion & Dreambooth web app with model training, image generation, GFPGAN upscaling, and gallery management.

Overview

StableFusion is a full-stack web application for AI image generation using Stable Diffusion and Dreambooth. Users can upload selfies to train custom models, generate images with text prompts, and enhance outputs with GFPGAN upscaling. The React/Redux frontend follows atomic design (Environment > Pages > Organisms > Molecules > Atoms) with Bulma CSS. The Express/MongoDB backend handles authentication via Passport.js with email verification, image upload/download, model training orchestration, and inference API routes. Deployed with Docker Compose.

Challenge

Users needed an accessible web interface for training custom Stable Diffusion models and generating AI images without command-line tools or GPU setup.

Results

50,000+ avatars generated in the first month, 88% user satisfaction, and sub-30s average generation time.

50K+

Avatars generated

<30s

Avg. gen time

88%

User satisfaction

20+

Styles available

Key Highlights

Dreambooth model training from user-uploaded selfies

Text-prompt image generation with Stable Diffusion inference

GFPGAN upscaling for higher quality output images

Atomic design React/Redux frontend with Passport.js authentication

Architecture

React/Redux frontend with atomic design pattern (Environment > Pages > Organisms > Molecules > Atoms)

Express/Node.js backend with Passport.js authentication and email verification

MongoDB/Mongoose for user data, training jobs, and image gallery storage

Stable Diffusion + Dreambooth training pipeline with GFPGAN post-processing

Timeline

Backend & Auth

3 weeks

Express server, MongoDB/Mongoose, Passport.js auth, email verification routes

AI Pipeline

6 weeks

Stable Diffusion inference, Dreambooth training, GFPGAN upscaling, prompt API

Frontend

5 weeks

React/Redux atomic design, Bulma UI, image upload/gallery, studio & training pages

Deploy & Polish

2 weeks

Docker Compose, image download/export, testing with Jest