Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -117,32 +117,29 @@ def alphafold(input_set, mode, impute):
     data = clean_data(input_set)
 
     if len(data) == 0:
-        print('Feature vectore generation terminated. Please enter a query or check your input format.')
+        st.write('Feature vectore generation terminated. Please enter a query or check your input format.')
     else:
         data = add_uniprot_sequence(data)
-        st.write(data)
         match = data[(data.wt_sequence_match == 'm')]
-        st.write(match)
         org_len = len(match)
         iso = data[(data.wt_sequence_match == 'i')]
         noMatch = data[(data.wt_sequence_match != 'm') & (data.wt_sequence_match != 'i')]
-        st.write(noMatch)
         if len(noMatch) == len(data) :
-            print('>> Aminoacid at the position could not be mapped to canonical or isoform sequence. Please check the input amino acid.')
-            print('\n')
+            st.write('>> Aminoacid at the position could not be mapped to canonical or isoform sequence. Please check the input amino acid.')
+            st.write('\n')
         elif len(noMatch) > 0:
-            print(
+            st.write(
                 f'>> {len(noMatch)} of {len(data)} datapoints has not been mapped to any sequence. These datapoints are omitted.')
-            print('\n')
+            st.write('\n')
         if len(iso) > 0:
-            print(f'>> {len(iso)} of {len(data)} datapoints has been mapped to isoform sequences. These datapoints are omitted.')
-            print('\n')
+            st.write(f'>> {len(iso)} of {len(data)} datapoints has been mapped to isoform sequences. These datapoints are omitted.')
+            st.write('\n')
         if len(match) == 0:
-            print('>> Feature generation terminated due to failed mapping of input amino acid to UniProt sequence.')
-            print('\n')
+            st.write('>> Feature generation terminated due to failed mapping of input amino acid to UniProt sequence.')
+            st.write('\n')
         else:
-            print(f'>> {len(match)} of {len(data)} datapoints has been mapped to canonical sequences. Proceeding with these datapoins.')
-            print('\n')
+            st.write(f'>> {len(match)} of {len(data)} datapoints has been mapped to canonical sequences. Proceeding with these datapoins.')
+            st.write('\n')
             if (len(iso) != 0) | (len(noMatch) != 0):
                 print('Omitted datapoints are:', noMatch.datapoint.to_list() + iso.datapoint.to_list())
             data = match[['uniprotID', 'wt', 'pos', 'mut', 'datapoint']]