Update app.py

app.py

@@ -41,12 +41,12 @@ with st.sidebar:
     input_type = st.radio('Select input type', ['Enter sequence', 'Upload FASTA file'])
     if input_type == 'Enter sequence':
         if st.session_state.example_seq_button:
-            sequence = st.text_area('Enter protein sequence in FASTA format.', 
+            st.session_state['sequence'] = st.text_area('Enter protein sequence in FASTA format.', 
                 value='>sp|O18783|PLMN_NOTEU\n'
                 'MEYGKVIFLFLLFLKSGQGESLENYIKTEGASLSNSQKKQFVASSTEECEALCEKETEFVCRSFEHYNKEQKCVIMSENSKTSSVERKRDVVLFEKRIYLSDCKSGNGRNYRGTLSKTKSGITCQKWSDLSPHVPNYAPSKYPDAGLEKNYCRNPDDDVKGPWCYTTNPDIRYEYCDVPECEDECMHCSGENYRGTISKTESGIECQPWDSQEPHSHEYIPSKFPSKDLKENYCRNPDGEPRPWCFTSNPEKRWEFCNIPRCSSPPPPPGPMLQCLKGRGENYRGKIAVTKSGHTCQRWNKQTPHKHNRTPENFPCRGLDENYCRNPDGELEPWCYTTNPDVRQEYCAIPSCGTSSPHTDRVEQSPVIQECYEGKGENYRGTTSTTISGKKCQAWSSMTPHQHKKTPDNFPNADLIRNYCRNPDGDKSPWCYTMDPTVRWEFCNLEKCSGTGSTVLNAQTTRVPSVDTTSHPESDCMYGSGKDYRGKRSTTVTGTLCQAWTAQEPHRHTIFTPDTYPRAGLEENYCRNPDGDPNGPWCYTTNPKKLFDYCDIPQCVSPSSFDCGKPRVEPQKCPGRIVGGCYAQPHSWPWQISLRTRFGEHFCGGTLIAPQWVLTAAHCLERSQWPGAYKVILGLHREVNPESYSQEIGVSRLFKGPLAADIALLKLNRPAAINDKVIPACLPSQDFMVPDRTLCHVTGWGDTQGTSPRGLLKQASLPVIDNRVCNRHEYLNGRVKSTELCAGHLVGRGDSCQGDSGGPLICFEDDKYVLQGVTSWGLGCARPNKPGVYVRVSRYISWIEDVMKNN')
         else:
-            sequence = st.text_input('Enter protein sequence in FASTA format.')
-        name = sequence.split('\n')[0].strip('>')
+            st.session_state['sequence'] = st.text_input('Enter protein sequence in FASTA format.')
+        st.session_state['name'] = st.session_state['sequence'].split('\n')[0].strip('>')
         st.button('Use example sequence', on_click=click_button)
     else:
         protein_input = st.file_uploader('Choose file')
@@ -54,14 +54,14 @@ with st.sidebar:
             protein_input_stringio = StringIO(protein_input.getvalue().decode("utf-8"))
             fasta_sequences = SeqIO.parse(protein_input_stringio, 'fasta')
             for fasta in fasta_sequences:
-                name, sequence = fasta.id, str(fasta.seq)
+                st.session_state['name'], st.session_state['sequence'] = fasta.id, str(fasta.seq)
 
-    email = st.text_input('Enter your email for InterProScan query: ')
+    st.session_state['email'] = st.text_input('Enter your email for InterProScan query: ')
 
     # prevent user from clicking 'Find domains' button if email or sequence is empty
     domains_submitted = False  
     if st.button('Find domains'):
-        if email and sequence:
+        if email in st.session_state and sequence in st.session_state:
             domains_submitted = True
             st.session_state.disabled = True
         else:
@@ -69,12 +69,13 @@ with st.sidebar:
     else:
         with domain_tab:
             st.warning('Please enter your query and click "Find domains" to see domains in sequence.')
+
 with domain_tab:
     no_domains = False
     error_in_interproscan = False
     if domains_submitted:
         with st.spinner('Finding domains in sequence using InterProScan. This may take a while...'):
-            result = find_domains(email, sequence, name)
+            result = find_domains(st.session_state.email, st.session_state.sequence, st.session_state.name)
         result_text = result[0]
         if result_text == 'Domains found.':
             st.success(result_text + ' You can now see function predictions for the sequence in the "Function predictions" tab.')
@@ -96,7 +97,7 @@ with domain_tab:
             st.download_button(
                 label="Download domains in sequence as CSV",
                 data=domains_csv,
-                file_name=f"{name}_domains.csv",
+                file_name=f"{st.session_state.name}_domains.csv",
                 mime="text/csv",
             )
 
@@ -129,7 +130,7 @@ with pred_tab:
                 st.download_button(
                     label="Download function predictions as CSV",
                     data=pred_csv,
-                    file_name=f"{name}_function_predictions.csv",
+                    file_name=f"{st.session_state.name}_function_predictions.csv",
                     mime="text/csv",
                 )