ok need to observe whats happening when the checkpoints come back

3 files changed, 35 insertions, 12 deletions

diff --git a/app/client/modules/pix2pix/pix2pix.actions.js b/app/client/modules/pix2pix/pix2pix.actions.js
index 8633c0a..778bc64 100644
--- a/app/client/modules/pix2pix/pix2pix.actions.js
+++ b/app/client/modules/pix2pix/pix2pix.actions.js
@@ -15,15 +15,15 @@ export const load_directories = (id) => (dispatch) => {
   const module = pix2pixModule.name
   util.allProgress([
     datasetLoader.load(module),
-    // actions.socket.list_directory({ module, dir: 'datasets' }),
-    // actions.socket.list_directory({ module, dir: 'results' }),
-    // actions.socket.list_directory({ module, dir: 'output' }),
+    actions.socket.list_directory({ module, dir: 'sequences/pix2pix/' }),
+    actions.socket.list_directory({ module, dir: 'datasets/pix2pix/' }),
+    actions.socket.list_directory({ module, dir: 'checkpoints/pix2pix/' }),
     // actions.socket.disk_usage({ module, dir: 'datasets' }),
   ], (percent, i, n) => {
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
-    const [datasetApiReport] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const [datasetApiReport, sequences, datasets, checkpoints] = res //, datasets, results, output, datasetUsage, lossReport] = res
     const {
       folderLookup,
       fileLookup,
@@ -34,13 +34,21 @@ export const load_directories = (id) => (dispatch) => {
     } = datasetApiReport
     // console.log(datasetUsage)
 
-    // // also show the various flat audio files we have, in the input area..
-    // const flatDatasets = datasets.filter(s => s.name.match(/(wav|aiff?|flac|mp3)$/) && !s.dir)
-    // const builtDatasets = datasets.filter(s => s.dir)
-    // builtDatasets.forEach(dir => {
-    //   const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
-    //   dataset.isBuilt = true
-    // })
+    // const flatDatasets = datasets.filter(s => !s.dir)
+    sequences.filter(s => s.dir).forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.isBuilt = true
+    })
+
+    datasets.filter(s => s.dir).forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasDataset = true
+    })
+
+    checkpoints.filter(s => s.dir).forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasCheckpoints = true
+    })
 
     // flatDatasets.forEach(file => {
     //   file.uuid = uuidv1()
@@ -102,6 +110,9 @@ export const load_directories = (id) => (dispatch) => {
         fileLookup,
         datasetLookup,
         folders, files,
+        sequences,
+        datasets,
+        checkpoints,
         // checkpoints,
         // output,
       },
diff --git a/app/client/modules/pix2pix/pix2pix.tasks.js b/app/client/modules/pix2pix/pix2pix.tasks.js
index eaecca8..9572418 100644
--- a/app/client/modules/pix2pix/pix2pix.tasks.js
+++ b/app/client/modules/pix2pix/pix2pix.tasks.js
@@ -5,6 +5,17 @@ import types from '../../types'
 
 import actions from '../../actions'
 
+export const live_task = (dataset, checkpoint, epochs=1) => dispatch => {
+  const task = {
+    module: 'pix2pix',
+    activity: 'live',
+    dataset: dataset.name,
+    checkpoint: checkpoint,
+  }
+  console.log(task)
+  return actions.queue.add_task(task)
+}
+
 // export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
 //   const task = {
 //     module: 'samplernn',
diff --git a/app/client/modules/pix2pix/views/pix2pix.live.js b/app/client/modules/pix2pix/views/pix2pix.live.js
index 76b6727..7651dcc 100644
--- a/app/client/modules/pix2pix/views/pix2pix.live.js
+++ b/app/client/modules/pix2pix/views/pix2pix.live.js
@@ -11,6 +11,7 @@ import Button from '../../../common/button.component'
 import { startRecording, stopRecording, saveFrame } from '../../../live/player'
 
 import * as liveActions from '../../../live/live.actions'
+import * as pix2pixTasks from '../pix2pix.tasks'
 
 class Pix2PixLive extends Component {
   constructor(props){
@@ -45,7 +46,7 @@ class Pix2PixLive extends Component {
     this.props.actions.seek(frame)
   }
   start(){
-    //
+    // pix2pixTasks.live_task()
   }
   kill(){
     //