Added image and description utils

utils.py

@@ -5,6 +5,7 @@ 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)
@@ -15,6 +16,13 @@ class Test:
             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
@@ -22,4 +30,12 @@ class Test:
     def get_answers(self, idx):
         responces = self.data[idx]["responces"]
         answers = [responce["answer"] for responce in responces]
-        return answers
+        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