Update app.py

app.py

@@ -1,38 +1,61 @@
 import gradio as gr
+import datetime
 
-# Custom supportive chatbot function
-def mental_health_bot(message):
-    message = message.lower().strip()
-
-    responses = {
-        "lonely": "You're not alone. I'm here to listen. Would you like to talk about what's making you feel this way?",
-        "heartbroken": "Heartbreak can feel really heavy. Want to share what happened? I'm here for you.",
-        "not able to study": "It’s okay to feel stuck. Maybe take a short break, or we can talk about what’s bothering you?",
-        "lackin in physics": "Struggling is part of learning. Maybe I can help you with a study tip or we can just talk about what’s making it tough.",
-        "anxious": "Anxiety can be overwhelming. Want to talk through it together?",
-        "depressed": "You matter. Even if it doesn’t feel like it right now, things can get better. I'm here for you.",
-        "suicidal": "I'm really sorry you're feeling this way. You're not alone. Please consider reaching out to a mental health professional or a helpline right away. Your life is important."
-    }
-
-    # Check if any keyword is matched
-    for keyword in responses:
-        if keyword in message:
-            return responses[keyword]
-
-    # Default fallback
-    return "I'm here for you. Want to talk more about how you're feeling?"
-
-# Gradio interface setup
-chat_interface = gr.Interface(
-    fn=mental_health_bot,
-    inputs=gr.Textbox(lines=2, placeholder="How are you feeling today?"),
-    outputs="text",
-    title="Teen Mental Health Chatbot 💬",
-    description="A kind and simple chatbot to support your feelings. You're not alone. Let's talk.",
-    theme="soft"
+# Keywords and professional responses
+topic_responses = {
+    "lonely": "It's completely natural to feel lonely at times. You're not alone — many people experience this. Would you like to talk about what's been going on recently?",
+    "heartbroken": "Heartbreak can be deeply painful. It's okay to grieve. Healing takes time, and I'm here to listen if you’d like to share more about it.",
+    "anxious": "Anxiety can feel overwhelming, but recognizing it is a strong first step. Would you like to explore grounding techniques or simply talk about what’s making you anxious?",
+    "depressed": "Feeling low can be incredibly hard. If you're feeling overwhelmed, speaking to a mental health professional can help. I’m also here to talk if you'd like.",
+    "not able to study": "Lack of focus can stem from stress, fatigue, or emotional strain. Do you want to break down your routine or talk about what's affecting your concentration?",
+    "lackin in physics": "Academic challenges are common, and it's okay to seek help. Would you like a study tip or a motivational push to get back on track?",
+    "suicidal": "I'm really sorry you're feeling this way. You're important and your life matters. Please seek immediate help from a professional or call a helpline in your area. You're not alone."
+}
+
+# Fallback generic emotional support
+fallback_response = (
+    "Thank you for sharing that with me. Emotions can be difficult to navigate, but I'm here to listen and support you."
+    " Would you like to talk more about what's been on your mind lately?"
 )
 
-# Launch the app
-if __name__ == "__main__":
-    chat_interface.launch()
+# Logging (for future upgrade with secure storage, optional)
+def log_interaction(user_input, response):
+    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    print(f"[{timestamp}] User: {user_input}\nResponse: {response}\n")
+
+# Chatbot response function
+def mental_health_bot(user_input):
+    user_input_lower = user_input.lower()
+    for keyword, response in topic_responses.items():
+        if keyword in user_input_lower:
+            log_interaction(user_input, response)
+            return response
+    log_interaction(user_input, fallback_response)
+    return fallback_response
+
+# Gradio interface
+def build_interface():
+    with gr.Blocks(theme=gr.themes.Base(), title="Teen Mental Health Chatbot") as demo:
+        gr.Markdown("""
+        # 💬 Teen Mental Health Chatbot
+        
+        This chatbot provides a safe, non-judgmental space for teenagers to talk about their feelings.
+        While it does not replace professional help, it can offer emotional support and understanding.
+        """)
+
+        chatbot = gr.Textbox(placeholder="How are you feeling today?", label="Your Message", lines=3)
+        output = gr.Textbox(label="Chatbot Response")
+
+        submit_btn = gr.Button("Send")
+        submit_btn.click(fn=mental_health_bot, inputs=chatbot, outputs=output)
+
+        gr.Markdown("""
+        > ⚠️ This is not a substitute for professional care. If you're in crisis, please contact a licensed therapist or a suicide prevention helpline in your region.
+        """)
+    return demo
+
+# Run the chatbot
+demo_app = build_interface()
+
+demo_app.launch()