import Heurist from 'heurist'

const heurist = new Heurist({
  apiKey: process.env['HEURIST_API_KEY']
});

async function generateImage() {
  const fluxLoraTask = new FluxLoraTask({
    workflow_id: '3',
    prompt: 'A photo of GECKO with dark green skin spots playing american football, cartoon, comic, anime',
    lora_name: 'gecko',
    timeout_seconds: 300 // optional
  });

  const response = await heurist.workflow.executeWorkflowAndWaitForResult(
    fluxLoraTask,
    600000,  // 10 minutes timeout
    3000     // 3 seconds polling interval
  );

  console.log('Generated image URL:', response.result);
}

{
  "task_id": "9034a6ca80",
  "task_type": "txt2img",
  "workflow_id": "3",
  "status": "finished",
  "timeout_seconds": 300,
  "create_timestamp": 1738916890,
  "result": "https://sequencer-v2.heurist.xyz/files/flux-lora/flux-lora-0x737c3fcc8def561dbE0D3de8Cc3B0646574b7651-9034a6ca80.jpg",
  "miner_id": "0x737c3fcc8def561dbE0D3de8Cc3B0646574b7651",
  "queue_position": 1,
  "inference_latency": 18.42825,
  "msg": ""
}

const heurist = new Heurist({
  apiKey: process.env['HEURIST_API_KEY']  // format: consumerId#apiKey
});

const fluxLoraTask = new FluxLoraTask({
  consumer_id: 'consumerId',
  api_key: 'apiKey',
  ...
});

class FluxLoraTask extends WorkflowTask {
  constructor(options: FluxLoraTaskOptions)
  get task_type(): WorkflowTaskType.FluxLora
  get task_details(): {
    parameters: {
      prompt: string
      lora_name: string
      width?: number
      height?: number
      steps?: number
      noise_seed?: number
    }
  }
}

enum WorkflowTaskType {
  FluxLora = 'flux-lora'
}