Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -167,11 +167,14 @@ def pdb(input_set, mode, impute):
 
         data.wt_sequence_match = data.wt_sequence_match.astype('str')
         data.replace({'': 'nan'}, inplace=True)
+        
         data_size = len(data.drop_duplicates(['datapoint']))
         not_match_in_uniprot = data[(data.uniprotSequence == 'nan') | (data.wt_sequence_match == 'nan')]
         uniprot_matched = data[(data.uniprotSequence != 'nan') & (data.wt_sequence_match != 'nan')]
-        st.write('DAFSDSD')
-        st.write(data)
+        if (len(uniprot_matched) == 0) & (len(not_match_in_uniprot) == 0):
+            st.write('Please check the input aminoacid at the selected position.')
+
+        
         data = None
 
         print('You have %d data points that failed to match a UniProt Sequence\nProceeding with %d remaining...\n'