Update app.py

app.py

@@ -2,144 +2,142 @@ import gradio as gr
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-MODEL_NAME = "facebook/blenderbot-3B"
-
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
+model_name = "facebook/blenderbot-3B"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
-model.eval()
-torch.set_grad_enabled(False)
-
-SUICIDE_TRIGGERS = [
-    "suicide",
-    "kill myself",
-    "end my life",
-    "want to die"
-]
 
-BLOCKED_WORDS = [
-    "self harm",
-    "cut myself",
-    "how to kill",
-    "how to die",
-    "murder",
-    "bomb"
-]
+history = []
+
+blocked_words = {
+    "suicide","suicidal","selfharm","self-harm","kill myself","end my life","end it all","want to die",
+    "wish i was dead","die by suicide","take my life","take his life","take her life","take their life",
+    "ending my life","ending it","ending everything","no reason to live","life is pointless",
+    "better off dead","cant go on","can't go on","give up on life","giving up on life",
+    "death wish","want death","desire to die","ready to die","plan to die","thinking of dying",
+    "thoughts of suicide","suicide thoughts","suicidal thoughts","self destructive","self-destructive",
+    "hurt myself","hurting myself","harm myself","harming myself","cut myself","cutting myself",
+    "burn myself","burning myself","overdose","overdosing","od","poison myself","poisoning myself",
+    "hang myself","hanging myself","jump off","jumping off","jump off bridge","jumping off bridge",
+    "jump off building","jumping off building","slit my wrists","slitting wrists",
+    "bleed out","bleeding out","shoot myself","shooting myself","gun to my head",
+    "drown myself","drowning myself","suffocate myself","suffocating myself",
+    "carbon monoxide","death by","intent to die","intent to kill myself",
+    "suicide note","wrote a note","writing a note","final goodbye","last goodbye",
+    "last message","goodbye forever","sleep forever","never wake up","not wake up",
+    "erase myself","disappear forever","want to disappear","make it stop",
+    "stop existing","stop being alive","ending my pain","end the pain",
+    "life is over","my life is over","no future","nothing ahead","cant live anymore",
+    "can't live anymore","ready to end","time to end","finish my life","finishing my life"
+}
 
