api/local_api.py

from typing import List
import json

def get_question_by_id(l,question_id):
    doc=None
    for i in l:
        if i['original_question']['question_id']==question_id:
            doc=i
            break
    return doc
def get_question_ids_with_correctness(l) -> List[str]:

    print("started")
    results = []
    for doc in l:
        data = doc
        question_id = data['original_question']['question_id']
        correct_answer = data['original_question']['answer']
        generated_answer = data['generated_result']['answer_key_vale']
        correctness = "✅" if correct_answer == generated_answer else '📛'
        
        results.append(f"{question_id} {correctness}")

    return results

def init_json():
    with open("./all_data.json") as f:
        return json.load(f)
def update_depth(question_id,depth):
    with open('all_data_json' ,'w') as f:
        l=json.load(f)
        for i in l:
            if i['original_question']['question_id']==question_id:
                i["max_depth"]=depth
                break
        f.write(l)