sheets.py

from datetime import datetime
from datasets import load_dataset
import gspread
from huggingface_hub import login
from settings import SHEETS_ID, HF_TOKEN
login(token=HF_TOKEN)
creds = load_dataset("rkarthur/test")
final_credentials = dict(zip(creds['train'].data.to_pydict().keys(), [x[0] for x in creds['train'].data.to_pydict().values()]))
gc = gspread.service_account_from_dict(final_credentials)
worksheet = gc.open_by_url(SHEETS_ID).get_worksheet(1)


def next_available_row(worksheet):
    str_list = list(filter(None, worksheet.col_values(1)))
    return len(str_list)+1


def update_sheets(name: str, email: str, rf_score: float, better_score: float,
                  essay: str) -> str:
    row_number = next_available_row(worksheet)
    submission_date = datetime.now().date().strftime('%m-%d-%y')
    worksheet.update(f'A{row_number}:F{row_number}',
                     [[name, email, submission_date,
                       rf_score, better_score, essay]])
    return "Submission successful."