// Install: npm install axios
import axios from 'axios';

const apiKey = 'YOUR_API_KEY';
const baseURL = 'https://api.bestimage.ai';

// Create image generation task
async function generateImage() {
  try {
    const response = await axios.post('/api/v1/image/task', {
      model_name: 'flux-context-max',
      prompt: 'A beautiful sunset over mountains',
      width: "16",
      height: "9"
    }, {
      baseURL,
      headers: {
        'Authorization': `Bearer ${apiKey}`,
        'Content-Type': 'application/json'
      }
    });

    const { task_id } = response.data.data;
    console.log('Task created:', task_id);
    
    // Poll for results
    return await pollResult(task_id);
  } catch (error) {
    console.error('Error:', error.response?.data || error.message);
  }
}

// Poll for task completion
async function pollResult(taskId) {
  while (true) {
    try {
      const response = await axios.get('/api/v1/image/' + taskId, {
        baseURL,
        headers: { 'Authorization': `Bearer ${apiKey}` }
      });

      const { task_status, task_result } = response.data.data;
      
      if (task_status === 'succeed') {
        console.log('Images generated:', task_result.images);
        return task_result.images;
      } else if (task_status === 'failed') {
        throw new Error('Task failed');
      }
      
      // Wait 2 seconds before next poll
      await new Promise(resolve => setTimeout(resolve, 2000));
    } catch (error) {
      console.error('Polling error:', error);
      break;
    }
  }
}

generateImage();