from django.core.management import BaseCommand
from vacancies.main.vector_store import vector_store
from vacancies.main.models import CustomerCV, RecommendedVacancy


class Command(BaseCommand):
    help = "Generates new recommended vacancies"

    def handle(self, *args, **options):
        customer_cvs = CustomerCV.objects.all()
        for customer_cv in customer_cvs:
            documents = vector_store.similarity_search(customer_cv.content, k=10)
            for document in documents:
                RecommendedVacancy.objects.get_or_create(
                    customer=customer_cv.customer,
                    vacancy_id=document.metadata["vacancy_id"],
                    defaults=dict(is_shown=False),
                )