diff --git a/.gitignore b/.gitignore index fd3dbb5..9fd9904 100644 --- a/.gitignore +++ b/.gitignore @@ -34,3 +34,8 @@ yarn-error.log* # typescript *.tsbuildinfo next-env.d.ts +config.py +__pycache__/ +.idea/braintok-2.iml +*.pdf +.idea/ \ No newline at end of file diff --git a/main.py b/main.py index e2af2ef..0f82984 100644 --- a/main.py +++ b/main.py @@ -1,22 +1,34 @@ + from config import API_KEY from openai import OpenAI from fileFormatConverter import convert_file -client = OpenAI(api_key=API_KEY) def main(): - inputFileName = input("enter input file name with extension (supported types: pdf, docx): ") - outputFileName = input("enter output file name without extension:") - convert_file(inputFileName, outputFileName) + #inputFileName = input("enter input file name with extension (supported types: pdf, docx): ") + # outputFileName = input("enter output file name without extension:") - with open(f'{outputFileName}.jsonl', 'rb') as file: - file_response = client.files.create( - file=file, - purpose="fine-tune" - ) + inputFileName = "HW07.pdf" + outputFileName = "demo" + convert_file(inputFileName, outputFileName) + client = OpenAI(api_key=API_KEY) + + with open(f'{outputFileName}.jsonl', 'r') as file: + # file_response = client.files.create( + # file=file, + # purpose="fine-tune" + # ) + # fine_tuned_model = client.fine_tuning.jobs.create( + # training_file=file_response.id, + # model="gpt-4o-mini-2024-07-18" + # ) + homework = file.read() stream = client.chat.completions.create( - model="gpt-4o", - messages=[{"role": "user", "content": "Give me an estimation how long this homework will take me"}], + model="gpt-4o-mini-2024-07-18", + messages=[{"role": "user", "content": f""" + This is my homework: {homework} + If this homework consists of several separate problems, split them. + For each problem give me problem number, problem name, and problem text"""}], stream=True, ) for chunk in stream: