Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -257,13 +257,22 @@ def pdb(input_set, mode, impute):
                 if response.status_code == 200:
                     st.write('here1')
                     # Save the PDB file to a local file
-                    with open(f"path_pdb/{search}.pdb", "wb") as f:
-                        f.write(response.content)
+                    try:
+                        with open(Path(path_to_output_files / search/'.pdb'), "wb") as f:
+                            st.write('WRITING TO FILE')
+                            f.write(response.content)
+                    except:
+                        st.write('ERRor')
                     st.write(f"PDB file {search}.pdb downloaded successfully.")
                 else:
-                    st.write('here2')
+                    st.write('Here2')
                     st.write(f"Failed to retrieve PDB file for {search}.")
                 st.write('what')
+                existing_pdb = list(Path(path_to_output_files / 'pdb_structures').glob("*"))
+                st.write('existing_pdb3', existing_pdb)
+
+
+                
     
                 resolution_method = parser.get_structure(search, file)
                 for record in SeqIO.parse(file, "pdb-seqres"):