devve1 commited on
Commit
259d5fa
1 Parent(s): 2da2cef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -508,6 +508,7 @@ if __name__ == '__main__':
508
  disabled=False if st.session_state.id_chat != 'New Chat' else True
509
  ):
510
  st.session_state.id_chat = 'New Chat'
 
511
  with col2:
512
  if st.button(
513
  'Delete Chat',
@@ -523,6 +524,11 @@ if __name__ == '__main__':
523
  if conversation != conversations[st.session_state.id_chat]:
524
  st.session_state.id_chat = conversation
525
  break
 
 
 
 
 
526
 
527
  def generate_conv_title(llm: vllm.LLM, tokenizer: AutoTokenizer):
528
  if st.session_state.chat_id == 'New Chat':
 
508
  disabled=False if st.session_state.id_chat != 'New Chat' else True
509
  ):
510
  st.session_state.id_chat = 'New Chat'
511
+ st.session_state.messages = []
512
  with col2:
513
  if st.button(
514
  'Delete Chat',
 
524
  if conversation != conversations[st.session_state.id_chat]:
525
  st.session_state.id_chat = conversation
526
  break
527
+
528
+ conversations.pop(tmp)
529
+ with open(conversations_path, 'wb') as fp:
530
+ packed_bytes = msgpack.packb(conversations, use_bin_type=True)
531
+ fp.write(packed_bytes)
532
 
533
  def generate_conv_title(llm: vllm.LLM, tokenizer: AutoTokenizer):
534
  if st.session_state.chat_id == 'New Chat':