Build a Chatbot

1
2
3
4
5
6
7

from transformers import pipeline
import random

def chatbot(message):
    classifier = pipeline("zero-shot-classification")
    result = classifier(message, ["greeting", "question", "goodbye"])
    return responses[result["labels"][0]]

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

import random

responses = {
    "greeting": ["Hello!", "Hi there!", "Hey!"],
    "goodbye": ["Bye!", "See you!", "Goodbye!"],
    "thanks": ["You're welcome!", "Happy to help!"],
    "default": ["Tell me more!", "Interesting!", "I see."]
}

def get_intent(text):
    text = text.lower()
    if any(w in text for w in ["hi", "hello", "hey"]):
        return "greeting"
    if any(w in text for w in ["bye", "goodbye"]):
        return "goodbye"
    if any(w in text for w in ["thanks", "thank"]):
        return "thanks"
    return "default"

def chatbot(text):
    intent = get_intent(text)
    return random.choice(responses[intent])

print(chatbot("Hello!"))
print(chatbot("Thanks for help"))
print(chatbot("Goodbye"))