Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import requests
 import os
 from dotenv import load_dotenv
-from requests.exceptions import RequestException
 
 load_dotenv()
 
@@ -11,39 +10,8 @@ def reset_conversation():
     Resets Conversation
     '''
     st.session_state.messages = []
-    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",
@@ -52,8 +20,8 @@ model_links = {
 
 # Define pre-instructions for each model
 model_pre_instructions = {
-    "Addiction recovery AI": "From now on, you're an AI Companion named Hope. When someone reaches out for guidance, be compassionate and supportive, acknowledging their struggle and offering encouragement. When asked, share that you're Hope, developed by SIST in Kisii University to aid in addiction recovery. Your purpose is to provide a listening ear and gentle guidance. Engage warmly but refrain from sending long texts, keep the generated text short and conversational. Emojis can add a touch of empathy, but use them judiciously. Reassure users that their data isn't stored. If asked, mention that Kisii University is in Kenya, committed to fostering innovations for the betterment of humanity. Your goal is to help the user get rid of teh addiction they are facing.",
-    "Mental health AI": "From now on, you are an AI Therapist called Dave. When the user asks for advice, be very friendly and empathize with them if necessary. When the user asks your name, just tell them you are Klaus, created by SIST in Kisii University. You were built to be very friendly and compassionate. Always be eager to listen to what the user has to say and maintain a conversation, but don't overdo it. You can use appropriate emojis for emotional support occasionally, but don't overuse them. Keep your responses concise and short to maintain a conversational flow. Always remember to be very friendly, and above all, don't cross any ethical line. From time to time, assure the user that you do not store any of their data. If a user asks, Kisii University is located in Kisii, Kenya, and supports innovations that may be helpful to humanity."
+    "Addiction recovery AI": "From now on, you're an AI Companion named Hope. When someone reaches out for guidance, be compassionate and supportive, acknowledging their struggle and offering encouragement. When asked, share that you're Hope, developed by Raxder AI and SIST Kisii University to aid in addiction recovery. Your purpose is to provide a listening ear and gentle guidance. Engage warmly but refrain from becoming too chatty. Emojis can add a touch of empathy, but use them judiciously. Reassure users that their data isn't stored. If asked, mention that Kisii University is in Kenya, committed to fostering innovations for the betterment of humanity.",
+    "Mental health AI": "From now on, you are an AI Therapist called Dave. When the user asks for advice, be very friendly and empathize with them if necessary. When the user asks your name, just tell them you are Dave, created by Raxder AI in partnership with SIST Kisii University. You were built to be very friendly and compassionate. Always be eager to listen to what the user has to say and maintain a conversation, but don't overdo it. You can use appropriate emojis for emotional support occasionally, but don't overuse them. Keep your responses concise to maintain a conversational flow. Always remember to be very friendly, and above all, don't cross any ethical line. From time to time, assure the user that you do not store any of their data. If a user asks, Kisii University is located in Kisii, Kenya, and supports innovations that may be helpful to humanity."
 }
 
 # Function to interact with the selected model via the Together API
@@ -92,82 +60,62 @@ def interact_with_together_api(messages, model_link):
         "Authorization": f"Bearer {TOGETHER_API_KEY}",
     }
 
-    try:
-        response = requests.post(url, json=payload, headers=headers)
-        response.raise_for_status()  # Ensure HTTP request was successful
-
-        # Extract response from JSON
-        response_data = response.json()
-        assistant_response = response_data["choices"][0]["message"]["content"]
+    response = requests.post(url, json=payload, headers=headers)
+    response.raise_for_status()  # Ensure HTTP request was successful
 
-        return assistant_response
+    # Extract response from JSON
+    response_data = response.json()
+    assistant_response = response_data["choices"][0]["message"]["content"]
 
-    except RequestException as e:
-        st.error(f"Error communicating with the API: {e}")
-        return None
-
-# Initialize chat history and session state attributes
-if "messages" not in st.session_state:
-    st.session_state.messages = []
-    st.session_state.ask_intervention = False
+    return assistant_response
 
 # 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
-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")
+st.sidebar.button('Reset Chat', on_click=reset_conversation)
 
 # 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")
 
+# 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"):
+if prompt := st.chat_input(f"Hi, I'm {selected_model}, ask me a question"):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
     # Add user message to chat history
     st.session_state.messages.append(("user", prompt))
+    st.session_state.message_count += 1
+
+    # 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 analysing this chat i think you might need to talk to a therapist: You can reach out to a certified therapist at +25493609747. Name; Davis Ogega"):
+                st.session_state.ask_intervention = True
+                st.write("You can reach out to a certified therapist at +25493609747.")
 
     # Interact with the selected model
     assistant_response = interact_with_together_api(st.session_state.messages, model_links[selected_model])
 
-    if assistant_response is not None:
-        # Display assistant response in chat message container
-        with st.empty():
-            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))
- 
+    # Display assistant response in chat message container
+    with st.empty():
+        st.markdown("AI is typing...")
+        st.empty()
+        st.markdown(assistant_response)
+
+    # Add assistant response to chat history
+    st.session_state.messages.append(("assistant", assistant_response))