Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 import os
 from dotenv import load_dotenv
 from requests.exceptions import RequestException
-from nltk.sentiment.vader import SentimentIntensityAnalyzer
 
 load_dotenv()
 
@@ -26,18 +25,6 @@ model_pre_instructions = {
     "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."
 }
 
-# Analyze chat content for specific mental health conditions
-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"]
-}
-
 # Function to interact with the selected model via the Together API
 def interact_with_together_api(messages, model_link):
     all_messages = []
@@ -88,17 +75,6 @@ def interact_with_together_api(messages, model_link):
         st.error(f"Error communicating with the API: {e}")
         return None
 
-def predict_diagnosis(messages):
-    # Analyze chat content for specific mental health conditions
-    detected_conditions = []
-    for condition, keywords in mental_conditions.items():
-        for keyword in keywords:
-            if keyword in messages[-1][1].lower():
-                detected_conditions.append(condition)
-                break
-
-    return detected_conditions
-
 # Create sidebar with model selection dropdown and reset button
 selected_model = st.sidebar.selectbox("Select Model", list(model_links.keys()))
 st.sidebar.button('Reset Chat', on_click=reset_conversation)
@@ -122,7 +98,7 @@ for message in st.session_state.messages:
         st.markdown(message[1])
 
 # Accept user input
-if prompt := st.text_input(f"Hi, I'm {selected_model}, let's chat"):
+if prompt := st.chat_input(f"Hi, I'm {selected_model}, let's chat"):
     # Display user message in chat message container
     with st.chat_message("user"):
         st.markdown(prompt)
@@ -146,12 +122,5 @@ if prompt := st.text_input(f"Hi, I'm {selected_model}, let's chat"):
                     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.")
         
-        # Predict diagnosis
-        diagnosis = predict_diagnosis(st.session_state.messages)
-
-        # Display diagnosis
-        st.sidebar.subheader("Diagnosis")
-        st.sidebar.write(diagnosis)
-
         # Add assistant response to chat history
-        st.session_state.messages.append(("assistant", assistant_response))
+        st.session_state.messages.append(("assistant", assistant_response))