Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -718,29 +718,31 @@ def alphafold(input_set, mode, impute):
                         if sasa_pos != np.NaN:
                             sasa = 'nan'
                             for filename in  list(Path(path_to_output_files / 'freesasa_files').glob("*")):
-
-                                if 'readme' not in filename:
-                                    st.write('xdfkjlxdjf', filename)
-                                    st.write('string filename')
-                                    st.write(str(filename))
-                                    st.write('string filename split')
-                                    st.write(str(filename).split('.'))
+                                st.write('xdfkjlxdjf', filename)
+                                st.write('string filename')
+                                st.write(str(filename))
+                                st.write('string filename split')
+                                st.write(str(filename).split('.'))
+                                try:
                                     st.write(list(filter(None, str(filename).split('.')))[0].split('/')[-1].split('-')[1].upper())
                                     fname = list(filter(None, str(filename).split('.')))[0].split('/')[-1].split('-')[1].upper()
                                     st.write('fname', fname)
-                
-                                    if uniprotID == fname:
-                                        files = open(filename, 'r')
-                                        file = files.readlines()
-                                        for k in file:
-                                            if str(k.strip()[10:13]) == str(sasa_pos):
-                                                residue = str(k[4:7].strip())
-                                                if wt == threeToOne(residue):
-                                                    sasa = str(k[22:28]).strip('\n')
-                                                elif wt != threeToOne(residue):
-                                                    sasa = str(k[22:28]).strip('\n') + '*'
-                                                else:
-                                                    sasa = 'nan'
+                                except IndexError:
+                                    
+                                    st.write('IndexError')
+                                    fname = ''
+                                if uniprotID == fname:
+                                    files = open(filename, 'r')
+                                    file = files.readlines()
+                                    for k in file:
+                                        if str(k.strip()[10:13]) == str(sasa_pos):
+                                            residue = str(k[4:7].strip())
+                                            if wt == threeToOne(residue):
+                                                sasa = str(k[22:28]).strip('\n')
+                                            elif wt != threeToOne(residue):
+                                                sasa = str(k[22:28]).strip('\n') + '*'
+                                            else:
+                                                sasa = 'nan'
                 
                         else:
                             sasa = 'nan'