testing / app.py
rumaisa1054's picture
Update app.py
57d0b95 verified
import streamlit as st
def main():
st.title("Doctor Chatbot")
# Initialize chat history if not already initialized
if "chat_messages" not in st.session_state:
st.session_state.chat_messages = []
# Display chat history
for message in st.session_state.chat_messages:
if message["role"] == "user":
st.text_area("User:", message["content"], height=40, key=message["content"], disabled=True)
else:
st.text_area("AI:", message["content"], height=40, key=message["content"], disabled=True)
# User input
if prompt := st.chat_input("Welcome - How can I help you?"):
# Display user's message in chat message container
with st.chat_message("user"):
st.markdown(prompt)
# Add user message to chat history
st.session_state.chat_messages.append({"role": "user", "content": prompt})
# Get AI response using responsr function
response = "responsr(prompt)"
# Display assistant's response in chat message container
with st.chat_message("assistant"):
st.markdown(response)
# Add assistant's response and audio to chat history
st.session_state.chat_messages.append({
"role": "assistant",
"content": response,
})
if __name__ == "__main__":
main()