pix2pix datasets in full effect

8 files changed, 30 insertions, 59 deletions

diff --git a/app/client/modules/pix2pix/pix2pix.actions.js b/app/client/modules/pix2pix/pix2pix.actions.js
index 5d53136..82311ad 100644
--- a/app/client/modules/pix2pix/pix2pix.actions.js
+++ b/app/client/modules/pix2pix/pix2pix.actions.js
@@ -9,11 +9,12 @@ import actions from '../../actions'
 
 import { allProgress } from '../../util'
 
+import pix2pixModule from './pix2pix.module'
+
 export const load_directories = (id) => (dispatch) => {
-  const module = 'samplernn'
+  const module = pix2pixModule.name
   allProgress([
     datasetLoader.load(module),
-    actions.task.index({ module }),
     // actions.socket.list_directory({ module, dir: 'datasets' }),
     // actions.socket.list_directory({ module, dir: 'results' }),
     // actions.socket.list_directory({ module, dir: 'output' }),
@@ -22,8 +23,7 @@ export const load_directories = (id) => (dispatch) => {
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
-    const [datasetApiReport, tasks] = res //, datasets, results, output, datasetUsage, lossReport] = res
-
+    const [datasetApiReport] = res //, datasets, results, output, datasetUsage, lossReport] = res
     const {
       folderLookup,
       fileLookup,
@@ -102,21 +102,21 @@ export const load_directories = (id) => (dispatch) => {
         fileLookup,
         datasetLookup,
         folders, files,
-        checkpoints,
-        output,
+        // checkpoints,
+        // output,
       },
     })
     if (id) {
       console.log('folder id', id)
       dispatch({
-        type: types.pix2pix.set_folder,
-        folder_id: id,
+        type: types.dataset.set_folder,
+        data: {
+          folder_id: id,
+          module
+        },
       })
     }
   }).catch(e => {
     console.error(e)
   })
 }
-
-export const set_folder = (folder) => { types.pix2pix.set_folder, folder }
-
diff --git a/app/client/modules/pix2pix/pix2pix.module.js b/app/client/modules/pix2pix/pix2pix.module.js
index 63b1e94..65e173a 100644
--- a/app/client/modules/pix2pix/pix2pix.module.js
+++ b/app/client/modules/pix2pix/pix2pix.module.js
@@ -1,6 +1,6 @@
 const pix2pixModule = {
   name: 'pix2pix',
-  datatype: 'audio',
+  datatype: 'video',
 }
 
 export default pix2pixModule
diff --git a/app/client/modules/pix2pix/pix2pix.reducer.js b/app/client/modules/pix2pix/pix2pix.reducer.js
index fbd38d1..10135a5 100644
--- a/app/client/modules/pix2pix/pix2pix.reducer.js
+++ b/app/client/modules/pix2pix/pix2pix.reducer.js
@@ -1,39 +1,20 @@
 import types from '../../types'
