From 934480585249ad800d07c617efd6aea52be4da61 Mon Sep 17 00:00:00 2001
From: sc0v1n0 <sc0v1n001@gmail.com>
Date: Sat, 26 Aug 2023 15:59:33 -0300
Subject: [PATCH] style: update names, remove comments

---
 src/components/query.py | 21 ++++++++-------------
 1 file changed, 8 insertions(+), 13 deletions(-)

diff --git a/src/components/query.py b/src/components/query.py
index 093a832..01d3e74 100644
--- a/src/components/query.py
+++ b/src/components/query.py
@@ -1,14 +1,9 @@
 def query_name_title(dataset, nome, top_n):
-    nome_pesquisa = nome.upper()
-    # Recebe o valor da id K-Classes
-    filme = dataset[dataset['title'] == nome_pesquisa][['clusters_genre_type']]
-    # resetar o index, como possui apenas um linha, acessar mais rápido
-    reseta_filme = filme.reset_index()
-    # Adicionar a célula da K-Classes, a uma variável
-    resetado_filme = reseta_filme.at[0, 'clusters_genre_type']
-    # Transforma variável string em inteiro
-    k_id = int(resetado_filme)
-    # Procura filmes com semelhança usando a k-id = K-Classes
-    recomendacoes = dataset[dataset['clusters_genre_type'] == k_id][['title', 'gender_type']][:int(top_n)]
-    recomendacoes.set_index('title')
-    return recomendacoes
\ No newline at end of file
+    new_search = nome.upper()
+    movie = dataset[dataset['title'] == new_search][['clusters_genre_type']]
+    reset_movie = movie.reset_index()
+    reset_movie = reset_movie.at[0, 'clusters_genre_type']
+    k_id = int(reset_movie)
+    result = dataset[dataset['clusters_genre_type'] == k_id][['title', 'gender_type']][:int(top_n)]
+    result.set_index('title')
+    return result
\ No newline at end of file