Spaces:
Configuration error
Configuration error
File size: 3,065 Bytes
a59ab81 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import streamlit as st
from streamlit_pills import pills
from st_utils import (
add_builder_config,
add_sidebar,
get_current_state,
)
current_state = get_current_state()
####################
#### STREAMLIT #####
####################
st.set_page_config(
page_title="Welcome to ChatDoctor: your virtual primary care physician assistant",
page_icon="💬",
layout="centered",
initial_sidebar_state="auto",
menu_items=None,
)
st.title("Welcome to ChatDoctor: your virtual primary care physician assistant")
st.info(
"Hello, my name is ChatDoctor and I am trained to provide medical diagnoses and advice.",
icon="ℹ️",
)
if "metaphor_key" in st.secrets:
st.info("**NOTE**: The ability to add web search is enabled.")
add_builder_config()
add_sidebar()
# add pills
selected = pills(
"Sample prompts",
[
"I want a medical diagnoses for ...",
"I want medical advice about ...",
],
clearable=True,
index=None,
)
if "messages" not in st.session_state.keys(): # Initialize the chat messages history
st.session_state.messages = [
{"role": "assistant", "content": "How may I help you, today?"}
]
def add_to_message_history(role: str, content: str) -> None:
message = {"role": role, "content": str(content)}
st.session_state.messages.append(message) # Add response to message history
for message in st.session_state.messages: # Display the prior chat messages
with st.chat_message(message["role"]):
st.write(message["content"])
# TODO: this is really hacky, only because st.rerun is jank
if prompt := st.chat_input(
"Your question",
): # Prompt for user input and save to chat history
# TODO: hacky
if "has_rerun" in st.session_state.keys() and st.session_state.has_rerun:
# if this is true, skip the user input
st.session_state.has_rerun = False
else:
add_to_message_history("user", prompt)
with st.chat_message("user"):
st.write(prompt)
# If last message is not from assistant, generate a new response
if st.session_state.messages[-1]["role"] != "assistant":
with st.chat_message("assistant"):
with st.spinner("Thinking..."):
response = current_state.builder_agent.chat(prompt)
st.write(str(response))
add_to_message_history("assistant", str(response))
else:
pass
# check agent_ids again
# if it doesn't match, add to directory and refresh
agent_ids = current_state.agent_registry.get_agent_ids()
# check diff between agent_ids and cur agent ids
diff_ids = list(set(agent_ids) - set(st.session_state.cur_agent_ids))
if len(diff_ids) > 0:
# # clear streamlit cache, to allow you to generate a new agent
# st.cache_resource.clear()
st.session_state.has_rerun = True
st.rerun()
else:
# TODO: set has_rerun to False
st.session_state.has_rerun = False
|