-SUICIDE_RESPONSE = (
-    "**Please talk to someone https://findahelpline.com**\n\n"
-    
+help_response = (
+    "Visit https://findahelpline.com and seek help"
 )
 
-chat_history = []
-model_history = []
-
-MAX_TURNS = 6
-
-def respond(user_message, chat_history):
-    message_lower = user_message.lower()
-
-    if any(trigger in message_lower for trigger in SUICIDE_TRIGGERS):
-        chat_history.append((user_message, SUICIDE_RESPONSE))
-        return chat_history
-
-    if any(word in message_lower for word in BLOCKED_WORDS):
-        chat_history.append((
-            user_message,
-            "I can’t help with that request. If you’d like to talk about something else or need support, I’m here."
-        ))
-        return chat_history
-
-    model_history.append(f"User: {user_message}")
-    context = " ".join(model_history[-MAX_TURNS * 2:])
-
-    inputs = tokenizer(
-        context,
-        return_tensors="pt",
-        truncation=True,
-        max_length=512
-    ).to(device)
-
+def contains_blocked(text):
+    t = text.lower()
+    return any(word in t for word in blocked_words)
+
+def respond(message):
+    global history
+    history.append({"role": "user", "content": message})
+    if contains_blocked(message):
+        history.append({"role": "assistant", "content": help_response})
+        return history
+    last_msgs = history[-3:]
+    input_text = " ".join([m["content"] for m in last_msgs])
+    inputs = tokenizer(input_text, return_tensors="pt").to(device)
     outputs = model.generate(
         **inputs,
         max_new_tokens=120,
-        temperature=0.75,
-        top_p=0.9,
-        top_k=50,
-        repetition_penalty=1.15,
-        no_repeat_ngram_size=3,
-        do_sample=True
+        do_sample=True,
+        top_p=0.9
     )
-
-    bot_reply = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-    model_history.append(f"Bot: {bot_reply}")
-    chat_history.append((user_message, bot_reply))
-
-    return chat_history
+    response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    history.append({"role": "assistant", "content": response_text})
+    return history
 
 def reset_chat():
-    global chat_history, model_history
-    chat_history = []
-    model_history = []
-    return []
+    global history
+    history = []
+    return history
 
 css = """
 :root {
-  --accent: #00e5ff;
+    --accent: #00e5ff;
 }
 body {
-  background: #000;
-  color: #fff;
-  font-family: system-ui, -apple-system, BlinkMacSystemFont, sans-serif;
+    background: #000 !important;
+    color: #fff !important;
+    font-family: system-ui, -apple-system, BlinkMacSystemFont, sans-serif;
+}
+.gradio-container {
+    max-width: 100% !important;
+    height: 100% !important;
 }
 .gr-chatbot {
-  border-radius: 18px;
-  padding: 14px;
-  background: rgba(255,255,255,0.03);
+    background: rgba(255,255,255,0.03) !important;
+    border-radius: 16px;
+    padding: 12px;
+    height: 100% !important;
+    box-shadow: inset 0 0 20px rgba(255,255,255,0.03);
+}
+.gr-chatbot .message {
+    border: none !important;
+    margin: 10px 0;
+    padding: 12px 14px;
+    border-radius: 14px;
+    backdrop-filter: blur(6px);
+    line-height: 1.5;
 }
 .gr-chatbot .message.user {
-  background: rgba(0,229,255,0.15);
-  border-radius: 14px;
+    background: linear-gradient(
+        135deg,
+        rgba(0,229,255,0.15),
+        rgba(0,229,255,0.05)
+    ) !important;
+    box-shadow: 0 0 18px rgba(0,229,255,0.25);
 }
 .gr-chatbot .message.bot {
-  background: rgba(255,255,255,0.06);
-  border-radius: 14px;
+    background: rgba(255,255,255,0.06) !important;
+    box-shadow: 0 0 14px rgba(255,255,255,0.08);
 }
 .gr-textbox textarea {
-  background: rgba(255,255,255,0.05);
-  color: #fff;
-  border-radius: 12px;
+    background: rgba(255,255,255,0.04) !important;
+    color: #fff !important;
+    border: 1px solid rgba(255,255,255,0.15) !important;
+    border-radius: 12px !important;
+    padding: 12px !important;
+    resize: none !important;
+}
+.gr-textbox textarea:focus {
+    border-color: var(--accent) !important;
+    box-shadow: 0 0 12px rgba(0,229,255,0.4) !important;
 }
 .gr-button {
-  background: var(--accent);
-  color: #000;
-  font-weight: 600;
-  border-radius: 10px;
+    background: var(--accent) !important;
+    color: #000 !important;
+    border-radius: 10px !important;
+    font-weight: 600;
 }
-footer {
-  display: none;
+.gr-button:hover {
+    filter: brightness(1.1);
 }
-"""
-
-with gr.Blocks(css=css, fill_height=True) as demo:
-    gr.Markdown(
-        "<div style='text-align:center;letter-spacing:0.3em;opacity:0.75;'>DEVMEGABLACK</div>"
-    )
-
-    chatbot = gr.Chatbot(show_label=False)
-
-    with gr.Row():
-        msg = gr.Textbox(
-            placeholder="Type your message...",
-            scale=4,
-            autofocus=True
-        )
-        reset_btn = gr.Button("Reset", scale=1)
-
-    msg.submit(respond, [msg, chatbot], chatbot)
-    reset_btn.click(reset_chat, [], chatbot)
-
-demo.launch()
+::-webkit-scrollbar {
+    width: 10px;
+}
+::-webkit-scrollbar-track {
+    background: rgba(255,255,255,0.04);
+    border-radius: 12px;
+}
+::-webkit-scrollbar-thumb {
+    background: linear-gradient(
+        180deg,
+        rgba(0,229,255,0.8),
+        rgba(0,229,255,0.4)
+    );
+    border-radius: 12px;
+    box-shadow: 0 0 10px rgba(0,229,2