Skip to content

Commit

Permalink
add more guard rails to remove double chat
Browse files Browse the repository at this point in the history
  • Loading branch information
dillonalaird committed Oct 2, 2024
1 parent 3bcdc9c commit 097360d
Showing 1 changed file with 17 additions and 11 deletions.
28 changes: 17 additions & 11 deletions examples/chat/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
"style": {"bottom": "calc(50% - 4.25rem", "right": "0.4rem"},
}
# set artifacts remote_path to WORKSPACE
artifacts = va.tools.Artifacts(WORKSPACE / "artifacts.pkl")
artifacts = va.tools.meta_tools.Artifacts(WORKSPACE / "artifacts.pkl")
if Path("artifacts.pkl").exists():
artifacts.load("artifacts.pkl")
else:
Expand Down Expand Up @@ -109,16 +109,22 @@ def main():
len(st.session_state.messages) == 0
or prompt != st.session_state.messages[-1]["content"]
):
st.session_state.messages.append(
{"role": "user", "content": prompt}
)
messages.chat_message("user").write(prompt)
message_thread = threading.Thread(
target=update_messages,
args=(st.session_state.messages, message_lock),
)
message_thread.daemon = True
message_thread.start()
# occassionally resends the last user message twice
user_messages = [msg for msg in st.session_state.messages if msg["role"] == "user"]
last_user_message = None
if len(user_messages) > 0:
last_user_message = user_messages[-1]["content"]
if last_user_message is None or last_user_message != prompt:
st.session_state.messages.append(
{"role": "user", "content": prompt}
)
messages.chat_message("user").write(prompt)
message_thread = threading.Thread(
target=update_messages,
args=(st.session_state.messages, message_lock),
)
message_thread.daemon = True
message_thread.start()
st.session_state.input_text = ""

with tabs[1]:
Expand Down

0 comments on commit 097360d

Please sign in to comment.