Create app.py

app.py

@@ -0,0 +1,60 @@
+import streamlit as st
+from transformers import pipeline
+
+# Configure the Streamlit page
+st.set_page_config(page_title="AI Friend Therapist Assistant", layout="centered")
+
+# Load the Hugging Face summarization model (cached to avoid reloading)
+@st.cache(allow_output_mutation=True)
+def load_summarizer():
+    return pipeline("summarization", model="facebook/bart-large-cnn")
+
+summarizer = load_summarizer()
+
+# Initialize session state to store conversation history
+if 'conversation' not in st.session_state:
+    st.session_state.conversation = []
+
+# App title and disclaimer
+st.title("AI Friend Therapist Assistant")
+st.write(
+    "I'm here to listen. Share your thoughts below.\n\n"
+    "*Disclaimer: I am not a licensed therapist. If you're in crisis, please seek professional help immediately.*"
+)
+
+# Text area for user input
+user_input = st.text_area("Your message", height=150)
+
+# When the user clicks "Send", process the input
+if st.button("Send"):
+    if user_input.strip():
+        # Append the message to conversation history
+        st.session_state.conversation.append(user_input)
+        
+        # Define keywords that might indicate suicidal ideation or distress
+        flagged_keywords = [
+            "suicide", "kill myself", "end my life", "self-harm", 
+            "self harm", "hopeless", "despair", "worthless", "can't go on"
+        ]
+        # Check if any flagged keyword is present in the input
+        if any(keyword in user_input.lower() for keyword in flagged_keywords):
+            st.error(
+                "Your message indicates you may be experiencing distress. "
+                "If you're in immediate danger, please call **911** immediately. "
+                "For non-immediate crisis support in Canada, call **Crisis Services Canada at 1.833.456.4566** "
+                "or visit [crisisservicescanada.ca](https://www.crisisservicescanada.ca/)."
+            )
+        else:
+            st.success("Message received. Thank you for sharing.")
+    else:
+        st.warning("Please enter a message.")
+
+# Option to generate a therapist summary of the conversation
+if st.button("Generate Therapist Summary"):
+    if st.session_state.conversation:
+        conversation_text = "\n".join(st.session_state.conversation)
+        summary = summarizer(conversation_text, max_length=100, min_length=30, do_sample=False)
+        st.subheader("Therapist Summary")
+        st.write(summary[0]['summary_text'])
+    else:
+        st.info("No conversation to summarize yet.")