add folder_id to dataset object

2 files changed, 18 insertions, 1 deletions

diff --git a/app/client/dataset/dataset.loader.js b/app/client/dataset/dataset.loader.js
index 76d8b16..f1ff75e 100644
--- a/app/client/dataset/dataset.loader.js
+++ b/app/client/dataset/dataset.loader.js
@@ -20,6 +20,7 @@ export const unsortedFolder = (module, instantiate=false) => {
 export const emptyDataset = (module, datasetLookup, name, folder) => {
   const dataset = {
     name,
+    folder_id: folder.id,
     input: [],
     checkpoints: [],
     output: [],
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index 78def7a..d4e390c 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -152,8 +152,24 @@ class Pix2PixHDTrain extends Component {
 
   renderEditor(){
     const { pix2pixhd, queue, remote } = this.props
-    const { checkpoint, folder_id } = pix2pixhd
     const { dataset, processed } = this.state
+    let { checkpoint, folder_id } = pix2pixhd
+
+    const currentDataset = this.props.pix2pixhd.data.datasetLookup[this.state.dataset]
+    if (currentDataset && currentDataset.folder_id) {
+      folder_id = currentDataset.folder_id || folder_id
+      console.log("dataset folder id", folder_id)
+      const processed = currentDataset.input
+        .map(f => this.props.pix2pixhd.data.fileLookup[f])
+        .reduce((a, b) => {
+          console.log(b)
+          return b.processed || a
+        }, false)
+      console.log("is processed?", processed)
+    }
+    console.log(currentDataset)
+    console.log("so folder id is..", folder_id)
+
     return (
       <div>
         <Group title='Sequence Editor'>