AI Email Reply Assistant Project

"""Draft reviewable email replies from approved templates."""

from __future__ import annotations

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


TEMPLATES = [
    ("request course information syllabus topics", "course_information", "Thanks for your interest. Please review the current course page for the syllabus, duration, and prerequisites."),
    ("ask class schedule batch timing", "schedule", "Thanks for checking. Current batch dates and timings are available from the training support team."),
    ("payment invoice receipt refund", "billing", "Thanks for letting us know. Please share the transaction reference through the official support channel so the billing team can review it."),
    ("technical error login upload access", "technical_support", "Sorry about the difficulty. Please share the exact error, device, browser, and steps to reproduce through the support channel."),
]


@dataclass(frozen=True)
class Draft:
    category: str | None
    text: str
    confidence: float


def draft_reply(message: str, threshold: float = 0.12) -> Draft:
    prompts = [item[0] for item in TEMPLATES]
    vectorizer = TfidfVectorizer(ngram_range=(1, 2), stop_words="english")
    matrix = vectorizer.fit_transform(prompts + [message])
    scores = cosine_similarity(matrix[-1], matrix[:-1])[0]
    index = int(scores.argmax())
    confidence = float(scores[index])
    if confidence < threshold:
        return Draft(None, "Thank you for your message. A team member will review it and respond through the official support channel.", confidence)
    return Draft(TEMPLATES[index][1], TEMPLATES[index][2], confidence)


def main() -> None:
    print(draft_reply("Please send the course syllabus and topics"))
    print("Draft only: verify facts and recipients before sending any email.")


if __name__ == "__main__":
    main()