Update app.py

app.py

@@ -81,7 +81,6 @@ with st.form('mform', clear_on_submit=False):
     mode = source
     impute = impute
     submitted = st.form_submit_button(label="Submit", help=None, on_click=None, args=None, kwargs=None, type="secondary", disabled=False, use_container_width=False)
-    st.write('submitted', submitted)
     print('*****************************************')
     print('Feature vector generation is in progress. \nPlease check log file for updates..')
     print('*****************************************')
@@ -95,64 +94,18 @@ with st.form('mform', clear_on_submit=False):
         try:
             if mode == 1:
                 selected_df = pdb_featureVector.pdb(input_set, mode, impute)
-                """
-                int_builder = GridOptionsBuilder.from_dataframe(selected_df)
-                int_builder.configure_default_column(editable=False, filterable=True, cellStyle={'text-align': 'center'})
-                int_builder.configure_pagination(enabled=True, paginationAutoPageSize=False, paginationPageSize=10)
-                int_builder.configure_selection(selection_mode='multiple', use_checkbox=True)
-                gridoptions = int_builder.build()
-                int_return = AgGrid(selected_df,
-                            width='100%',
-                            height=(len(selected_df) + 4) * 35.2 + 3,
-                            theme='light',
-                            enable_enterprise_modules=False,
-                            gridOptions=gridoptions,
-                            fit_columns_on_grid_load=False,
-                            update_mode=GridUpdateMode.SELECTION_CHANGED, # or MODEL_CHANGED
-                            custom_css={".ag-header-cell-label": {"justify-content": "center"}})
-                st.write('INT return1', int_return)
-                """
                 st.success('Feature vector successfully created.')
     
                 
             elif mode == 2:
                 selected_df = alphafold_featureVector.alphafold(input_set, mode, impute)
-                """
-                int_builder = GridOptionsBuilder.from_dataframe(selected_df)
-                int_builder.configure_default_column(editable=False, filterable=True, cellStyle={'text-align': 'center'})
-                int_builder.configure_pagination(enabled=True, paginationAutoPageSize=False, paginationPageSize=10)
-                int_builder.configure_selection(selection_mode='multiple', use_checkbox=True)
-                gridoptions = int_builder.build()
-                int_return = AgGrid(selected_df,
-                            width='100%',
-                            height=(len(selected_df) + 4) * 35.2 + 3,
-                            theme='light',
-                            enable_enterprise_modules=False,
-                            gridOptions=gridoptions,
-                            fit_columns_on_grid_load=False,
-                            update_mode=GridUpdateMode.SELECTION_CHANGED, # or MODEL_CHANGED
-                            custom_css={".ag-header-cell-label": {"justify-content": "center"}})
-                st.success('Feature vector successfully created.')
-                st.write('INT return2', int_return)
-                """
             else:
-                
-                int_return = pd.DataFrame()
-                st.write('INT return3', int_return)
-            #st.write('INTTTT', int_return)
-            #selected_row = int_return["selected_rows"]
-            #st.write('selected_rows', selected_rows)
-            #st.write('selected_row', selected_row)
-            #selected_df = pd.DataFrame(selected_row, columns=selected_df.columns)
+                selected_df =  pd.DataFrame()
+
             st.write(selected_df)
-            st.text_input("Enter filename", key="filename")
-            submit = st.form_submit_button("Download", on_click=selected_df)
-            
         except:
-            pass
             selected_df = pd.DataFrame()
-            #int_return = pd.DataFrame()
-            #st.write('INT return4', int_return)
-    
-    
-    
+            pass
+            
+    st.text_input("Enter filename", key="filename")
+    submit = st.form_submit_button("Download", on_click=selected_df)