From 55df7e617feb374c546d72a958b601402dbd7c5d Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Thu, 22 Aug 2019 16:02:40 +0200
Subject: samplernn generate tasks

---
 app/client/modules/samplernn/samplernn.tasks.js      | 15 +++++++++++++++
 app/client/modules/samplernn/views/samplernn.show.js |  9 +++++++++
 2 files changed, 24 insertions(+)

(limited to 'app/client/modules')

diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
index bb22791..55828b8 100644
--- a/app/client/modules/samplernn/samplernn.tasks.js
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -22,6 +22,21 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
   console.log(task)
   return actions.queue.add_task(task)
 }
+export const generate_task = (dataset, folder_id, sample_length=5, n_samples=6, epoch=0) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'generate',
+    dataset: dataset.name,
+    epoch: epoch || (dataset.checkpoints.length && dataset.checkpoints[0].epoch) || 0,
+    folder_id: folder_id,
+    opt: {
+      sample_length: 44100 * sample_length,
+      n_samples,
+    }
+  }
+  console.log(task)
+  return actions.queue.add_task(task)
+}
 export const fetch_task = (url, folder_id, file_id, dataset) => dispatch => {
   if (! url) return console.log('input file inaccessible (no url)')
   const task = {
diff --git a/app/client/modules/samplernn/views/samplernn.show.js b/app/client/modules/samplernn/views/samplernn.show.js
index 7585e6c..6226fd6 100644
--- a/app/client/modules/samplernn/views/samplernn.show.js
+++ b/app/client/modules/samplernn/views/samplernn.show.js
@@ -93,6 +93,15 @@ class SampleRNNShow extends Component {
           <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 6)}>6x</span>
           <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 18)}>18x</span>
         </div>
+        <div class={'actions'}>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 5, 12)}>generate</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 0.5, 120)}>rhythm</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 2, 30)}>short</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 10, 6)}>medium</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 15, 4)}>long</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 30, 2)}>longer</span>
+          <span class='link' onClick={() => remote.generate_task(dataset, samplernn.folder_id, 60, 1)}>longest</span>
+        </div>
         {dataset.isBuilt
         ? <div class='subtext'>
             {'fetched '}
-- 
cgit v1.2.3-70-g09d2