get_total_msg.py

def get_total_msg(user:str, df:pd.DataFrame) -> str:
  """ 
  Return the total msgs 'user' has sent 
  Args:
  ----
  user: the user to extract his/her total count of messages
  df: pd.DataFrame: dataframe that stores information about each user and message sent

  Returns:
  --------
  str
  """

  try:
    if user in df['Sender'].unique().tolist():
      text = f"{user} has sent a total message of {len(df[df['Sender'] == user])}"
    else: st.info(f"{user} is not present in the data")

  except Exception as e : st.warning('error', e)

  return text