helper.py

import os

def get_label_for_query(image_url, model_name):
      fourway_label = image_url.split('/')[-2]
  
  if fourway_label=='both_correct':
    return 'Correct'
  
  if fourway_label=='both_wrong':
    return 'Wrong'
  
  if fourway_label == 'chm_correct_knn_incorrect' and model_name == 'CHM':
    return 'Correct'
  elif fourway_label == 'knn_correct_chm_incorrect' and model_name == 'KNN':
   return 'Correct'
 
  return 'Wrong'

def get_category(image_url):
  return image_url.split('/')[-2]

def translate_winds_to_names(winds):
  return [folder_to_name[x] for x in winds]