Update app.py

app.py

@@ -15,8 +15,10 @@ load_dotenv()
 UPLOAD_DIR = "uploaded_data"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
-CSV_FILE_PATH = "anomalia_vendas.csv"
+DEFAULT_CSV_PATH = "anomalia_vendas.csv"
+UPLOADED_CSV_PATH = os.path.join(UPLOAD_DIR, "tabela.csv")
 SQL_DB_PATH = "data.db"
+
 HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
@@ -43,53 +45,46 @@ query_cache = {}
 history_log = []  
 recent_history = []  
 show_history_flag = False
+engine = None 
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def create_or_load_sql_database(csv_path, sql_db_path):
-    if os.path.exists(sql_db_path):
-        print("Banco de dados SQL já existe. Carregando...")
-        return create_engine(f"sqlite:///{sql_db_path}")
+# === Controle do CSV ativo ===
+def get_active_csv_path():
+    """Retorna o CSV ativo: o carregado ou o padrão."""
+    if os.path.exists(UPLOADED_CSV_PATH):
+        logging.info(f"[CSV] Usando arquivo CSV carregado: {UPLOADED_CSV_PATH}")
+        return UPLOADED_CSV_PATH
     else:
-        print("Banco de dados SQL não encontrado. Criando...")
-        engine = create_engine(f"sqlite:///{sql_db_path}")
-        df = pd.read_csv(csv_path, sep=";", on_bad_lines="skip")
-        print(f"CSV carregado: {len(df)} linhas, {len(df.columns)} colunas")
-        df.to_sql("anomalia_vendas", engine, index=False, if_exists="replace")
-        print("Banco de dados SQL criado com sucesso!")
-        return engine
-
-def load_uploaded_csv_and_create_db(uploaded_file):
-    if uploaded_file is None:
-        return None
-
-    print(f"[UPLOAD] CSV recebido: {uploaded_file}")
-
-    engine = create_engine(f"sqlite:///{SQL_DB_PATH}")
-    df = pd.read_csv(uploaded_file, sep=";", on_bad_lines="skip")
+        logging.info(f"[CSV] Usando arquivo CSV padrão: {DEFAULT_CSV_PATH}")
+        return DEFAULT_CSV_PATH
+
+def create_engine_and_load_db(csv_path, sql_db_path):
+    logging.info(f"[ENGINE] Criando engine com CSV: {csv_path}")
+    engine = create_engine(f"sqlite:///{sql_db_path}")
+    df = pd.read_csv(csv_path, sep=";", on_bad_lines="skip")
     df.to_sql("anomalia_vendas", engine, index=False, if_exists="replace")
-    print("Banco recriado com base no novo CSV.")
-    print(f"CSV carregado: {len(df)} linhas, {len(df.columns)} colunas")
-    print(f"[DEBUG] Novo engine criado: {engine}")
+    logging.info(f"[ENGINE] Banco criado com sucesso com {len(df)} linhas e {len(df.columns)} colunas")
     return engine
 
-engine = create_or_load_sql_database(CSV_FILE_PATH, SQL_DB_PATH)
-
-def refresh_sql_agent():
-    global db, sql_agent
-    db = SQLDatabase(engine=engine)
-    sql_agent = create_sql_agent(
-        ChatOpenAI(model="gpt-4o-mini", temperature=0),
-        db=db,
-        agent_type="openai-tools",
-        verbose=True,
-        max_iterations=40,
-        return_intermediate_steps=True
-    )
-    print("[SQL_AGENT] Atualizado com novo banco de dados.")
-
-refresh_sql_agent()
+# === Upload do CSV via UI ===
+def handle_csv_upload(file):
+    global engine, db
+    try:
+        with open(UPLOADED_CSV_PATH, "wb") as f:
+            f.write(file.read())
+        logging.info(f"[UPLOAD] CSV salvo como: {UPLOADED_CSV_PATH}")
+        engine = create_engine_and_load_db(UPLOADED_CSV_PATH, SQL_DB_PATH)
+        db = SQLDatabase(engine=engine)
+        logging.info("[UPLOAD] Novo banco carregado e DB atualizado.")
+    except Exception as e:
+        logging.error(f"[ERRO] Falha ao processar novo CSV: {e}")
 
+# === Inicialização ===
+engine = create_engine_and_load_db(get_active_csv_path(), SQL_DB_PATH)
+db = SQLDatabase(engine=engine)
+llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
+sql_agent = create_sql_agent(llm, db=db, agent_type="openai-tools", verbose=True, max_iterations=40, return_intermediate_steps=True)
 
 def generate_initial_context(db_sample):
     return (
@@ -224,15 +219,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
             history_output = gr.JSON()
             history_btn.click(toggle_history, inputs=[], outputs=history_output)
-
-        def handle_csv_upload(file):
-            global engine
-            try:
-                engine = load_uploaded_csv_and_create_db(file)
-                if engine is not None:
-                    refresh_sql_agent()
-            except Exception as e:
-                print(f"[ERRO] Falha ao processar novo CSV: {e}")
     
         csv_file.change(handle_csv_upload, inputs=csv_file, outputs=csv_file)