Spaces:

HUBioDataLab
/

ASCARIS

Sleeping

App Files Files Community

fatmacankara commited on Jul 22, 2023

Commit

c8b993f

•

1 Parent(s): 0a0d39e

Update app.py

Browse files

Files changed (1) hide show

app.py +88 -84

app.py CHANGED Viewed

@@ -11,6 +11,33 @@ import argparse
 from st_aggrid import AgGrid, GridOptionsBuilder, JsCode,GridUpdateMode
 import base64
 showWarningOnDirectExecution = False
 # Check if 'key' already exists in session_state
 # If not, then initialize it
 if 'visibility' not in st.session_state:
@@ -55,89 +82,66 @@ with st.form('mform'):
     mode = source
     impute = impute
-print('*****************************************')
-print('Feature vector generation is in progress. \nPlease check log file for updates..')
-print('*****************************************')
-mode = int(mode)
-with st.spinner('In progress...This may take a while...'):
-    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.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.')
-        else:
             int_return = pd.DataFrame()
-    except:
-        pass
-        selected_df = pd.DataFrame()
-        int_return = pd.DataFrame()
-def download_button(object_to_download, download_filename):
-    if isinstance(object_to_download, pd.DataFrame):
-        object_to_download = object_to_download.to_csv(index=False)
-    # Try JSON encode for everything else
-    else:
-        object_to_download = json.dumps(object_to_download)
-    try:
-        # some strings <-> bytes conversions necessary here
-        b64 = base64.b64encode(object_to_download.encode()).decode()
-    except AttributeError as e:
-        b64 = base64.b64encode(object_to_download).decode()
-    dl_link = f"""<html><head><title>Start Auto Download file</title><script src="http://code.jquery.com/jquery-3.2.1.min.js"></script><script>$('<a href="data:text/csv;base64,{b64}" download="{download_filename}">')[0].click()</script></head></html>"""
-    return dl_link
-def download_df():
-    components.html(
-        download_button(selected_df, st.session_state.filename),
-        height=0,
-    )
-selected_row = int_return["selected_rows"]
-selected_df = pd.DataFrame(selected_row, columns=selected_df.columns)
-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)

 from st_aggrid import AgGrid, GridOptionsBuilder, JsCode,GridUpdateMode
 import base64
 showWarningOnDirectExecution = False
+def download_button(object_to_download, download_filename):
+    if isinstance(object_to_download, pd.DataFrame):
+        object_to_download = object_to_download.to_csv(index=False)
+    # Try JSON encode for everything else
+    else:
+        object_to_download = json.dumps(object_to_download)
+    try:
+        # some strings <-> bytes conversions necessary here
+        b64 = base64.b64encode(object_to_download.encode()).decode()
+    except AttributeError as e:
+        b64 = base64.b64encode(object_to_download).decode()
+    dl_link = f"""<html><head><title>Start Auto Download file</title><script src="http://code.jquery.com/jquery-3.2.1.min.js"></script><script>$('<a href="data:text/csv;base64,{b64}" download="{download_filename}">')[0].click()</script></head></html>"""
+    return dl_link
+def download_df():
+    components.html(
+        download_button(selected_df, st.session_state.filename),
+        height=0,
+    )
 # Check if 'key' already exists in session_state
 # If not, then initialize it
 if 'visibility' not in st.session_state:
     mode = source
     impute = impute
+    print('*****************************************')
+    print('Feature vector generation is in progress. \nPlease check log file for updates..')
+    print('*****************************************')
+    mode = int(mode)
+    with st.spinner('In progress...This may take a while...'):
+        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.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.')
+            else:
+                int_return = pd.DataFrame()
+        except:
+            pass
+            selected_df = pd.DataFrame()
             int_return = pd.DataFrame()
+    selected_row = int_return["selected_rows"]
+    selected_df = pd.DataFrame(selected_row, columns=selected_df.columns)
+    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)