Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import requests
 import pandas as pd
 from difflib import get_close_matches
+import re  # Importa o módulo de expressões regulares
 
 def fetch_data_to_dataframe(query, limit=50, source="mercadolibre"):
     if source == "mercadolibre":
@@ -18,6 +19,14 @@ def fetch_data_to_dataframe(query, limit=50, source="mercadolibre"):
                 return df
     return pd.DataFrame()
 
+def refinar_resultados(df):
+    # Filtra itens que não contêm "kit" no título e não indicam múltiplas unidades
+    df_refinado = df[~df['Title'].str.contains("kit", case=False)]
+    # Expressão regular para identificar números seguidos por palavras relacionadas a quantidade
+    padrao_unidades = r'\b(\d+)\s*(unidade|unidades|pacote|pacotes|caixa|caixas)\b'
+    df_refinado = df_refinado[~df_refinado['Title'].str.contains(padrao_unidades, case=False, regex=True)]
+    return df_refinado
+
 def filtrar_itens_similares(df, termo_pesquisa, limite=5):
     titulos = df['Title'].tolist()
     titulos_similares = get_close_matches(termo_pesquisa, titulos, n=limite, cutoff=0.1)
@@ -29,6 +38,7 @@ def integrated_app(query):
     if df.empty:
         return "Nenhum dado encontrado. Tente uma consulta diferente.", pd.DataFrame()
 
+    df = refinar_resultados(df)  # Chama a função de refinamento
     df_similares = filtrar_itens_similares(df, query)
     if df_similares.empty:
         return "Nenhum item similar encontrado.", pd.DataFrame()