Update app.py

app.py

@@ -1,57 +1,38 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
 import gradio as gr
 
-# Load DialoGPT-medium
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
-
-# Global state to keep track of chat
-chat_history_ids = None
-
-def respond(user_input, history):
-    global chat_history_ids
-
-    # Tokenize the input with EOS token
-    new_input_ids = tokenizer.encode(user_input + tokenizer.eos_token, return_tensors='pt')
-
-    # Append to history if available
-    bot_input_ids = torch.cat([chat_history_ids, new_input_ids], dim=-1) if chat_history_ids is not None else new_input_ids
-
-    # Generate response
-    chat_history_ids = model.generate(
-        bot_input_ids,
-        max_length=1000,
-        pad_token_id=tokenizer.eos_token_id,
-        temperature=0.7,
-        top_k=50,
-        top_p=0.95,
-        do_sample=True
-    )
-
-    # Decode only the new tokens
-    response = tokenizer.decode(chat_history_ids[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
-
-    # Truncate output to avoid HTTP overflow (safe zone ~500 chars)
-    response = response[:500]
-
-    # Append to history
-    history.append((user_input, response))
-    return history, history
-
-# Build Gradio Blocks UI
-with gr.Blocks(title="Teen Mental Health Chatbot") as demo:
-    gr.Markdown("### 💬 Teen Mental Health Chatbot\n_Not a replacement for therapy, but here to talk._")
-
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Tell me how you're feeling...")
-    state = gr.State([])
-
-    send = gr.Button("Send")
-
-    send.click(fn=respond, inputs=[msg, state], outputs=[chatbot, state])
-    msg.submit(fn=respond, inputs=[msg, state], outputs=[chatbot, state])
-
-# Required to run on Hugging Face
+# 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"
+)
+
+# Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    chat_interface.launch()
+