T / src /streamlit_app.py
Dhruv-Ty's picture
Update src/streamlit_app.py
41cd254 verified
raw
history blame contribute delete
834 Bytes
import os
import streamlit as st
from streamlit_chat_widget import chat_input_widget
def app():
st.title("Doctor AI Assistant")
if "chat_history" not in st.session_state:
st.session_state.chat_history = [
"Hello! I am your Doctor AI Assistant. How can I help you?"
]
# Display chat history
for message in st.session_state.chat_history:
st.write(message)
# Display the chat input widget at the bottom
user_input = chat_input_widget()
if user_input:
if "text" in user_input:
user_text =user_input["text"]
st.session_state.chat_history.append(f"You: {user_text}")
elif "audioFile" in user_input:
audio_bytes = bytes(user_input["audioFile"])
st.audio(audio_bytes)
if __name__ == "__main__":
app()