Update code/process_input.py

code/process_input.py

@@ -6,15 +6,22 @@ def clean_data(input_set):
         if ',' in input_set:
             input_set = [i.strip() for i in input_set.split(',')]
             initial_len = len(input_set)
+            
             for i in input_set:
                 data = data.append(pd.Series([j.strip() for j in i.split('-')]), ignore_index=True)
             data.columns = ['uniprotID', 'wt', 'pos', 'mut']
+            if initial_len != len(data):
+                st.write(f'{initial_len- len(data)} of {initial_len} datapoints is omitted. Check your input.')
         elif '\t' in input_set:
             input_set = [i.strip() for i in input_set.split('\t')]
             initial_len = len(input_set)
+            
             for i in input_set:
                 data = data.append(pd.Series([j.strip() for j in i.split('-')]), ignore_index=True)
             data.columns = ['uniprotID', 'wt', 'pos', 'mut']
+            if initial_len != len(data):
+                st.write(f'{initial_len- len(data)} of {initial_len} datapoints is omitted. Check your input.')
+            for i in input_set:
 
         elif '-' in input_set:
             data = data.append(pd.Series([j.strip() for j in input_set.split('-')]), ignore_index=True)
@@ -38,8 +45,7 @@ def clean_data(input_set):
             data.at[i, 'datapoint'] = data.at[i, 'uniprotID'] + data.at[i, 'wt'] + str(data.at[i, 'pos']) + data.at[i, 'mut']
 
         data = data.astype(str)
-        if initial_len != len(data):
-            st.write(f'{initial_len- len(data)} of {initial_len} datapoints is omitted. Check your input.')
+        
         return data
     except ValueError:
         st.write('Your input is in the wrong format. Please see the example.')