Update app.py

app.py

@@ -74,10 +74,22 @@ def load_uploaded_csv_and_create_db(uploaded_file):
     return engine
 
 engine = create_or_load_sql_database(CSV_FILE_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 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()
+
 
 def generate_initial_context(db_sample):
     return (
@@ -214,23 +226,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             history_btn.click(toggle_history, inputs=[], outputs=history_output)
 
         def handle_csv_upload(file):
-            global engine, db, sql_agent
-
+            global engine
             try:
                 engine = load_uploaded_csv_and_create_db(file)
                 if engine is not None:
-                    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("[UPLOAD] Banco e agente SQL atualizados com sucesso.")
+                    refresh_sql_agent()
             except Exception as e:
-                print(f"[ERRO] Falha ao processar novo CSV: {e}")
+                print(f"[ERRO] Falha ao processar novo CSV: {e}
     
         csv_file.change(handle_csv_upload, inputs=csv_file, outputs=csv_file)