AI Image Generation

import { AIImageGeneration } from '@/blocks/ai-tools/ai-image-generation'

export default function Example() {
  const handleGenerate = () => {
    // Call your AI image generation API
    console.log('Starting generation...')
  }

  const handleDownload = (variationIndex: number) => {
    // Download the selected variation
    console.log('Downloading variation:', variationIndex)
  }

  const handleShare = (variationIndex: number) => {
    // Share the selected variation
    console.log('Sharing variation:', variationIndex)
  }

  return (
    <AIImageGeneration
      modelName="DALL-E 3"
      variationCount={4}
      onGenerate={handleGenerate}
      onDownload={handleDownload}
      onShare={handleShare}
      onVariationSelect={(i) => console.log('Selected:', i)}
    />
  )
}