Update app.py

app.py

@@ -7,6 +7,7 @@ from streamlit_feedback import streamlit_feedback
 
 from rag_chain.chain import get_rag_chain
 
+# Langsmith client for the feedback system
 client = Client()
 
 # Streamlit page configuration
@@ -113,19 +114,19 @@ if prompt:
 
         try:
             partial_message = ""
-            with st.spinner(" "):
-                with collect_runs() as cb:  # Collect runs for feedback in langsmith
-                    for chunk in chain.stream({"message": prompt}):
-                        partial_message += chunk
-                        message_placeholder.markdown(partial_message + "|")
-                    st.session_state.run_id = cb.traced_runs[0].id
+            # Collect runs for feedback using Langsmith
+            with st.spinner(" "), collect_runs() as cb:
+                for chunk in chain.stream({"message": prompt}):
+                    partial_message += chunk
+                    message_placeholder.markdown(partial_message + "|")
+                st.session_state.run_id = cb.traced_runs[0].id
+            message_placeholder.markdown(partial_message)
         except openai.BadRequestError:
             st.warning(openai_api_error_message, icon="🙁")
             st.stop()
         except Exception as e:
             st.warning(base_error_message, icon="🙁")
             st.stop()
-        message_placeholder.markdown(partial_message)
 
         # Add the full response to the history
         st.session_state["history"].append((prompt, partial_message))
@@ -133,7 +134,7 @@ if prompt:
         # Add AI message to memory after the response is generated
         memory.chat_memory.add_ai_message(AIMessage(content=partial_message))
 
-        # add the full response to the message history
+        # Add the full response to the message history
         st.session_state["messages"].append(ChatMessage(
             role="assistant", content=partial_message))
 
@@ -158,9 +159,9 @@ st.sidebar.markdown(
     Thank you! Let's get started. 🚀
 
     **Note**:\n\n
-    
-    This AI assistant is designed to provide guidance and general information about the services offered by Tall Tree Health.
-    It is not intended for seeking medical advice and should not be used as such.
+
+    This AI assistant is designed to provide guidance and general information about the services offered by Tall Tree Health. 
+    It is not intended for seeking medical advice and should not be used as such. 
     The information provided by this generative AI technology cannot replace the advice of qualified healthcare professionals.
     """
 )
@@ -172,6 +173,7 @@ if st.session_state.get("run_id"):
     run_id = st.session_state.run_id
     feedback = streamlit_feedback(
         feedback_type=feedback_option,
+        optional_text_label="[Optional] Please provide an explanation",
         key=f"feedback_{run_id}",
     )
     score_mappings = {
@@ -196,6 +198,7 @@ if st.session_state.get("run_id"):
                 run_id,
                 feedback_type_str,
                 score=score,
+                comment=feedback.get("text"),
             )
             st.session_state.feedback = {
                 "feedback_id": str(feedback_record.id),