def get_word_counts(df):
    """Gets the total word count and the total unique lemmas.

    Parameters
    ----------
    df : pandas.DataFrame
        The DataFrame to be analyzed.

    """

    # Small fix for programa and programar.
    df.loc[df['lemma_lower'] == "programa", "lemma_lower"] = "programar"

    words = df[df["is_alphabet"] == True]["text_lower"].count()
    print("Words:", words)

    unique_words = df[df["is_alphabet"] == True]["lemma_lower"].nunique()
    print("Unique words:", unique_words)
Gets the total word count and the total unique lemmas.

    Parameters
    ----------
    df : pandas.DataFrame
        The DataFrame to be analyzed.

    
scripts/step3.py