rwayz commited on
Commit
dc74977
·
1 Parent(s): a10e0b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,7 +16,7 @@ load_dotenv()
16
  UPLOAD_DIR = "uploaded_data"
17
  os.makedirs(UPLOAD_DIR, exist_ok=True)
18
 
19
- DEFAULT_CSV_PATH = "anomalia_vendas.csv"
20
  UPLOADED_CSV_PATH = os.path.join(UPLOAD_DIR, "tabela.csv")
21
  SQL_DB_PATH = "data.db"
22
 
@@ -64,7 +64,7 @@ def create_engine_and_load_db(csv_path, sql_db_path):
64
  logging.info(f"[ENGINE] Criando engine com CSV: {csv_path}")
65
  engine = create_engine(f"sqlite:///{sql_db_path}")
66
  df = pd.read_csv(csv_path, sep=";", on_bad_lines="skip")
67
- df.to_sql("anomalia_vendas", engine, index=False, if_exists="replace")
68
  logging.info(f"[ENGINE] Banco criado com sucesso com {len(df)} linhas e {len(df.columns)} colunas")
69
  return engine
70
 
@@ -153,7 +153,7 @@ def query_sql_agent(user_query, selected_model_name):
153
  query_cache[user_query] = greeting_response
154
  return greeting_response
155
 
156
- column_data = pd.read_sql_query("SELECT * FROM anomalia_vendas LIMIT 10", engine)
157
  llama_instruction = query_with_llama(user_query, column_data, selected_model_name)
158
 
159
  if not llama_instruction:
 
16
  UPLOAD_DIR = "uploaded_data"
17
  os.makedirs(UPLOAD_DIR, exist_ok=True)
18
 
19
+ DEFAULT_CSV_PATH = "tabela.csv"
20
  UPLOADED_CSV_PATH = os.path.join(UPLOAD_DIR, "tabela.csv")
21
  SQL_DB_PATH = "data.db"
22
 
 
64
  logging.info(f"[ENGINE] Criando engine com CSV: {csv_path}")
65
  engine = create_engine(f"sqlite:///{sql_db_path}")
66
  df = pd.read_csv(csv_path, sep=";", on_bad_lines="skip")
67
+ df.to_sql("tabela", engine, index=False, if_exists="replace")
68
  logging.info(f"[ENGINE] Banco criado com sucesso com {len(df)} linhas e {len(df.columns)} colunas")
69
  return engine
70
 
 
153
  query_cache[user_query] = greeting_response
154
  return greeting_response
155
 
156
+ column_data = pd.read_sql_query("SELECT * FROM tabela LIMIT 10", engine)
157
  llama_instruction = query_with_llama(user_query, column_data, selected_model_name)
158
 
159
  if not llama_instruction: