Update app.py

app.py

@@ -130,20 +130,21 @@ if st.button("Summarize Chat"):
     st.subheader("Chat Summary")
     st.write(summary)
 if st.button("Detect Disorders"):
-    if chat_history.strip():
-        disorders = detect_disorders(chat_history.split("\n"))
-        st.write("**Detected Disorders:**")
+    if st.session_state["chat_history"]:
+        disorders = detect_disorders(st.session_state["chat_history"])
+        st.subheader("Detected Disorders:")
         for disorder in disorders:
             st.write(f"- {disorder}")
     else:
         st.error("❌ Please enter chat history.")
 
 if st.button("Get Treatment Recommendations"):
-    if chat_history.strip():
-        detected_disorders = detect_disorders(chat_history.split("\n"))
+    if st.session_state["chat_history"]:
+        detected_disorders = detect_disorders(st.session_state["chat_history"])
         treatments = get_treatment(detected_disorders)
-        st.write("**Treatment Recommendations:**")
+        st.subheader("Treatment Recommendations:")
         for disorder, treatment in treatments.items():
             st.write(f"**{disorder}:** {treatment}")
     else:
         st.error("❌ Please enter chat history.")
+