Create app.py

app.py

@@ -0,0 +1,179 @@
+import streamlit as st
+import pandas as pd
+from groq import Groq
+import nltk
+from nltk.sentiment import SentimentIntensityAnalyzer
+
+# Download NLTK data
+nltk.download('vader_lexicon')
+
+# Initialize the Sentiment Analyzer
+sia = SentimentIntensityAnalyzer()
+
+# Initialize the Groq client with your API key
+client = Groq(api_key="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")
+
+# Custom CSS for full-page background image
+def add_fullpage_background_image():
+    st.markdown(
+        f"""
+        <style>
+        .stApp {{
+            background-image: url("https://raw.githubusercontent.com/ansabb420/ClariMind/main/img.jpg");
+            background-size: cover;
+            background-repeat: no-repeat;
+            background-position: center;
+            background-attachment: fixed;
+            height: 100vh;
+            overflow: hidden;
+        }}
+        </style>
+        """,
+        unsafe_allow_html=True
+    )
+
+# Add background image
+add_fullpage_background_image()
+
+# Define the system message for the model
+system_message = {
+    "role": "system",
+    "content": "You are a mental health assistant trained to identify symptoms of ADHD, PTSD, and schizophrenia based on user responses and provide appropriate recommendations, resources, and prescriptions based on the patient's condition. Start conversations with empathy and stay supportive, like an expert psychotherapist."
+}
+
+# Function to reset the chat
+def reset_chat():
+    st.session_state.messages = []
+    st.session_state.chat_title = "New Chat"
+
+# Initialize session state variables
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+    st.session_state.chat_title = "CLARIMIND"
+if 'questionnaire_open' not in st.session_state:
+    st.session_state.questionnaire_open = False
+if 'questionnaire_submitted' not in st.session_state:
+    st.session_state.questionnaire_submitted = False
+
+# Sidebar for user inputs
+with st.sidebar:
+    st.header("User Inputs")
+    name = st.text_input("Name")
+    age = st.number_input("Age", min_value=1, max_value=100, value=25)
+    location = st.text_input("Location")
+    gender = st.selectbox("Gender", options=["Male", "Female", "Other"])
+
+    if st.button("Reset Chat"):
+        reset_chat()
+
+# Button to open the questionnaire
+if st.button("Please fill the questionnaire"):
+    st.session_state.questionnaire_open = True
+
+# Display the questionnaire if the button was clicked
+if st.session_state.questionnaire_open and not st.session_state.questionnaire_submitted:
+    st.header("Mental Health Screening Questionnaire")
+    st.write("Please answer the following questions to help us understand your condition.")
+
+    # Questionnaire for mental health screening
+    attention_span = st.radio("Do you find it difficult to concentrate for extended periods?", ["Yes", "No"])
+    restlessness = st.radio("Do you often feel restless or unable to sit still?", ["Yes", "No"])
+    intrusive_memories = st.radio("Do you experience intrusive or distressing memories of traumatic events?", ["Yes", "No"])
+    flashbacks = st.radio("Do you frequently relive past traumatic events (e.g., through flashbacks)?", ["Yes", "No"])
+    paranoia = st.radio("Do you feel like others are watching or trying to harm you?", ["Yes", "No"])
+    auditory_hallucinations = st.radio("Do you hear voices that others cannot hear?", ["Yes", "No"])
+    difficulty_completing_tasks = st.radio("Do you often start tasks but fail to complete them?", ["Yes", "No"])
+    emotional_numbness = st.radio("Do you feel emotionally numb or detached from people?", ["Yes", "No"])
+    delusions = st.radio("Do you hold beliefs that others find unusual or illogical?", ["Yes", "No"])
+    sleep_disturbances = st.radio("Do you experience difficulty falling or staying asleep?", ["Yes", "No"])
+    hypervigilance = st.radio("Are you easily startled or always on guard?", ["Yes", "No"])
+    impulsivity = st.radio("Do you act on impulses without considering the consequences?", ["Yes", "No"])
+
+# After submitting the questionnaire
+if st.button("Submit Questionnaire"):
+    patient_persona = (
+        f"Patient Profile:\n"
+        f"- Name: {name}\n"
+        f"- Age: {age}\n"
+        f"- Location: {location}\n"
+        f"- Gender: {gender}\n"
+        f"- Symptoms:\n"
+        f"  * Attention Span Issues: {'Yes' if attention_span == 'Yes' else 'No'}\n"
+        f"  * Restlessness: {'Yes' if restlessness == 'Yes' else 'No'}\n"
+        f"  * Intrusive Memories: {'Yes' if intrusive_memories == 'Yes' else 'No'}\n"
+        f"  * Flashbacks: {'Yes' if flashbacks == 'Yes' else 'No'}\n"
+        f"  * Paranoia: {'Yes' if paranoia == 'Yes' else 'No'}\n"
+        f"  * Auditory Hallucinations: {'Yes' if auditory_hallucinations == 'Yes' else 'No'}\n"
+        f"  * Difficulty Completing Tasks: {'Yes' if difficulty_completing_tasks == 'Yes' else 'No'}\n"
+        f"  * Emotional Numbness: {'Yes' if emotional_numbness == 'Yes' else 'No'}\n"
+        f"  * Delusions: {'Yes' if delusions == 'Yes' else 'No'}\n"
+        f"  * Sleep Disturbances: {'Yes' if sleep_disturbances == 'Yes' else 'No'}\n"
+        f"  * Hypervigilance: {'Yes' if hypervigilance == 'Yes' else 'No'}\n"
+        f"  * Impulsivity: {'Yes' if impulsivity == 'Yes' else 'No'}\n"
+    )
+
+    st.session_state.patient_persona = patient_persona
+
+    questionnaire_data = {
+        "Name": name,
+        "Age": age,
+        "Location": location,
+        "Gender": gender,
+        "Attention Span": attention_span,
+        "Restlessness": restlessness,
+        "Intrusive Memories": intrusive_memories,
+        "Flashbacks": flashbacks,
+        "Paranoia": paranoia,
+        "Auditory Hallucinations": auditory_hallucinations,
+        "Difficulty Completing Tasks": difficulty_completing_tasks,
+        "Emotional Numbness": emotional_numbness,
+        "Delusions": delusions,
+        "Sleep Disturbances": sleep_disturbances,
+        "Hypervigilance": hypervigilance,
+        "Impulsivity": impulsivity
+    }
+
+    df = pd.DataFrame([questionnaire_data])  
+    df.to_csv("mental_health_responses.csv", mode='a', header=not pd.io.common.file_exists("mental_health_responses.csv"), index=False)
+
+    st.session_state.questionnaire_open = False
+    st.session_state.questionnaire_submitted = True
+    st.success("Thank you for completing the questionnaire! Your patient profile has been created.")
+
+# Chat functionality
+st.title(st.session_state.chat_title)
+
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.markdown(message["content"])
+
+user_input = st.chat_input("How can I assist you today?")
+if user_input:
+    sentiment_scores = sia.polarity_scores(user_input)
+    sentiment = "Neutral"
+    if sentiment_scores['compound'] > 0.05:
+        sentiment = "Positive"
+    elif sentiment_scores['compound'] < -0.05:
+        sentiment = "Negative"
+
+    st.session_state.messages.append({"role": "user", "content": f"{user_input} (Sentiment: {sentiment})"})
+
+    persona = st.session_state.patient_persona if 'patient_persona' in st.session_state else "Patient has not filled the questionnaire."
+    messages = [system_message, {"role": "user", "content": persona}] + st.session_state.messages
+
+    try:
+        completion = client.chat.completions.create(
+            model="llama3-8b-8192",
+            messages=messages,
+            temperature=1,
+            max_tokens=1024,
+            top_p=1,
+            stream=False,
+        )
+        response_content = completion.choices[0].message.content if completion.choices else "Sorry, I couldn't generate a response."
+    except Exception as e:
+        response_content = f"Error: {str(e)}"
+
+    st.session_state.messages.append({"role": "assistant", "content": response_content})
+    with st.chat_message("assistant"):
+        st.markdown(response_content)