progress

app.py

@@ -11,6 +11,7 @@ ENTRY_METHODS = dict(
 DOCUMENTATION = st.container()
 MODE_SELECTION = st.container()
 TRANSCRIPT_ENTRY = st.container()
+RUNTIME = st.container()
 RESULTS = st.container()
 
 
@@ -90,12 +91,18 @@ if __name__ == '__main__':
     if 'mode' not in st.session_state:
         st.session_state.mode = tiger.RUN_MODES['all']
         st.session_state.disable_off_target_checkbox = True
+    if 'off_targets_checked' not in st.session_state:
+        st.session_state.off_targets_checked = False
     if 'entry_method' not in st.session_state:
         st.session_state.entry_method = ENTRY_METHODS['manual']
         st.session_state.manual_entry_disabled = False
         st.session_state.fasta_entry_disabled = True
     if 'run' not in st.session_state:
         st.session_state.run = False
+    if 'results_ready' not in st.session_state:
+        st.session_state.results_ready = False
+        st.session_state.on_target = pd.DataFrame()
+        st.session_state.off_target = pd.DataFrame()
 
     # title and documentation
     with DOCUMENTATION:
@@ -139,31 +146,50 @@ if __name__ == '__main__':
         )
         st.button(label='Get predictions!', on_click=process_input)
 
-    with RESULTS:
+    with RUNTIME:
+        # check and clear run signal and existing results
         if st.session_state.run:
             st.session_state.run = False
-            on_target, off_target = tiger.tiger_exhibit(
+            st.session_state.results_ready = False
+
+            # run model and signal results are ready
+            st.session_state.off_targets_checked = st.session_state.check_off_targets
+            st.session_state.on_target, st.session_state.off_target = tiger.tiger_exhibit(
                 transcripts=st.session_state.transcripts,
                 mode={v: k for k, v in tiger.RUN_MODES.items()}[st.session_state.mode],
                 status=st.empty(),
                 progress_bar=st.progress(0),
-                check_off_targets=st.session_state.check_off_targets
+                check_off_targets=st.session_state.off_targets_checked
             )
+            st.session_state.results_ready = True
 
-    # # valid input
-    # if src_seq and all([True if nt.upper() in NUCLEOTIDE_TOKENS.keys() else False for nt in src_seq]):
-    #     on_target, off_target = tiger_exhibit(pd.DataFrame(dict(id=['ManualEntry'], seq=[src_seq])),
-    #                                           status_bar, status_text, check_off_targets=option == 'On and Off Target')
-    #     on_target.rename(columns={'Guide': '23 nt guide sequence'}, inplace=True)
-    #     if len(on_target) > 0:
-    #         if on_target.iloc[0]['On-target ID'] == 0:
-    #             on_target.drop(['On-target ID'], axis=1, inplace=True)
-    #     st.write('On-target predictions: ', on_target)
-    #     st.download_button(label='Download', data=convert_df(on_target), file_name='on_target.csv', mime='text/csv')
-    #     if option == 'On and Off Target' and len(off_target) > 0:
-    #         off_target.rename(columns={'Guide': '23 nt guide sequence'}, inplace=True)
-    #         st.write('Off-target predictions: ', off_target)
-    #         st.download_button(label='Download', data=convert_df(off_target), file_name='off_target.csv', mime='text/csv')
-    #     elif option == 'On and Off Target' and len(off_target) == 0:
-    #         st.write('We did not find any off-target effects!')
-    #
+    with RESULTS:
+        on_target_results = st.empty()
+        off_target_results = st.empty()
+
+        # display results if they are ready
+        if st.session_state.results_ready:
+            with on_target_results:
+                st.write('On-target predictions:', st.session_state.on_target)
+                st.download_button(
+                    label='Download on-target predictions',
+                    data=convert_df(st.session_state.on_target),
+                    file_name='on_target.csv',
+                    mime='text/csv'
+                )
+            with off_target_results:
+                if st.session_state.off_targets_checked and len(st.session_state.off_target) > 0:
+                    st.write('Off-target predictions:', st.session_state.off_target)
+                    st.download_button(
+                        label='Download off-target predictions',
+                        data=convert_df(st.session_state.off_target),
+                        file_name='off_target.csv',
+                        mime='text/csv'
+                    )
+                elif st.session_state.off_targets_checked and len(st.session_state.off_target) == 0:
+                    st.write('We did not find any off-target effects!')
+
+        # otherwise, clear our results
+        else:
+            on_target_results.empty()
+            off_target_results.empty()