Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -262,10 +262,19 @@ def pdb(input_set, mode, impute):
                     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('ERROR')
                     st.write(f"PDB file {search}.pdb downloaded successfully.")
+
+
+                    from huggingface_hub import HfApi
+                    api = HfApi()
+                    api.upload_file(
+                        path_or_fileobj=response.content,
+                        path_in_repo="out_files/pdb/pdb_structures/",
+                        repo_id="HUBioDataLab/ASCARIS",
+                        repo_type="space")
+
                 else:
                     st.write('Here2')
                     st.write(f"Failed to retrieve PDB file for {search}.")
@@ -273,8 +282,7 @@ def pdb(input_set, mode, impute):
                 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"):