Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import streamlit as st
 import requests
 import os
@@ -13,36 +14,6 @@ def reset_conversation():
     st.session_state.messages = []
     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",
@@ -105,20 +76,9 @@ def interact_with_together_api(messages, model_link):
         st.error(f"Error communicating with the API: {e}")
         return None
 
-# Initialize chat history
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-
 # Create sidebar with model selection dropdown and reset button
 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 after at least 3 messages are input by the user
-if len(st.session_state.messages) >= 3:
-    diagnosis = analyze_diagnosis(st.session_state.messages, model_links[selected_model])
-    if diagnosis:
-        st.sidebar.subheader("Diagnosis")
-        st.sidebar.write(diagnosis)
+st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # Add cautionary message about testing phase at the bottom of the sidebar
 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.")
@@ -127,6 +87,17 @@ 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")
 
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+    st.session_state.message_count = 0
+    st.session_state.ask_intervention = False
+
+# Display chat messages from history on app rerun
+for message in st.session_state.messages:
+    with st.chat_message(message[0]):
+        st.markdown(message[1])
+
 # Accept user input
 if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     # Display user message in chat message container
@@ -134,6 +105,7 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
         st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append(("user", prompt))
+    st.session_state.message_count += 1
 
     # Interact with the selected model
     assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
@@ -144,13 +116,12 @@ if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
             st.markdown("AI is typing...")
             st.empty()
             st.markdown(assistant_response)
-
             # Check if intervention is needed based on bot response
             if any(keyword in prompt.lower() for keyword in ["human", "therapist", "someone", "died", "death", "help", "suicide", "suffering", "crisis", "emergency", "support", "depressed", "anxiety", "lonely", "desperate", "struggling", "counseling", "distressed", "hurt", "pain", "grief", "trauma", "abuse", "danger", "risk", "urgent", "need assistance"]):
                 # Intervention logic here
                 if not st.session_state.ask_intervention:
                     if st.button("After the analysing our session you may need some extra help, so you can reach out to a certified therapist at +25493609747 Name: Ogega feel free to talk"):
                         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))