fschwartzer commited on
Commit
03d5b19
1 Parent(s): e5b70df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -131,16 +131,15 @@ def search_with_fallback(query, df, limit=50):
131
 
132
  def integrated_app(query, titulo, EC, PU):
133
  df_mercadolibre = fetch_data_to_dataframe(query, 50, "mercadolibre")
134
- print(df_mercadolibre)
135
  df_combined = pd.concat([df_mercadolibre, data_crawler], ignore_index=True)
136
- print(df_combined)
137
 
138
  if df_combined.empty:
139
  return "Nenhum dado encontrado. Tente uma consulta diferente.", pd.DataFrame()
140
 
141
  # Pass whether "conjunto" is part of the original query
142
- include_word = ["conjunto"] if include_conjunto else []
143
- df_refined = refinar_resultados(df_combined, include_word=include_word)
144
 
145
 
146
  df_similares = search_with_fallback(query, df_refined)
 
131
 
132
  def integrated_app(query, titulo, EC, PU):
133
  df_mercadolibre = fetch_data_to_dataframe(query, 50, "mercadolibre")
134
+ include_conjunto = False # Default value
135
  df_combined = pd.concat([df_mercadolibre, data_crawler], ignore_index=True)
 
136
 
137
  if df_combined.empty:
138
  return "Nenhum dado encontrado. Tente uma consulta diferente.", pd.DataFrame()
139
 
140
  # Pass whether "conjunto" is part of the original query
141
+ include_conjunto = "conjunto" in query.lower()
142
+ df_refined = refinar_resultados(df_combined, include_word=include_conjunto)
143
 
144
 
145
  df_similares = search_with_fallback(query, df_refined)