import configparser
import search_engine_model as sem
        
def main():
    config_manager_obj = configparser.ConfigParser()
    config_manager_obj.read('./config.cfg')
    
    image_root_dir = config_manager_obj['SEARCH_ENGINE']['image_root_dir']
    csv_file_path = config_manager_obj['SEARCH_ENGINE']['csv_file_path']

    model_manager_obj = sem.SearchEngineModel(image_root_dir, csv_file_path)
    search_results = model_manager_obj.search_image_by_text_prompt('Cat', 10)
    print('Search results found by text prompt -> ', search_results)
    
    search_results = model_manager_obj.search_image_by_image_prompt('./dataset/000000001675.jpg', 10)
    print('Search results found by image prompt -> ', search_results)

main()