AI FAQ Chatbot Project with Python

"""A retrieval-based FAQ chatbot that answers only from approved content."""

from __future__ import annotations

from dataclasses import dataclass
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.metrics.pairwise import cosine_similarity


FAQS = [
    ("What are the class timings?", "Weekday and weekend batch timings are listed on the current course page."),
    ("Do you provide certificates?", "Course-completion certificates are issued according to the published eligibility requirements."),
    ("Is online training available?", "Online and classroom options depend on the course and current batch schedule."),
    ("How can I contact support?", "Use the contact page to reach the training support team."),
    ("Are projects included?", "Project practice is included where it is listed in the selected course curriculum."),
    ("Where is the institute located?", "Check the contact page for the current address and directions."),
]


@dataclass(frozen=True)
class BotReply:
    answer: str
    matched_question: str | None
    confidence: float


class FaqBot:
    def __init__(self, faqs=FAQS, threshold: float = 0.18):
        self.questions = [item[0] for item in faqs]
        self.answers = [item[1] for item in faqs]
        self.threshold = threshold
        self.vectorizer = TfidfVectorizer(stop_words="english", ngram_range=(1, 2))
        self.matrix = self.vectorizer.fit_transform(self.questions)

    def answer(self, query: str) -> BotReply:
        if not query.strip():
            return BotReply("Please enter a question.", None, 0.0)
        scores = cosine_similarity(self.vectorizer.transform([query]), self.matrix)[0]
        index = int(scores.argmax())
        confidence = float(scores[index])
        if confidence < self.threshold:
            return BotReply("I could not find that in the approved FAQs. Please contact support.", None, confidence)
        return BotReply(self.answers[index], self.questions[index], confidence)


def main() -> None:
    bot = FaqBot()
    for query in ("Can I attend online?", "When are classes held?", "Can you repair my laptop?"):
        print(query, "->", bot.answer(query))


if __name__ == "__main__":
    main()