import numpy as np
import pandas as pd
from search_engine_model import SearchEngineModel

def main():
    search_engine_model = SearchEngineModel()
    model, preprocess = search_engine_model.load_clip_model()
    text_prompt = 'cat'
    text_embeddings = search_engine_model.encode_text(model, text_prompt)

    random_features = np.random.rand(50, 512)
    input_df = pd.DataFrame(random_features)
    names_column = [f'image_{it}.png' for it in range(0, len(random_features))]
    input_df.insert(0, 'images_names', names_column)

    search_result = search_engine_model.search_image_by_text_prompt(text_embeddings, input_df)

main()