1 files changed, 31 insertions, 2 deletions
diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index 372b3bc..524d650 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -17,34 +17,63 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
         loading: false,
         data: action.data,
       }
+
     case types.socket.connect:
       return {
         ...state,
       }
+
     case types.task.task_begin:
       return {
         ...state,
       }
+
     case types.task.task_finish:
       return {
         ...state,
       }
+
+    case types.folder.create:
+      if (action.data.module === 'samplernn') {
+        return {
+          ...state,
+          loading: false,
+          // folder: {
+          //   ...action.data,
+          //   input: [],
+          //   checkpoints: [],
+          //   output: [],
+          // }
+        }
+      }
+      return state
+
     case types.file.create:
       if (state.folder.id === action.data.folder_id) {
         return {
           ...state,
-          files: [action.data].concat(state.files),
+          loading: false,
+          folder: {
+            ...state.folder,
+            input: [action.data].concat(state.folder.input),
+          },
         }
       }
       return state
+
     case types.folder.upload_complete:
       if (state.folder.id === action.folder) {
         return {
           ...state,
-          files: [action.files].concat(state.files),
+          loading: false,
+          folder: {
+            ...state.folder,
+            input: [action.data].concat(state.folder.input),
+          },
         }
       }
       return state
+
     case types.socket.status:
       return samplernnSocket(state, action.data)