import pandas as pd

FILE_NAME = "gaia.jsonl"
#FILE_NAME = "gaia_20.jsonl"

def get_questions(level):
    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