Spaces:
Configuration error
Configuration error
| 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 | |