From 72ea43b6fc51e42a580b77953936cdaf1f95724f Mon Sep 17 00:00:00 2001
From: leo <leolswq@163.com>
Date: Wed, 6 Mar 2024 22:13:04 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=9A=E4=B9=89=E5=88=97=E5=90=8D=E6=98=A0?=
 =?UTF-8?q?=E5=B0=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 data.db | Bin 28672 -> 28672 bytes
 main.py |  16 +++++++++++++---
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/data.db b/data.db
index 100264c054a6e13f08a893f17b76e6f2f8d26801..6f43ea817c9971475d7bdeed1e72b1ee6c5e8305 100644
GIT binary patch
delta 62
zcmZp8z}WDBae_2s?L--8#@dYue<T<=CdW$N1(McMKytIB^lHY*-=z5`UzXVqr1>^4
OknLjxitEc?3;+PitQM#M

delta 58
zcmZp8z}WDBae_2s;Y1l{#=?yWe<T<=C&x<O1(McMKytIB^lC;R=cvqCAZN1dY#?W|
IzWl`i08Xb9LjV8(

diff --git a/main.py b/main.py
index b34f205..235da6e 100644
--- a/main.py
+++ b/main.py
@@ -9,6 +9,8 @@ from db_utils import BatchData, get_session
 st.session_state.setdefault('data_table', [])
 
 
+# session: Session = next(get_session())
+
 def get_data_from_db():
     logger.info("init")
     return (BatchDataRead.from_orm(db_obj).dict()
@@ -28,10 +30,17 @@ def update_handler(*args, **kwargs):
 
 edited_df = st.data_editor(df, key="edited_info",
                            hide_index=True,
-                           on_change=update_handler,
-                           column_order=('id', 'year', 'is_validation'),
                            use_container_width=True,
-                           )  # 👈 An editable dataframe
+                           on_change=update_handler,
+                           column_order=('id', 'year', 'census_batch', 'id_code', 'precision', 'is_train', 'is_validation'),
+                           column_config={
+                               "year": st.column_config.NumberColumn("年份", format="%d 年", ),
+                               'census_batch': "普查批次",
+                               'id_code': "编号",
+                               'precision': "精度",
+                               'is_train': "是否是训练集",
+                               'is_validation': "是否是验证集",
+                           })
 
 favorite_command = edited_df.loc[edited_df["id"].idxmax()]
 data_table_change_info = st.session_state['edited_info']
@@ -44,6 +53,7 @@ for id_, update_data in edited_rows.items():
     for field in update_data:
         setattr(row_db, field, update_data[field])
     st.text(f"update: {id_=}, {update_data=}")
+    st.toast('Your edited image was saved!', icon='😍')
     session.commit()
     st.rerun()
 logger.info('end')