rwayz commited on
Commit
99d8de2
·
1 Parent(s): 6aa407c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -21
app.py CHANGED
@@ -103,12 +103,6 @@ def handle_csv_upload(file):
103
  logging.error(f"[ERRO] Falha ao processar novo CSV: {e}")
104
  return f"❌ Erro ao processar CSV: {e}"
105
 
106
- # === Inicialização ===
107
- engine = create_engine_and_load_db(get_active_csv_path(), SQL_DB_PATH)
108
- db = SQLDatabase(engine=engine)
109
- llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
110
- sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True, max_iterations=40, return_intermediate_steps=True)
111
-
112
  def reset_app():
113
  global engine, db, sql_agent, query_cache, history_log, recent_history
114
  try:
@@ -127,15 +121,10 @@ def reset_app():
127
  except Exception as e:
128
  return f"❌ Erro ao resetar: {e}"
129
 
130
- def export_history_json():
131
- path = "history_log.json"
132
- pd.DataFrame(history_log).to_json(path, orient="records", indent=2)
133
- return path
134
-
135
- def export_history_csv():
136
- path = "history_log.csv"
137
- pd.DataFrame(history_log).to_csv(path, index=False, sep=";")
138
- return path
139
 
140
  def generate_initial_context(db_sample):
141
  return (
@@ -245,12 +234,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
245
  with gr.Row():
246
  with gr.Column(scale=1):
247
  gr.Markdown("## ⚙️ Configurações")
248
- model_selector = gr.Dropdown(list(LLAMA_MODELS.keys()), label="Modelo LLM", value="LLaMA 70B")
249
- csv_file = gr.File(label="📂 Enviar novo CSV", file_types=[".csv"])
250
  upload_feedback = gr.Markdown()
251
  reset_btn = gr.Button("🔄 Resetar")
252
- export_json = gr.Button("📤 Exportar JSON")
253
- export_csv = gr.Button("📤 Exportar CSV")
254
 
255
  with gr.Column(scale=4):
256
  gr.Markdown("# 🧠 Anomalia Agent")
@@ -279,8 +266,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
279
  history_btn.click(toggle_history, outputs=history_output)
280
  csv_file.change(handle_csv_and_clear_chat, inputs=csv_file, outputs=[upload_feedback, chatbot])
281
  reset_btn.click(reset_all, outputs=[upload_feedback, chatbot, csv_file])
282
- export_json.click(lambda: export_history_json(), outputs=download_file)
283
- export_csv.click(lambda: export_history_csv(), outputs=download_file)
284
 
285
  if __name__ == "__main__":
286
  demo.launch(share=False)
 
103
  logging.error(f"[ERRO] Falha ao processar novo CSV: {e}")
104
  return f"❌ Erro ao processar CSV: {e}"
105
 
 
 
 
 
 
 
106
  def reset_app():
107
  global engine, db, sql_agent, query_cache, history_log, recent_history
108
  try:
 
121
  except Exception as e:
122
  return f"❌ Erro ao resetar: {e}"
123
 
124
+ engine = create_engine_and_load_db(get_active_csv_path(), SQL_DB_PATH)
125
+ db = SQLDatabase(engine=engine)
126
+ llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
127
+ sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True, max_iterations=40, return_intermediate_steps=True)
 
 
 
 
 
128
 
129
  def generate_initial_context(db_sample):
130
  return (
 
234
  with gr.Row():
235
  with gr.Column(scale=1):
236
  gr.Markdown("## ⚙️ Configurações")
237
+ model_selector = gr.Dropdown(list(LLAMA_MODELS.keys()), label=" ", value="LLaMA 70B")
238
+ csv_file = gr.File(label=" ", file_types=[".csv"])
239
  upload_feedback = gr.Markdown()
240
  reset_btn = gr.Button("🔄 Resetar")
 
 
241
 
242
  with gr.Column(scale=4):
243
  gr.Markdown("# 🧠 Anomalia Agent")
 
266
  history_btn.click(toggle_history, outputs=history_output)
267
  csv_file.change(handle_csv_and_clear_chat, inputs=csv_file, outputs=[upload_feedback, chatbot])
268
  reset_btn.click(reset_all, outputs=[upload_feedback, chatbot, csv_file])
 
 
269
 
270
  if __name__ == "__main__":
271
  demo.launch(share=False)