from sklearn.feature_extraction.text import TfidfVectorizer
import pandas as pd
from icecream import ic


def demo():
    vectorizer = TfidfVectorizer()

    lst = ["Hello I am the teacher",
           "I am the teacher Hello",
           "teacher Hello"]

    vectors = vectorizer.fit_transform(lst)

    print(vectors)
    print(vectorizer.get_feature_names_out())
    df = pd.DataFrame(vectors.toarray().transpose(),
                      index=vectorizer.get_feature_names_out())

    print(df)


if __name__ == "__main__":
    demo()