import numpy as np
import faiss
from utils import get_model, get_img_files, batch_process_images
from django.conf import settings
import tensorflow as tf
import os
import glob

model = get_model()

img_files=glob.glob('./static/data/data/*.jpg')
img_files=sorted(img_files)

features = batch_process_images(img_files, model)

dimension = features.shape[1]  
index = faiss.IndexFlatIP(dimension) 

faiss.normalize_L2(features)

index.add(features) 

faiss.write_index(index, 'faiss_index_database')

os.rename('faiss_index_database', 'static/model/faiss_index_database')