AI Image Generator Development

I am building a website that lets visitors create their own images on demand, and I need the complete AI generation workflow put in place.

Scope
• Integrate an off-the-shelf diffusion or similar model (e.g., Stable Diffusion, DALL·E, or another GPU-friendly alternative) so the site can accept a text prompt and return a downloadable PNG/JPG.
• Ensure every generated image follows a professional and clean aesthetic by default; prompts may vary, but the base model and post-processing should avoid noisy, low-quality artefacts.
• Provide a lightweight, responsive front-end component (HTML/CSS/JS or React—whichever fits neatly with my current stack) to capture the prompt, show generation progress, and display results.
• Build a secure back-end endpoint (Python, Node, or comparable) that queues requests, handles concurrent users, and stores generated files for 24 hours.
• Include prompt history for the session so users can quickly regenerate or edit previous requests.

Deliverables
1. Source code for both front-end and back-end, clearly organised and commented.
2. Read-me explaining local setup, environment variables (API keys, model weights, GPU requirements), and deployment steps to a typical cloud VM or container.
3. Short screen-capture demo showing the feature in action from prompt entry to download.
4. Basic error handling and graceful fall-back when GPU resources are exhausted.

I already host the main site; your module only needs to integrate via a single route and match the existing colour palette. Once everything runs smoothly on a test server, I will move it to production.