Update app.py

app.py

@@ -14,6 +14,36 @@ def reset_conversation():
     st.session_state.ask_intervention = False
     return None
 
+def analyze_diagnosis(messages, model_link):
+    # Extract user messages
+    user_messages = [message[1] for message in messages if message[0] == "user"]
+
+    # Define mental conditions and associated keywords
+    mental_conditions = {
+        "Depression": ["depression", "sad", "hopeless", "lonely", "empty", "worthless", "miserable"],
+        "Anxiety": ["anxiety", "nervous", "worried", "fearful", "panicked", "stressed", "tense"],
+        "Panic disorder": ["panic attack", "panic", "scared", "terrified", "frightened", "hyperventilate", "heart racing"],
+        "Bipolar disorder": ["bipolar", "manic", "mania", "euphoric", "energetic", "depressed", "hopeless"],
+        "Schizophrenia": ["schizophrenia", "hallucination", "delusion", "paranoia", "disorganized", "psychotic", "dissociation"],
+        "PTSD": ["ptsd", "trauma", "nightmare", "flashback", "startled", "avoidance", "hypervigilance"],
+        "Obsessive-Compulsive Disorder": ["ocd", "obsession", "compulsion", "intrusive thought", "ritual", "cleaning", "checking"],
+        "Eating disorder": ["eating disorder", "anorexia", "bulimia", "binge eating", "body image", "weight obsession", "purging"],
+        "Substance use disorder": ["substance use", "drug addiction", "alcoholism", "substance abuse", "withdrawal", "craving", "dependency"],
+        "Attention-deficit/hyperactivity disorder (ADHD)": ["adhd", "attention deficit", "hyperactivity", "impulsivity", "inattention", "restlessness", "fidgeting"],
+        "Borderline personality disorder": ["borderline personality", "emotional instability", "impulsivity", "fear of abandonment", "unstable relationships", "self-harm", "mood swings"],
+        "Postpartum depression": ["postpartum depression", "baby blues", "motherhood depression", "crying spells", "irritability", "hopelessness", "fatigue"],
+        "Social anxiety disorder": ["social anxiety", "fear of social situations", "embarrassment", "humiliation", "avoidance", "self-consciousness", "panic"]
+    }
+
+    # Count occurrences of keywords for each mental condition
+    condition_counts = {condition: sum(message.lower().count(keyword) for message in user_messages for keyword in keywords) 
+                        for condition, keywords in mental_conditions.items()}
+
+    # Determine the condition with the highest count
+    diagnosis = max(condition_counts, key=condition_counts.get)
+
+    return diagnosis
+
 # Define model links for the Addiction Recovery and Mental Health models
 model_links = {
     "Addiction recovery AI": "NousResearch/Nous-Hermes-2-Yi-34B",
@@ -85,6 +115,34 @@ if "messages" not in st.session_state:
 selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
 reset_button = st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
+# Add diagnostic feature to the sidebar
+if st.session_state.messages:
+    with st.sidebar:
+        st.subheader("Diagnosis")
+        st.markdown("<div style='color: #1E90FF; font-size: 16px; font-weight: bold;'>Analyzing...</div>", unsafe_allow_html=True)
+
+        diagnosis = analyze_diagnosis(st.session_state.messages, model_links[selected_model])
+        if diagnosis:
+            st.markdown(f"<div style='color: #FF6347; font-size: 18px; font-weight: bold;'>Diagnosis: {diagnosis}</div>", unsafe_allow_html=True)
+
+# Add additional features on the sidebar
+st.sidebar.markdown("---")
+st.sidebar.subheader("Additional Features")
+st.sidebar.markdown("📅 Schedule Appointment")
+st.sidebar.markdown("📝 Take Notes")
+st.sidebar.markdown("🎵 Relaxing Music")
+st.sidebar.markdown("🥗 Healthy Recipes")
+st.sidebar.markdown("💤 Sleep Tracker")
+
+# Add cautionary message about testing phase at the bottom of the sidebar
+st.sidebar.markdown("---")
+st.sidebar.markdown("**Note**: This model is still in the beta phase. Responses may be inaccurate or undesired. Use it cautiously, especially for critical issues.")
+
+# Add logo and text to the sidebar
+st.sidebar.markdown("---")
+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")
+
 # Accept user input
 if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     # Display user message in chat message container
@@ -111,4 +169,5 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
                         st.write("You can reach out to a certified therapist at +25493609747.")
 
         # Add assistant response to chat history
-        st.session_state.messages.append(("assistant", assistant_response))
+        st.session_state.messages.append(("assistant", assistant_response))
+