+import datasetReducer from '../../dataset/dataset.reducer'
 
 const pix2pixInitialState = {
   loading: true,
+  progress: { i: 0, n: 0 },
   error: null,
-  folders: [],
   folder_id: 0,
   data: null,
-  lossReport: null,
 }
 
 const pix2pixReducer = (state = pix2pixInitialState, action) => {
-  // console.log(action.type)
-  switch(action.type) {
-    case types.pix2pix.init:
-      return {
-        ...state,
-        loading: false,
-        data: action.data,
-      }
-
-    case types.socket.connect:
-      return {
-        ...state,
-      }
-
-    case types.task.task_begin:
-      return {
-        ...state,
-      }
-
-    case types.task.task_finish:
-      return {
-        ...state,
-      }
+  if (action.data && action.data.module === 'pix2pix') {
+    state = datasetReducer(state, action)
+  }
 
+  switch (action.type) {
     default:
       return state
   }
diff --git a/app/client/modules/pix2pix/views/pix2pix.show.js b/app/client/modules/pix2pix/views/pix2pix.show.js
index b4cdc50..ef4b906 100644
--- a/app/client/modules/pix2pix/views/pix2pix.show.js
+++ b/app/client/modules/pix2pix/views/pix2pix.show.js
@@ -37,6 +37,7 @@ class Pix2pixShow extends Component {
     const { pix2pix, match, history } = this.props
     const { folderLookup } = (pix2pix.data || {})
     const folder = (folderLookup || {})[pix2pix.folder_id] || {}
+
     return (
       <div className='app pix2pix'>
         <div class='heading'>
diff --git a/app/client/modules/pix2wav/pix2wav.module.js b/app/client/modules/pix2wav/pix2wav.module.js
index c0fe23a..bca268f 100644
--- a/app/client/modules/pix2wav/pix2wav.module.js
+++ b/app/client/modules/pix2wav/pix2wav.module.js
@@ -1,6 +1,6 @@
 const pix2wavModule = {
   name: 'pix2wav',
-  datatype: 'audio',
+  datatype: 'spectrogram',
 }
 
 export default pix2wavModule
diff --git a/app/client/modules/samplernn/samplernn.actions.js b/app/client/modules/samplernn/samplernn.actions.js
index e9248c0..d0fda31 100644
--- a/app/client/modules/samplernn/samplernn.actions.js
+++ b/app/client/modules/samplernn/samplernn.actions.js
@@ -9,11 +9,12 @@ import actions from '../../actions'
 
 import { allProgress } from '../../util'
 
+import samplernnModule from './samplernn.module'
+
 export const load_directories = (id) => (dispatch) => {
-  const module = 'samplernn'
+  const module = samplernnModule.name
   allProgress([
     datasetLoader.load(module),
-    actions.task.index({ module }),
     actions.socket.list_directory({ module, dir: 'datasets' }),
     actions.socket.list_directory({ module, dir: 'results' }),
     actions.socket.list_directory({ module, dir: 'output' }),
@@ -23,7 +24,7 @@ export const load_directories = (id) => (dispatch) => {
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
-    const [datasetApiReport, tasks, datasets, results, output, datasetUsage, lossReport] = res
+    const [datasetApiReport, datasets, results, output, datasetUsage, lossReport] = res
 
     const {
       folderLookup,
@@ -108,10 +109,12 @@ export const load_directories = (id) => (dispatch) => {
       },
     })
     if (id) {
-      console.log('folder id', id)
       dispatch({
-        type: types.samplernn.set_folder,
-        folder_id: id,
+        type: types.dataset.set_folder,
+        data: {
+          folder_id: id,
+          module
+        },
       })
     }
   }).catch(e => {
@@ -205,5 +208,3 @@ export const import_files = (state, datasetLookup, fileLookup) => (dispatch) =>
     console.error(e)
   })
 }
-
-export const set_folder = (folder) => { types.samplernn.set_folder, folder }
diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index 1efbc90..ce3a549 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -18,25 +18,12 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
   }
 
   switch(action.type) {
-
-    case types.samplernn.set_folder:
-      return {
-        ...state,
-        folder_id: action.folder_id,
-      }
-
     case types.samplernn.load_loss:
       return {
         ...state,
         lossReport: action.lossReport,
       }
 
-    case types.app.load_progress:
-      return {
-        ...state,
-        progress: action.progress
-      }
-
     default:
       return state
   }
diff --git a/app/client/modules/samplernn/views/samplernn.show.js b/app/client/modules/samplernn/views/samplernn.show.js
index 98314f9..f44deda 100644
--- a/app/client/modules/samplernn/views/samplernn.show.js
+++ b/app/client/modules/samplernn/views/samplernn.show.js
@@ -30,6 +30,7 @@ class SampleRNNShow extends Component {
     if (id) {
       if (parseInt(id)) localStorage.setItem('samplernn.last_id', id)
       if (! samplernn.folder || samplernn.folder.id !== id) {
+        console.log('looooooooooad', id)
         actions.load_directories(id)
       }
     }