Fangrui Liu commited on
Commit
a988660
β€’
1 Parent(s): 3ca6dfb

reset sel_sess everytime change the setting

Browse files
Files changed (1) hide show
  1. chat.py +9 -9
chat.py CHANGED
@@ -68,14 +68,6 @@ def on_session_change_submit():
68
  session_id=f"{st.session_state.user_name}?{st.session_state.current_sessions[elem]['session_id']}",
69
  )
70
  refresh_sessions()
71
- if len(st.session_state.session_editor["deleted_rows"]) > 0:
72
- try:
73
- dfl_indx = [
74
- x["session_id"] for x in st.session_state.current_sessions
75
- ].index("default")
76
- except ValueError:
77
- dfl_indx = 0
78
- st.session_state.sel_sess = st.session_state.current_sessions[dfl_indx]
79
  except Exception as e:
80
  sleep(2)
81
  st.error(f"{type(e)}: {str(e)}")
@@ -98,6 +90,13 @@ def refresh_sessions():
98
  st.session_state[
99
  "current_sessions"
100
  ] = st.session_state.session_manager.list_sessions(st.session_state.user_name)
 
 
 
 
 
 
 
101
  if type(st.session_state.current_sessions) is not dict and len(st.session_state.current_sessions) <= 0:
102
  st.session_state.session_manager.add_session(
103
  st.session_state.user_name,
@@ -149,7 +148,8 @@ def chat_page():
149
  dfl_indx = [
150
  x["session_id"] for x in st.session_state.current_sessions
151
  ].index("default")
152
- except ValueError:
 
153
  dfl_indx = 0
154
  st.selectbox(
155
  "Choose a session be chat:",
 
68
  session_id=f"{st.session_state.user_name}?{st.session_state.current_sessions[elem]['session_id']}",
69
  )
70
  refresh_sessions()
 
 
 
 
 
 
 
 
71
  except Exception as e:
72
  sleep(2)
73
  st.error(f"{type(e)}: {str(e)}")
 
90
  st.session_state[
91
  "current_sessions"
92
  ] = st.session_state.session_manager.list_sessions(st.session_state.user_name)
93
+ try:
94
+ dfl_indx = [
95
+ x["session_id"] for x in st.session_state.current_sessions
96
+ ].index("default")
97
+ except ValueError:
98
+ dfl_indx = 0
99
+ st.session_state.sel_sess = st.session_state.current_sessions[dfl_indx]
100
  if type(st.session_state.current_sessions) is not dict and len(st.session_state.current_sessions) <= 0:
101
  st.session_state.session_manager.add_session(
102
  st.session_state.user_name,
 
148
  dfl_indx = [
149
  x["session_id"] for x in st.session_state.current_sessions
150
  ].index("default")
151
+ except Exception as e:
152
+ print("*** ", str(e))
153
  dfl_indx = 0
154
  st.selectbox(
155
  "Choose a session be chat:",