Update app.py

app.py

@@ -93,9 +93,7 @@ with st.form('mform', clear_on_submit=False):
     with st.spinner('In progress...This may take a while...'):
         try:
             if mode == 1:
-                selected_df = pdb_featureVector.pdb(input_set, mode, impute)
-                st.success('Feature vector successfully created.')
-    
+                selected_df = pdb_featureVector.pdb(input_set, mode, impute)    
                 
             elif mode == 2:
                 selected_df = alphafold_featureVector.alphafold(input_set, mode, impute)
@@ -106,6 +104,12 @@ with st.form('mform', clear_on_submit=False):
         except:
             selected_df = pd.DataFrame()
             pass
-            
+    st.success('Feature vector successfully created.')
+
+
+
+with st.form("my_form", clear_on_submit=False):
     st.text_input("Enter filename", key="filename")
-    submit = st.form_submit_button("Download", on_click=selected_df)
+    submit = st.form_submit_button("Download", on_click=selected_df)
+
+