import path from 'path'
import fs from 'fs'

const name = 'biggan'
const cwd = process.env.BIGGAN_CWD || path.join(process.env.HOME, 'code/' + name + '/cli/')
const env = {
  LD_LIBRARY_PATH: [
    '/usr/local/cuda/lib64',
    process.env.FFMPEG_LIB_PATH,
    process.env.TENSORRT_LIB_PATH,
  ].filter(s => !!s).join(':'),
  TFHUB_CACHE_DIR: process.env.HOME + '/.tfhub',
  LC_ALL: 'C.UTF-8',
  LANG: 'C.UTF-8',
}

const invert = {
  type: 'pytorch_biggan',
  script: 'cli.py',
  params: (task) => {
    console.log(task)
    return [
      'biggan', 'extract_dense_vectors',
      '--folder_id', task.opt.folder_id,
      '--stochastic_clipping', '--label_clipping', '--clip_interval', 200
    ]
  }
}

const live = {
  type: 'pytorch_biggan',
  script: 'cli.py',
  params: (task) => {
    console.log(task)
    const opt = task.opt || {}
    return ['biggan', 'live']
  },
  // after: 'render',
}

const render = {
  type: 'pytorch_biggan',
  script: 'cli.py',
  params: (task) => {
    console.log(task)
    return [
      'cortex', 'upload_video',
    ]
  }
}

export default {
  name, cwd, env,
  activities: {
    invert, live, render,
  }
}