from insw_scrapper import create_gpt_training_data
from gemini_trainer import generate_gemini_training_file
from gpt_trainer import train_gpt

if __name__ == '__main__':
    # extract_table_from_pdf(pdf_path="HSCodeMasterBPS.pdf", start_page=864, skip_header=True)
    training_data = create_gpt_training_data( # hscode_bps_pdf="HSCodeMasterBPS.pdf",
                             mensa_import_excel="tblPibDtl.xlsx",
                             mensa_custom_correction_excel="MensaResumeHS.xlsx")
    # train_gpt(data=training_data)
    generate_gemini_training_file(training_data)


# See PyCharm help at https://www.jetbrains.com/help/pycharm/
