pix2pix datasets in full effect

4 files changed, 19 insertions, 37 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'>