cleanup

app.py

@@ -51,7 +51,7 @@ def initiate_run():
         transcripts = pd.DataFrame({
             tiger.ID_COL: ['ManualEntry'],
             tiger.SEQ_COL: [st.session_state.manual_entry]
-        })
+        }).set_index(tiger.ID_COL)
 
     # fasta file upload
     elif st.session_state.entry_method == ENTRY_METHODS['fasta']:
@@ -78,7 +78,6 @@ def initiate_run():
 
     # run model if we have any transcripts
     elif len(transcripts) > 0:
-        st.session_state.running = True
         st.session_state.transcripts = transcripts
 
 
@@ -90,8 +89,6 @@ if __name__ == '__main__':
         st.session_state.disable_off_target_checkbox = True
     if 'entry_method' not in st.session_state:
         st.session_state.entry_method = ENTRY_METHODS['manual']
-    if 'running' not in st.session_state:
-        st.session_state.running = False
     if 'transcripts' not in st.session_state:
         st.session_state.transcripts = None
     if 'input_error' not in st.session_state:
@@ -114,13 +111,13 @@ if __name__ == '__main__':
                 options=tuple(tiger.RUN_MODES.values()),
                 key='mode',
                 on_change=mode_change_callback,
-                disabled=st.session_state.running,
+                disabled=st.session_state.transcripts is not None,
             )
         with col2:
             st.checkbox(
                 label='Find off-target effects (slow)',
                 key='check_off_targets',
-                disabled=st.session_state.disable_off_target_checkbox or st.session_state.running
+                disabled=st.session_state.disable_off_target_checkbox or st.session_state.transcripts is not None
             )
 
     # transcript entry
@@ -129,25 +126,25 @@ if __name__ == '__main__':
             label='How would you like to provide transcript(s) of interest?',
             options=ENTRY_METHODS.values(),
             key='entry_method',
-            disabled=st.session_state.running
+            disabled=st.session_state.transcripts is not None
         )
         if st.session_state.entry_method == ENTRY_METHODS['manual']:
             st.text_input(
                 label='Enter a target transcript:',
                 key='manual_entry',
                 placeholder='Upper or lower case',
-                disabled=st.session_state.running
+                disabled=st.session_state.transcripts is not None
             )
         elif st.session_state.entry_method == ENTRY_METHODS['fasta']:
             st.file_uploader(
                 label='Upload a fasta file:',
                 key='fasta_entry',
-                disabled=st.session_state.running
+                disabled=st.session_state.transcripts is not None
             )
 
     # runtime
     with RUNTIME:
-        st.button(label='Get predictions!', on_click=initiate_run, disabled=st.session_state.running)
+        st.button(label='Get predictions!', on_click=initiate_run, disabled=st.session_state.transcripts is not None)
         progress = st.empty()
 
     # results
@@ -188,6 +185,5 @@ if __name__ == '__main__':
             check_off_targets=st.session_state.check_off_targets,
             status_update_fn=progress_update
         )
-        st.session_state.running = False
         st.session_state.transcripts = None
         st.experimental_rerun()