dataset/dataset.py

from typing import Optional


def get_response_from_huggingface_dataset(message: str, DATASET) -> Optional[str]:
    for data in DATASET["train"]:
        if "dialog" in data and len(data["dialog"]) > 1:
            input_text: str = data["dialog"][0].lower()
            response_text: str = data["dialog"][1]

            if input_text == message.lower():
                return response_text
    return None