Update app.py
Browse files
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':
|