Spaces:
Sleeping
Sleeping
File size: 2,702 Bytes
d0c513b 847aabf d0c513b |
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 |
import streamlit as st
from streamlit_app import model
from utils import logger
def page_config():
st.set_page_config(page_title="Physio Assist", layout="wide")
logger.log("info","Configuring the page layout and title")
def page_title():
st.title("🏋️♂️🧑⚕️ Welcome to Physio Assist: Your Physical Therapy Companion")
logger.log("info", "Setting the title to 'Physio Assist: Your Physical Therapy Companion'")
def get_or_greet_user_name():
if 'user_name' not in st.session_state:
st.session_state.user_name = None
logger.log("info","user_name not found in session_state, setting to None.")
if st.session_state.user_name is None:
logger.log("info","user_name is None, requesting user input.")
user_name = st.text_input("Please let me know your name:",
placeholder="Enter your name buddy")
if user_name:
st.session_state.user_name = user_name
logger.log("info", f"User entered name: {user_name}. Setting session_state.user_name.")
st.rerun()
else:
logger.log("info", f"User already entered a name: {st.session_state.user_name}. Displaying greeting.")
return st._bottom.subheader(f"Hello {st.session_state.user_name}! How can I assist you today?")
def display_chat():
logger.log("info","Displaying the chat history.")
if "messages" not in st.session_state:
st.session_state.messages = []
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
logger.log("info", f"Displayed {len(st.session_state.messages)} messages from the chat history.")
def handle_user_input():
logger.log("info", "Waiting for user input...")
prompt = st.chat_input("Ask me anything related to physiotherapy. E.g., 'How can I recover from a sprained ankle?'")
if prompt :
with st.chat_message("user"):
st.markdown(prompt)
if prompt:
st.session_state.messages.append({"role": "user", "content": prompt})
with st.spinner("Processing your query..."):
try:
response = model.get_physiotherapy_assistant_response(prompt)
with st.chat_message("assistant"):
st.markdown(response)
st.session_state.messages.append({"role": "assistant", "content": response})
logger.log("info",f"Assistant response: {response}")
except Exception as e:
st.error(f"An error occurred while processing your query: {str(e)}")
logger.log("error", f"Error processing user query: {str(e)}") |