Update src/streamlit_app/home_page.py

src/streamlit_app/home_page.py

@@ -1,59 +1,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("🌿🤸‍♂️ Physio-Flex: Your Path to a Healthier, Stronger You")
-    logger.log("info", "Setting the title to '🌿🤸‍♂️ Physio-Flex: Your Path to a Healthier, Stronger You'")
-
-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)}")
+import streamlit as st
+from streamlit_app import model
+from utils import logger
+
+
+def page_config():
+     st.set_page_config(page_title="Physio Flex", layout="wide")
+     logger.log("info","Configuring the page layout and title")
+
+def page_title():
+    st.title("🌿🤸‍♂️ Physio-Flex: Your Path to a Healthier, Stronger You")
+    logger.log("info", "Setting the title to '🌿🤸‍♂️ Physio-Flex: Your Path to a Healthier, Stronger You'")
+
+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)}")