fschwartzer commited on
Commit
59e4db6
1 Parent(s): ae59a9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -130,19 +130,17 @@ def search_with_fallback(query, df, limit=50):
130
  return pd.DataFrame()
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
  include_word = ["conjunto"] if include_conjunto else []
143
- df_refined = refinar_resultados(df_combined, include_word=include_conjunto)
144
 
145
-
146
  df_similares = search_with_fallback(query, df_refined)
147
 
148
  if df_similares.empty:
@@ -156,6 +154,7 @@ def integrated_app(query, titulo, EC, PU):
156
  valor_avaliacao = df_nearest['Price'].mean() * fator_avaliacao
157
  return f"Valor Médio do Bem: R$ {df_nearest['Price'].mean():.2f}, Fator de Avaliação: {fator_avaliacao*100:.2f}%, Valor de Avaliação: R$ {valor_avaliacao:.2f}", df_nearest
158
 
 
159
  iface = gr.Interface(fn=integrated_app,
160
  inputs=[gr.Textbox(label="Digite sua consulta"),
161
  gr.Dropdown(label="Classificação Contábil do Bem", choices=bens_df['TITULO'].unique().tolist(), value="MOBILIÁRIO EM GERAL"),
 
130
  return pd.DataFrame()
131
 
132
  def integrated_app(query, titulo, EC, PU):
133
+ include_conjunto = "conjunto" in query.lower()
134
  df_mercadolibre = fetch_data_to_dataframe(query, 50, "mercadolibre")
 
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_word = ["conjunto"] if include_conjunto else []
142
+ df_refined = refinar_resultados(df_combined, include_word=include_word)
143
 
 
144
  df_similares = search_with_fallback(query, df_refined)
145
 
146
  if df_similares.empty:
 
154
  valor_avaliacao = df_nearest['Price'].mean() * fator_avaliacao
155
  return f"Valor Médio do Bem: R$ {df_nearest['Price'].mean():.2f}, Fator de Avaliação: {fator_avaliacao*100:.2f}%, Valor de Avaliação: R$ {valor_avaliacao:.2f}", df_nearest
156
 
157
+
158
  iface = gr.Interface(fn=integrated_app,
159
  inputs=[gr.Textbox(label="Digite sua consulta"),
160
  gr.Dropdown(label="Classificação Contábil do Bem", choices=bens_df['TITULO'].unique().tolist(), value="MOBILIÁRIO EM GERAL"),