1 files changed, 29 insertions, 2 deletions

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index 1ee4886..370597b 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -31,15 +31,16 @@ export const load_directories = (id) => (dispatch) => {
       folders,
       files,
       unsortedFolder,
+      resultsFolder,
     } = datasetApiReport
     // console.log(datasetUsage)
 
     const sequenceDirectories = sequences.filter(s => s.dir)
-    console.log(sequenceDirectories)
+    // console.log(sequenceDirectories)
     sequenceDirectories.forEach(dir => {
       const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
       dataset.isBuilt = true
-      console.log(dir.name, dataset)
+      // console.log(dir.name, dataset)
     })
 
     datasets.filter(s => s.dir).forEach(dir => {
@@ -119,6 +120,7 @@ export const load_directories = (id) => (dispatch) => {
         sequences: sequenceDirectories,
         datasets,
         checkpoints: checkpointDirectories,
+        resultsFolder,
       },
     })
     if (id) {
@@ -135,3 +137,28 @@ export const load_directories = (id) => (dispatch) => {
     console.error(e)
   })
 }
+
+export const load_results = (id) => (dispatch) => {
+  const module = pix2pixhdModule.name
+  util.allProgress([
+    actions.folder.index({ name: 'results' }),
+    actions.file.index({ module, generated: 1 }),
+    actions.socket.list_sequences({ module, dir: 'results' }),
+    actions.socket.list_directory({ module, dir: 'renders' }),
+  ], (percent, i, n) => {
+    console.log('pix2pixhd load progress', i, n)
+    dispatch({ type: types.app.load_progress, progress: { i, n }})
+  }).then(res => {
+    const [folders, files, results, renders] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    console.log(files, results, renders)
+    dispatch({
+      type: types.pix2pixhd.load_results,
+      results: {
+        resultsFolder: folders[0],
+        files,
+        results,
+        renders,
+      }
+    })
+  })
+}
\ No newline at end of file