1 files changed, 30 insertions, 15 deletions
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index f5f4a1d..dafed79 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -52,7 +52,35 @@ class SampleRNNDatasets extends Component {
   render(){
     const { samplernn } = this.props
     const folder = samplernn.folder
-    if (!folder.name) return
+    if (!folder || !folder.name) return
+    return (
+      <div className='app'>
+        <div class='heading'>
+          <h1>{folder.name}</h1>
+        </div>
+        {folder.name !== 'unsorted' &&
+          <DatasetForm title='Add Files' folder={folder} canUpload canAddURL />
+        }
+        {this.renderDataset()}
+      </div>
+    )
+  }
+  renderDataset(){
+    return (
+      <div class='rows params datasets'>
+        <div class='row dataset'>
+          <div class='col'>input</div>
+          <div class='col'></div>
+          <div class='col'>checkpoint</div>
+          <div class='col'>output</div>
+        </div>
+        {this.renderGroups()}
+      </div>
+    )
+  }
+  renderGroups(){
+    const { samplernn } = this.props
+    const folder = samplernn.folder
     const { mapFn, sortFn } = util.sort.orderByFn('date desc')
     const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => {
       const dataset = pair[1]
@@ -92,20 +120,7 @@ class SampleRNNDatasets extends Component {
         </div>
       )
     })
-    return (
-      <div className='app'>
-        <DatasetForm folder={folder} canRename canUpload canAddURL />
-        <div class='rows params datasets'>
-          <div class='row dataset'>
-            <div class='col'>input</div>
-            <div class='col'></div>
-            <div class='col'>checkpoint</div>
-            <div class='col'>output</div>
-          </div>
-          {datasets}
-        </div>
-      </div>
-    )
+    return datasets
   }
 }