AI Avatar API — Add AI Portrait Generation to Your Product

// User requests an anime-style avatar
async function generateAvatar(userId, style) {
  const prompts = {
    anime: 'anime portrait, vibrant colors, detailed eyes --ar 1:1 --niji 6',
    professional: 'professional headshot, soft lighting, neutral background --ar 1:1',
    fantasy: 'fantasy character portrait, epic art, intricate armor --ar 1:1',
  };

  const response = await fetch(
    'https://api.midjourney-api.com/v1/submit-jobs',
    {
      method: 'POST',
      headers: {
        'Authorization': 'Bearer YOUR_API_KEY',
        'Content-Type': 'application/json',
      },
      body: JSON.stringify({
        prompt: prompts[style],
        webhookUrl: `https://your-app.com/webhook/avatar/${userId}`,
      }),
    }
  );

  const { taskId } = await response.json();
  // Save taskId → userId mapping, await webhook
  await db.pendingAvatars.insert({ userId, taskId });
}

// Webhook handler — receives result when image is ready
app.post('/webhook/avatar/:userId', async (req, res) => {
  const { userId } = req.params;
  const { status, imageUrl } = req.body;

  if (status === 'completed') {
    await db.users.update(userId, { avatarUrl: imageUrl });
    await notifyUser(userId, 'Your avatar is ready!');
  }

  res.sendStatus(200);
});