Update app.py

app.py

@@ -43,10 +43,13 @@ def interact_with_together_api(messages, model_link, diagnostic=False):
 
     if diagnostic:
         diagnostic_prompt = "Analyze the following conversation and predict the mental issue the user might be suffering from:"
-        all_messages.append({"role": "user", "content": diagnostic_prompt})
+        diagnostic_messages = [{"role": "system", "content": "You are an AI model specialized in mental health diagnosis."},
+                               {"role": "user", "content": diagnostic_prompt}]
         for message in messages:
             if message[0] == "user":
-                all_messages.append({"role": "user", "content": message[1]})
+                diagnostic_messages.append({"role": "user", "content": message[1]})
+
+        all_messages = diagnostic_messages
 
     url = "https://api.together.xyz/v1/chat/completions"
     payload = {
@@ -124,12 +127,6 @@ st.sidebar.markdown("**Note**: This model is still in the beta phase. Responses
 st.sidebar.image("https://assets.isu.pub/document-structure/221118065013-a6029cf3d563afaf9b946bb9497d45d4/v1/2841525b232adaef7bd0efe1da81a4c5.jpeg", width=200)
 st.sidebar.write("A product proudly developed by Kisii University")
 
-# Display diagnosis in the sidebar
-if st.session_state.message_count >= 4:
-    if "diagnosis" not in st.session_state or st.session_state.diagnosis is None:
-        st.session_state.diagnosis = diagnose_mental_health(st.session_state.messages)
-    st.sidebar.markdown(f"### Diagnosis:\n**{st.session_state.diagnosis}**")
-
 # Initialize chat history
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -137,6 +134,12 @@ if "messages" not in st.session_state:
     st.session_state.ask_intervention = False
     st.session_state.diagnosis = None
 
+# Display diagnosis in the sidebar
+if st.session_state.message_count >= 4:
+    if "diagnosis" not in st.session_state or st.session_state.diagnosis is None:
+        st.session_state.diagnosis = diagnose_mental_health(st.session_state.messages)
+    st.sidebar.markdown(f"### Diagnosis:\n**{st.session_state.diagnosis}**")
+
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
     with st.chat_message(message[0]):