main.py

"""
Data Scientist.: Dr.Eddy Giusepe Chirinos Isidro

main.py
=======

Objetivo: Neste repositório seguimos os passos de 🤗"John Mendoza"🤗
          para criar um Sistema de Recomendação. Este sistema serve, por exemplo
          para uma loja de e-commerce. Aqui usamos a API da OpenAI para obter nossos
          EMBEDDINGS e usamos o DB Redis para armazenar nossos dados.

Execução: primeiro --> Instancie seu DB redis (com docker) https://redis.io/docs/getting-started/install-stack/docker/
          segundo -->  $ python main.py

Versão: 1.0.0     
Data: 07/07/2023             
"""
from modules.Recommendation import CoreRecommendation


def get_recommend_items(text, top_k=2):
    api_url = 'https://fakestoreapi.com/products'
    c_search = CoreRecommendation(api_url)
    items = c_search.recommended_products(text, top_k)
    return items


text_search = 'I need Men casual t-shits'
items = get_recommend_items(text_search, top_k=2)
print(items)