Update app_v3.py

app_v3.py

@@ -341,9 +341,9 @@ def _(
 
     s5 = mo.md(f'''
         ###**Helper Purge Functions:**
-
-        These functions help you retrieve and mass delete ***(WARNING: purges all at once)*** deployments, data assets or repository assets (functions, models, etc.) that you have in the deployment space. This is meant to support fast cleanup.
-
+        
+        These functions help you retrieve, select and delete deployments, data assets or repository assets (functions, models, etc.) that you have in the deployment space. This is meant to support fast cleanup.
+        
         Select the tab based on what you want to delete, then click each of the buttons one by one after the previous gives a response.
     
         ---
@@ -536,17 +536,6 @@ def _(deployment_client, mo, pd):
             initial_selection=[0]
         )
 
-    # # Display the table
-    # mo.md(f"""---
-    # <br>
-    # <br>
-    # {selection_table}
-    # <br>
-    # <br>
-    # ---
-    # <br>
-    # <br>
-    # """)
     return (
         framework_mapping,
         preferred_order,
@@ -615,15 +604,6 @@ def _(
             justify="center", gap=1, align="center", wrap=True
         )
 
-    # Display the metadata inputs
-    # mo.vstack([
-    #     func_metadata,
-    #     mo.md("**Make sure to click the checkboxes before filling in descriptions and tags or they will reset.**"),
-    #     schema_metadata
-    #     ],
-    #     align="center",
-    #     gap=2
-    # )
     fm = mo.vstack([
         func_metadata,
         ],
@@ -974,20 +954,6 @@ def _(deployment_client, mo, pd, upload_button, uuid):
             initial_selection=[0]
         )
 
-
-    # mo.md(f"""
-    # <br>
-    # <br>
-    # {upload_func}
-    # <br>
-    # <br>
-    # ---
-    # {hw_selection_table}
-    # <br>
-    # <br>
-
-
-    # """)
     return (
         deployment_name,
         deployment_type,
@@ -1108,14 +1074,6 @@ def _(deploy_button, deployment_definition, mo):
         deploy_button.value
     ], justify="space-around", align="center")
 
-    # mo.md(f"""
-    # {deployment_definition}
-    # <br>
-    # <br>
-    # {deploy_fnc}
-
-    # ---
-    # """)
     return (deploy_fnc,)
 
 
@@ -1125,9 +1083,8 @@ def _(deployment_client, mo):
 
     def get_deployment_list():
         dep_df = deployment_client.deployments.list()
-        dep_df_processed = pd.DataFrame(dep_df)
-        deployment_df = mo.ui.table(dep_df_processed, initial_selection=[0])
-        return deployment_df
+        dep_df = pd.DataFrame(dep_df)
+        return dep_df
 
     def get_deployment_ids(df):
         dep_list = df['ID'].tolist()
@@ -1137,21 +1094,19 @@ def _(deployment_client, mo):
 
     def get_data_assets_list():
         data_a_df = deployment_client.data_assets.list()
-        data_a_df_processed = pd.DataFrame(data_a_df)
-        data_assets_df = mo.ui.table(data_a_df_processed, initial_selection=[0])
-        return data_assets_df
+        data_a_df = pd.DataFrame(data_a_df)
+        return data_a_df
 
     def get_data_asset_ids(df):
         data_asset_list = df['ASSET_ID'].tolist()
         return data_asset_list
 
     #----
-    
+
     def get_repository_list():
         rep_list_df = deployment_client.repository.list()
-        rep_list_df_processed = pd.DataFrame(rep_list_df)
-        repository_df = mo.ui.table(rep_list_df_processed, initial_selection=[0])
-        return repository_df
+        rep_list_df = pd.DataFrame(rep_list_df)
+        return rep_list_df
 
     def get_repository_ids(df):
         repository_list = df['ID'].tolist()
@@ -1214,6 +1169,24 @@ def _(deployment_client, mo):
         get_repository_list,
     )
 
+@app.cell
+def _(get_data_assets_tab, get_deployments_tab, get_repository_tab, mo):
+    if get_deployments_tab() is not None:
+        deployments_table = mo.ui.table(get_deployments_tab())
+    else:
+        deployments_table = mo.md("No Table Loaded")
+
+    if get_repository_tab() is not None:
+        repository_table = mo.ui.table(get_repository_tab())
+    else:
+        repository_table = mo.md("No Table Loaded")
+
+    if get_data_assets_tab() is not None:
+        data_assets_table = mo.ui.table(get_data_assets_tab())
+    else:
+        data_assets_table = mo.md("No Table Loaded")
+        
+    return data_assets_table, deployments_table, repository_table
 
 @app.cell
 def _(get_deployment_id_list, get_deployments_button, mo, purge_deployments):
@@ -1261,23 +1234,31 @@ def _(data_assets_purge_tab, deployments_purge_tab, mo, repository_purge_tab):
         {"Purge Deployments": deployments_purge_tab, "Purge Repository Assets": repository_purge_tab,"Purge Data Assets": data_assets_purge_tab }, lazy=False
     )
 
-    # asset_purge = mo.accordion(
-    #     {
-    #         """<br>
-    # #### **Supporting Cleanup Functionality, lists of different assets and purge them if needed** *(purges all detected)*
-    # <br>""": purge_tabs,
-    #     }
-    # )
-
     # asset_purge
     return (purge_tabs,)
 
+@app.cell
+def _(mo):
+    get_deployments_tab, set_deployments_tab = mo.state(None)
+    get_repository_tab, set_repository_tab = mo.state(None)
+    get_data_assets_tab, set_data_assets_tab = mo.state(None)
+    return (
+        get_data_assets_tab,
+        get_deployments_tab,
+        get_repository_tab,
+        set_data_assets_tab,
+        set_deployments_tab,
+        set_repository_tab,
+    )
+
 
 @app.cell(hide_code=True)
 def _(
+    data_assets_table,
     delete_data_assets,
     delete_deployments,
     delete_repository_items,
+    deployments_table,
     get_data_asset_ids,
     get_data_assets_list,
     get_deployment_ids,
@@ -1285,23 +1266,28 @@ def _(
     get_repository_ids,
     get_repository_list,
     mo,
+    repository_table,
+    set_data_assets_tab,
+    set_deployments_tab,
+    set_repository_tab,
 ):
     ### Temporary Function Purge - Assets
     get_data_assets_button = mo.ui.button(
         label="Get Data Assets Dataframe",
         on_click=lambda _: get_data_assets_list(),
+        on_change=lambda value: set_data_assets_tab(value),
         kind="neutral",
     )
 
     get_data_asset_id_list = mo.ui.button(
         label="Turn Dataframe into List of IDs",
-        on_click=lambda _: get_data_asset_ids(get_data_assets_button.value.value),
+        on_click=lambda _: get_data_asset_ids(get_data_assets_button.value),
         kind="neutral",
     )
 
     purge_data_assets = mo.ui.button(
         label="Purge Data Assets",
-        on_click=lambda _: delete_data_assets(get_data_asset_id_list.value),
+        on_click=lambda _: delete_data_assets(data_assets_table.value),
         kind="danger",
     )
 
@@ -1309,12 +1295,13 @@ def _(
     get_deployments_button = mo.ui.button(
         label="Get Deployments Dataframe",
         on_click=lambda _: get_deployment_list(),
+        on_change=lambda value: set_deployments_tab(value),
         kind="neutral",
     )
 
     get_deployment_id_list = mo.ui.button(
         label="Turn Dataframe into List of IDs",
-        on_click=lambda _: get_deployment_ids(get_deployments_button.value.value),
+        on_click=lambda _: get_deployment_ids(deployments_table.value),
         kind="neutral",
     )
 
@@ -1328,12 +1315,13 @@ def _(
     get_repository_button = mo.ui.button(
         label="Get Repository Dataframe",
         on_click=lambda _: get_repository_list(),
+        on_change=lambda value: set_repository_tab(value),
         kind="neutral",
     )
 
     get_repository_id_list = mo.ui.button(
         label="Turn Dataframe into List of IDs",
-        on_click=lambda _: get_repository_ids(get_repository_button.value.value),
+        on_click=lambda _: get_repository_ids(repository_table.value),
         kind="neutral",
     )