check if dataset is generated

1 files changed, 8 insertions, 21 deletions

diff --git a/app/client/modules/pix2pixhd/views/sequence.editor.js b/app/client/modules/pix2pixhd/views/sequence.editor.js
index 33ed629..135dfb3 100644
--- a/app/client/modules/pix2pixhd/views/sequence.editor.js
+++ b/app/client/modules/pix2pixhd/views/sequence.editor.js
@@ -18,26 +18,6 @@ const initialState = {
   title: null,
 }
 
-/*
-  so there are two things you could do with this
-  1) create an entirely new dataset
-  2) add frames to an existing dataset
-  ...
-  method 1)
-  - requires minimal setup, just a server script which..
-     - creates the new sequence folder
-     - symlinks the frames
-     - runs build_dataset
-     - create appropriate file object, using parent dataset's folder_id
-     - enqueue initial training
-       ---> we can then train on basic subsets, with more control, like we do already...
-     - tell parent dataset thing the new sequence name
-  ...
-  method 2)
-  - requires sequence editor to be aware of its own dataset
-  - requires sequence editor to know whether a sequence is original or not
-*/
-
 class SequenceEditor extends Component {
   state = { ...initialState }
 
@@ -80,9 +60,10 @@ class SequenceEditor extends Component {
   }
 
   render() {
-    const { app, remote, checkpoint, folder_id } = this.props
+    const { app, pix2pixhd, remote, checkpoint, folder_id, generated } = this.props
     const { cursor, selection, title } = this.state
     const path = "sequences/" + checkpoint.name
+    console.log(checkpoint, pix2pixhd)
     return (
       <div className='sequenceEditor row'>
         {selection
@@ -106,6 +87,11 @@ class SequenceEditor extends Component {
                 Create
               </Button>
             </Group>
+            <Group title='Salt dataset'>
+              {generated
+                ? 'Salting is only available on generated datasets.'
+                : 'Salting coming soon!'}
+            </Group>
           </div>
         : <div className='form'><Group title='New dataset'>Please select some frames</Group></div>
         }
@@ -145,6 +131,7 @@ function Frame ({ label, path, frame }) {
 
 const mapStateToProps = state => ({
   app: state.system.app,
+  pix2pixhd: state.module.pix2pixhd,
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({