Update code/manage_files.py

code/manage_files.py

@@ -1,4 +1,5 @@
 import os
+import streamlit as st
 from pathlib import Path
 def manage_files(mode):
     if mode== 1:
@@ -32,7 +33,11 @@ def manage_files(mode):
         alphafold_summary = path_to_input_files / 'alphafold_summary.txt'
         path_to_interfaces = path_to_input_files / 'H_sapiens_interfacesHQ.txt'
         # Unzip before using
-        alphafold_path = Path(path_to_input_files/'alphafold_structures')
+        from datasets import load_dataset
+        alphafold_path = load_dataset("HUBioDataLab/AlphafoldStructures")
+        st.write('path')
+        st.write(alphafold_path)
+        ###alphafold_path = Path(path_to_input_files/'alphafold_structures')
 
         os.makedirs('out_files', exist_ok=True)
         path_to_output_files = Path('out_files/alphafold')