Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -257,9 +257,9 @@ def pdb(input_set, mode, impute):
                 if response.status_code == 200:
                     st.write('here1')
                     # Save the PDB file to a local file
-                    st.write(Path(path_to_output_files / search/'.pdb'))
+                    #st.write(f'out_files/pdb/pdb_structures/{search}.pdb')
                     #try:
-                    #    with open(Path(path_to_output_files / search/'.pdb'), "wb") as f:
+                    #    with open(f'out_files/pdb/pdb_structures/{search}.pdb', "wb") as f:
                     #        st.write('WRITING TO FILE')
                     #except:
                     #    st.write('ERROR')
@@ -268,6 +268,7 @@ def pdb(input_set, mode, impute):
 
                     from huggingface_hub import HfApi
                     api = HfApi()
+                    st.write('api', api)
                     api.upload_file(
                         path_or_fileobj=response.content,
                         path_in_repo="out_files/pdb/pdb_structures/",
@@ -277,6 +278,7 @@ def pdb(input_set, mode, impute):
                 else:
                     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)