From 51d9212d97e3b326a8e9e6499bc83d27df2ef64c Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 5 Jun 2018 03:28:01 +0200
Subject: dont train images lol

---
 app/client/modules/samplernn/samplernn.show.js | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

(limited to 'app/client/modules/samplernn/samplernn.show.js')

diff --git a/app/client/modules/samplernn/samplernn.show.js b/app/client/modules/samplernn/samplernn.show.js
index f6ce179..8b6f357 100644
--- a/app/client/modules/samplernn/samplernn.show.js
+++ b/app/client/modules/samplernn/samplernn.show.js
@@ -57,23 +57,27 @@ class SampleRNNShow extends Component {
     )
   }
   datasetActions(dataset){
-    const actions = this.props.actions.samplernn
+    const { samplernn, remote } = this.props
+    const { fileLookup } = samplernn.data
+    const input = fileLookup[dataset.input[0]]
+    if (! input) return null
+    if (input.name.match(/(gif|jpe?g|png)$/i)) return null
     return (
       <div>
         <div class={'actions'}>
-          <span class='link' onClick={() => samplernnTasks.train_task(dataset, 1)}>train</span>
-          <span class='link' onClick={() => samplernnTasks.train_task(dataset, 2)}>2x</span>
-          <span class='link' onClick={() => samplernnTasks.train_task(dataset, 4)}>4x</span>
-          <span class='link' onClick={() => samplernnTasks.train_task(dataset, 6)}>6x</span>
-          <span class='link' onClick={() => samplernnTasks.train_task(dataset, 18)}>18x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, 1)}>train</span>
+          <span class='link' onClick={() => remote.train_task(dataset, 2)}>2x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, 4)}>4x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, 6)}>6x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, 18)}>18x</span>
         </div>
         {dataset.isBuilt
         ? <div class='subtext'>
             {'fetched '}
-            <span class='link' onClick={() => samplernnTasks.clear_cache_task(dataset)}>rm</span>
+            <span class='link' onClick={() => remote.clear_cache_task(dataset)}>rm</span>
           </div>
         : <div class='subtext'>
-            <span class='link' onClick={() => samplernnTasks.fetch_task(dataset)}>fetch</span>
+            <span class='link' onClick={() => remote.fetch_task(input.url)}>fetch</span>
           </div>
         }
       </div>
@@ -91,6 +95,7 @@ const mapStateToProps = state => ({
 
 const mapDispatchToProps = (dispatch, ownProps) => ({
   actions: bindActionCreators(samplernnActions, dispatch),
+  remote: bindActionCreators(samplernnTasks, dispatch),
   audioPlayer: bindActionCreators(audioPlayerActions, dispatch),
 })
 
-- 
cgit v1.2.3-70-g09d2