如何编辑Streamlit AgGrid行的单元格？

我已经创建了AgGrid通过从 csv 文件加载数据。我通过外部按钮逐行添加行。但是当我尝试编辑我添加的行时，它消失了。如果您能帮助我找出错误所在，我将不胜感激。代码如下。

import pandas as pd
import streamlit as st
from st_aggrid import AgGrid, GridUpdateMode,  JsCode
from st_aggrid.grid_options_builder import GridOptionsBuilder
import sys
import os
import altair as alt
from streamlit.runtime.legacy_caching import caching

def data_upload():
    df = pd.read_csv("data.csv")
    return df

if 'grid' in st.session_state:
    grid_table = st.session_state['grid']
    df = pd.DataFrame(grid_table['data'])
    df.to_csv(“data.csv”, index=False)
else:
    df = data_upload()

gd = GridOptionsBuilder.from_dataframe(df)
gd.configure_column("Location", editable=True)
gd.configure_column("HourlyRate", editable=True)
gd.configure_column("CollaboratorName", editable=True)

gridOptions = gd.build()


button = st.sidebar.button("Add Line")

if "button_state" not in st.session_state:
    st.session_state.button_state = False

if button or st.session_state.button_state:
    st.session_state.button_state = True
    data = [['', '', 0]]
    df_empty = pd.DataFrame(data, columns=['CollaboratorName', 'Location', "HourlyRate"])
    df = pd.concat([df, df_empty], axis=0, ignore_index=True)
    df.to_csv(“data.csv”, index=False)
    gd= GridOptionsBuilder.from_dataframe(df)


grid_table = AgGrid(df,
                    gridOptions=gridOptions,
                    fit_columns_on_grid_load=True,
                    height=500,
                    width='100%',
                    theme="streamlit",
                    key= 'unique',
                    update_mode=GridUpdateMode.GRID_CHANGED,
                    reload_data=True,
                    allow_unsafe_jscode=True,
                    editable=True
                    )
if 'grid' not in st.session_state:
    st.session_state['grid'] = grid_table
else:
    grid_table_df = pd.DataFrame(grid_table['data'])
    grid_table_df.to_csv(“data.csv”, index=False) 

您可以从这里看到正在运行的应用程序在此输入图像描述

这个有不同的方法，但目标可能是相同的。

如果值为，则创建两个单选按钮yes如果值为no没有新线。

如果您想添加新行，请选择“是”，然后添加您的条目​​。然后按更新按钮在侧边栏中。

如果您想编辑但不添加新行，请选择no，编辑现有条目，然后按更新按钮.

Code

import streamlit as st
from st_aggrid import AgGrid, GridOptionsBuilder
import pandas as pd


def data_upload():
    df = pd.read_csv("data.csv")
    return df


def show_grid(newline):
    st.header("This is AG Grid Table")
    df = data_upload()

    if newline == 'yes':
        data = [['', '', 0]]
        df_empty = pd.DataFrame(data, columns=['CollaboratorName', 'Location', "HourlyRate"])
        df = pd.concat([df, df_empty], axis=0, ignore_index=True)

    gb = GridOptionsBuilder.from_dataframe(df)
    gb.configure_default_column(editable=True)
    grid_table = AgGrid(
        df,
        height=400,
        gridOptions=gb.build(),
        fit_columns_on_grid_load=True,
        allow_unsafe_jscode=True,
    )
    return grid_table


def update(grid_table):
    grid_table_df = pd.DataFrame(grid_table['data'])
    grid_table_df.to_csv('data.csv', index=False)


# start
addline = st.sidebar.radio('Add New Line', options=['yes', 'no'], index=1, horizontal=True)
grid_table = show_grid(addline)
st.sidebar.button("Update", on_click=update, args=[grid_table])