1 files changed, 18 insertions, 3 deletions
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index b4a206a..c09eadc 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -137,6 +137,21 @@ const augment = {
     }
     return args
   },
+  listen: (task, res, i) => {
+    // extract the new path name
+    const lines = res.split('\n')
+    for (let line of lines) {
+      console.log(line)
+      if ( line.match(/^render_dir: /) ) {
+        let tag = line.split(': ')[1].trim()
+        task.opt.render_dir = tag
+        // task.opt.filename = filename
+        console.log(">>>>>> created dataset", tag)
+        return { type: 'progress', action: 'resolve_dataset', task }
+      }
+    }
+    return null
+  },
   after: 'render_recursive',
 }
 const clear_recursive = {
@@ -210,11 +225,11 @@ const render_recursive = {
       console.log('will not render mov')
       return 'CANCEL'
     }
-    const tag = task.dataset + '/' + pad(task.opt.epoch || 'latest', 2) + '_' + pad(task.opt.augment_take, 5) + '_' + pad(task.opt.augment_make, 5)
-    console.log('rendering recursve path:', tag)
+    const render_dir = task.opt.render_dir.replace('./recursive/', '')
+    console.log('rendering recursve path:', render_dir)
     return [
       '--path', 'recursive',
-      '--tag', tag,
+      '--tag', render_dir,
       '--module', task.module,
       '--endpoint', process.env.API_REMOTE + '/api/folder/' + task.opt.folder_id + '/upload/',
     ]