import pandas as pd

LEVEL = 1

FILE_NAME = "gaia_20.jsonl"
#FILE_NAME = "gaia_validation.jsonl"

def get_questions():
    df = pd.read_json(FILE_NAME, lines = True)
    level_df = df[df["Level"] == LEVEL]

    result = []
    
    for index, row in level_df.iterrows():
        result.append([row["Level"], row["Question"], row["file_name"], row["Final answer"]])

    return result