5 files changed, 15 insertions, 8 deletions
diff --git a/app/client/modules/pix2pix/live.component.js b/app/client/modules/pix2pix/live.component.js
index 9aef297..82f315c 100644
--- a/app/client/modules/pix2pix/live.component.js
+++ b/app/client/modules/pix2pix/live.component.js
@@ -61,7 +61,7 @@ class LivePix2Pix extends Component {
   }
   render(){
     return (
-      <div className='app'>
+      <div className='app centered'>
         <Player width={424} height={256} />
         <div className='params row'>
           <div className='column'>
diff --git a/app/client/modules/samplernn/samplernn.actions.js b/app/client/modules/samplernn/samplernn.actions.js
index ef3e417..357617a 100644
--- a/app/client/modules/samplernn/samplernn.actions.js
+++ b/app/client/modules/samplernn/samplernn.actions.js
@@ -17,12 +17,15 @@ export const load_directories = (id) => (dispatch) => {
     actions.socket.list_directory({ module: 'samplernn', dir: 'datasets' }),
     actions.socket.list_directory({ module: 'samplernn', dir: 'results' }),
     actions.socket.list_directory({ module: 'samplernn', dir: 'output' }),
+    actions.socket.disk_usage({ module: 'samplernn', dir: 'datasets' }),
     load_loss()(dispatch),
   ], (percent, i, n) => {
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
     // console.log(res)
-    const [folders, files, tasks, datasets, results, output, lossReport] = res
+    const [folders, files, tasks, datasets, results, output, datasetUsage, lossReport] = res
+
+    console.log(datasetUsage)
 
     const unsortedFolder = {
       id: 0,
@@ -265,6 +268,7 @@ export const import_files = (state, datasetLookup, fileLookup) => (dispatch) =>
   console.log(promises)
   return Promise.all(promises).then(data => {
     console.log(data)
+    window.location.href = '/samplernn/datasets/' + folder.id
   }).catch(e => {
     console.error(e)
   })
diff --git a/app/client/modules/samplernn/samplernn.new.js b/app/client/modules/samplernn/samplernn.new.js
index 8d8b68c..a6e4fd3 100644
--- a/app/client/modules/samplernn/samplernn.new.js
+++ b/app/client/modules/samplernn/samplernn.new.js
@@ -20,7 +20,11 @@ class SampleRNNNew extends Component {
   }
   render(){
     const { history } = this.props
-    return <NewDatasetForm module={samplernnModule} history={history} />        
+    return (
+      <div class='app samplernn'>
+        <NewDatasetForm module={samplernnModule} history={history} />
+      </div>
+    )
   }
 }
 const mapStateToProps = state => ({
diff --git a/app/client/modules/samplernn/samplernn.results.js b/app/client/modules/samplernn/samplernn.results.js
index d12e856..31079ce 100644
--- a/app/client/modules/samplernn/samplernn.results.js
+++ b/app/client/modules/samplernn/samplernn.results.js
@@ -54,13 +54,12 @@ class SampleRNNResults extends Component {
     })
 
     return (
-      <div className='app top left'>
-        <div className='heading'>
+      <div className='app samplernn'>
+        <div className='heading row middle'>
           <h1>SampleRNN Results</h1>
+          <Link to='/samplernn/new/'>new project</Link>
         </div>
         <div class='rows params renders'>
-          <Link to='/samplernn/new/'>new dataset</Link>
-          <br/><br/>
           {renders}
         </div>
       </div>
diff --git a/app/client/modules/samplernn/samplernn.show.js b/app/client/modules/samplernn/samplernn.show.js
index 1237d9b..982ec53 100644
--- a/app/client/modules/samplernn/samplernn.show.js
+++ b/app/client/modules/samplernn/samplernn.show.js
@@ -30,7 +30,7 @@ class SampleRNNShow extends Component {
     const { folderLookup } = (samplernn.data || {})
     const folder = (folderLookup || {})[samplernn.folder_id] || {}
     return (
-      <div className='app'>
+      <div className='app samplernn'>
         <div class='heading'>
           <div class='spaced'>
             <h1>{folder ? folder.name : <Loading />}</h1>