Archicise
Exercise

Design an AI Image Generation Platform

Design a platform like Midjourney or DALL-E that generates images from text prompts at scale.

Functional Requirements

  • Generate images from text prompts
  • Support different styles, sizes, and aspect ratios
  • Image editing and inpainting
  • Upscaling and variations
  • User galleries and sharing
  • Content moderation

Non-Functional Requirements

  • Generate images in under 30 seconds
  • Handle 10K+ concurrent generation requests
  • Store and serve billions of generated images
  • 99.9% uptime for generation service
  • Block harmful content generation

Questions to Consider

  • How do you manage GPU resources efficiently?
  • How do you handle the queue during peak load?
  • How do you implement effective content moderation?
Your Solution

Request Queue & Scheduling

Design the request queuing and job scheduling system. Consider priority queues, fair scheduling, rate limiting, and handling burst traffic.