utils.py

import json



class Test:
    def __init__(self):
        self.data = self.read_test_data()
        self.descriptions = self.read_image_descriptions()

    def __len__(self):
        return len(self.data)

    @staticmethod
    def read_test_data():
        with open("test.json") as f:
            test_data = json.load(f)
        return test_data

    @staticmethod
    def read_image_descriptions():
        with open("images/descriptions.txt") as f:
            lines = f.readlines()
            clean_lines = [line.strip() for line in lines]
        return clean_lines

    def get_question(self, idx):
        question = self.data[idx]["question"]
        return question
    
    def get_answers(self, idx):
        responces = self.data[idx]["responces"]
        answers = [responce["answer"] for responce in responces]
        return answers
    
    def get_description(self, idx):
        description = self.descriptions[idx]
        return description
    
    def get_image_path(self, idx):
        image_path = f"images/{idx}.jpg"
        return image_path