import { createClient, saveBlob } from 'chutes-js';

const client = createClient({ apiKey: process.env.CHUTES_API_KEY });

// 1. Chat Completion (OpenAI-compatible)
const response = await client.chat({
  model: 'deepseek-ai/DeepSeek-V3-0324',
  messages: [{ role: 'user', content: 'Hello!' }]
});
console.log(response.choices[0].message.content);

// 2. Image Generation (unified params)
const image = await client.image({
  model: 'flux',                // alias for FLUX.1-schnell
  prompt: 'A beautiful sunset over mountains',
  aspectRatio: '16:9',          // auto-converted to width/height
  steps: 25,
  cfgScale: 7
});
await saveBlob(image, './sunset.png');

// 3. Video Generation (I2V with URL auto-fetch)
const video = await client.video({
  model: 'wan',                              // alias for wan-2-2-i2v-14b-fast
  prompt: 'A cat playing piano smoothly',
  image: 'https://example.com/cat.jpg',      // URL auto-fetched to base64
  duration: 5,                               // seconds → frames via fps
  aspectRatio: '16:9'
});
await saveBlob(video, './cat-piano.mp4');

// 4. Audio Generation (Text-to-Speech)
const audio = await client.audio({
  model: 'tts',                 // alias for kokoro
  text: 'Hello, welcome to Chutes!',
  voice: 'af_heart',
  speed: 1.0
});
await saveBlob(audio, './welcome.wav');

const response = await client.chat({
  model: 'deepseek-ai/DeepSeek-V3-0324',
  messages: [{ role: 'user', content: 'Explain quantum computing' }]
});

for await (const chunk of client.chatStream({ model, messages })) {
  process.stdout.write(chunk.choices[0]?.delta?.content || '');
}

import { saveBlob } from 'chutes-js';

// Basic usage
const blob = await client.image({
  model: 'flux',
  prompt: 'A cyberpunk city at night'
});
await saveBlob(blob, './cyberpunk.png');

// With all options
const blob2 = await client.image({
  model: 'qwen-image',
  prompt: 'A serene Japanese garden',
  aspectRatio: '16:9',        // or use width/height directly
  steps: 50,
  cfgScale: 7.5,
  negativePrompt: 'blurry, low quality',
  seed: 42
});

// Image editing (requires reference images)
const edited = await client.image({
  model: 'qwen-edit',
  prompt: 'Add a rainbow to the sky',
  images: ['https://example.com/photo.jpg'],  // URLs auto-fetched
  aspectRatio: '1:1'
});

// Subdomain model (HiDream)
const image = await client.invoke(
  { name: 'hidream', username: 'chutes' },
  '/generate',
  { prompt: 'Cyberpunk city', resolution: '1024x1024', guidance_scale: 5 }
);

// Centralized model (requires model in payload)
const image2 = await client.invoke(
  'image.chutes.ai',
  '/generate',
  { model: 'qwen-image', prompt: 'A sunset', width: 1024, height: 1024 }
);

import { readFileSync } from 'fs';
import { saveBlob } from 'chutes-js';

// Using image URL (auto-fetched to base64)
const video = await client.video({
  model: 'wan',                              // alias for wan-2-2-i2v-14b-fast
  prompt: 'A cat playing piano elegantly',
  image: 'https://example.com/cat.jpg',      // URL auto-fetched
  duration: 5,                               // 5 seconds
  aspectRatio: '16:9'
});
await saveBlob(video, './output.mp4');

// Using local file (base64)
const imageBase64 = readFileSync('./input.png').toString('base64');
const video2 = await client.video({
  model: 'i2v',
  prompt: 'Ocean waves crashing on rocks',
  image: imageBase64,
  frames: 81,                   // or use duration in seconds
  fps: 16,
  cfgScale: 1,
  cfgScale2: 1,
  negativePrompt: 'blurry, static',
  seed: 12345
});

// Check for async job (some models return job ID)
if (video2 instanceof Blob) {
  await saveBlob(video2, './ocean.mp4');
} else {
  console.log('Job started:', video2.job_id);
  // Poll with client.getJobStatus(video2.job_id)
}

const video = await client.invoke(
  { name: 'wan-2-2-i2v-14b-fast', username: 'chutes' },
  '/generate',
  {
    prompt: 'A dog running',
    image: imageBase64,
    resolution: '480p',
    fps: 16,
    frames: 81,
    guidance_scale: 1,
    guidance_scale_2: 1
  }
);

import { saveBlob } from 'chutes-js';

// Kokoro TTS (50+ voices)
const audio = await client.audio({
  model: 'tts',                 // alias for kokoro
  text: 'Hello, welcome to Chutes AI!',
  voice: 'af_heart',            // American female
  speed: 1.0
});
await saveBlob(audio, './welcome.wav');

// CSM-1B (conversational)
const audio2 = await client.audio({
  model: 'csm',
  text: 'This is a test of the CSM model.',
  speaker: 0,
  maxDuration: 15              // 15 seconds (auto-converted to ms)
});
await saveBlob(audio2, './csm-test.wav');

// Kokoro via invoke
const audio = await client.invoke(
  { name: 'kokoro', username: 'chutes' },
  '/speak',
  { text: 'Hello world', voice: 'af_heart', speed: 1.0 }
);

// CSM via invoke
const audio2 = await client.invoke(
  { name: 'csm-1b', username: 'chutes' },
  '/speak',
  { text: 'Hello', speaker: 0, max_duration_ms: 10000 }
);

// Pattern: https://{name}-{username}.chutes.ai/{path}

// Subdomain model
const result = await client.invoke(
  { name: 'custom-model', username: 'myuser' },
  '/generate',
  { prompt: 'Hello', ...otherParams }
);

// Direct URL
const result2 = await client.invoke(
  'https://custom-endpoint.chutes.ai',
  '/api/v1/generate',
  { prompt: 'Hello' }
);

const status = await client.getJobStatus('job_abc123');
// { state: 'completed', result: {...} }

import { createClient, ChutesError, ValidationError } from 'chutes-js';

try {
  await client.image({
    model: 'flux',
    prompt: 'A sunset',
    steps: 500  // exceeds max
  });
} catch (e) {
  if (e instanceof ValidationError) {
    // Parameter validation failed
    console.log('Validation error:', e.message);
    // e.g., "steps exceeds maximum value of 100"
  } else if (e instanceof ChutesError) {
    // API error
    console.log('Status:', e.status);  // 429 = rate limited
    console.log('Body:', e.body);      // Raw error response
  }
}

import {
  // Client
  createClient,
  
  // Errors
  ChutesError,
  ValidationError,
  
  // Utilities
  parseSSE,
  saveBlob,
  
  // Model configs (advanced)
  IMAGE_MODELS,
  IMAGE_MODEL_ALIASES,
  VIDEO_MODELS,
  VIDEO_MODEL_ALIASES,
  AUDIO_MODELS,
  AUDIO_MODEL_ALIASES,
  ALL_MODELS,
  ALL_ALIASES
} from 'chutes-js';