Update ASCARIS.py

ASCARIS.py

@@ -33,18 +33,19 @@ st.write('')
 st.write('')
 
 with st.form('mform', clear_on_submit=False):
-    source = st.selectbox('Select the protein structure resource (1: PDB-SwissModel-Modbase, 2: AlphaFold)',[1,2])
+    #source = st.selectbox('Select the protein structure resource (1: PDB-SwissModel-Modbase, 2: AlphaFold)',[1,2])
+    source = 1
     impute = st.selectbox('Imputation',[True, False])
     
-    input_data = st.text_input('Enter SAV data points (Format Provided Below)\nQ9Y4W6-N-432-T'
+    input_data = st.text_input('Enter SAV data points (Example: Q9Y4W6-N-432-T')
             )
 
 
     parser = argparse.ArgumentParser(description='ASCARIS')
     
-    parser.add_argument('-s', '--source_option',
-                        help='Selection of input structure data.\n 1: PDB Structures (default), 2: AlphaFold Structures',
-                        default=1)
+    #parser.add_argument('-s', '--source_option',
+    #                    help='Selection of input structure data.\n 1: PDB Structures (default), 2: AlphaFold Structures',
+    #                    default=1)
     parser.add_argument('-i', '--input_datapoint',
                         help='Input file or query datapoint\n Option 1: Comma-separated list of identifiers (UniProt ID-wt residue-position-mutated residue (e.g. Q9Y4W6-N-432-T or Q9Y4W6-N-432-T, Q9Y4W6-N-432-T)) \n Option 2: Enter comma-separated file path')
     
@@ -64,6 +65,7 @@ with st.form('mform', clear_on_submit=False):
     
     
 selected_df = pd.DataFrame()
+st.write('The online tool may be slow, especially while processing multiple SAVs, please consider using the local programmatic version at https://github.com/HUBioDataLab/ASCARIS/')
 if submitted:
     with st.spinner('In progress...This may take a while...'):
         try:
@@ -89,11 +91,11 @@ if submitted:
             st.success('Feature vector successfully created.')
             csv = convert_df(selected_df)
     
-            st.download_button("Press to Download the Feature Vector", csv,"ASCARIS_SAV_rep.csv","text/csv",key='download-csv')
+            st.download_button("Press to Download the Feature Vector", csv,f"ASCARIS_SAV_rep_{input_set}.csv","text/csv",key='download-csv')
 
         else:
             st.success('Feature vector failed. Check log file.')
-            st.write('Failed here2')
+