Update app.py

app.py

@@ -12,6 +12,9 @@ import logging
 
 load_dotenv()
 
+UPLOAD_DIR = "uploaded_data"
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
 CSV_FILE_PATH = "anomalia_vendas.csv"
 SQL_DB_PATH = "data.db"
 HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
@@ -56,6 +59,23 @@ def create_or_load_sql_database(csv_path, sql_db_path):
         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
+
+    saved_path = os.path.join(UPLOAD_DIR, "tabela.csv")
+
+    with open(saved_path, "wb") as f:
+        f.write(uploaded_file.read())
+
+    print(f"[UPLOAD] Novo CSV salvo em: {saved_path}")
+    
+    engine = create_engine(f"sqlite:///{SQL_DB_PATH}")
+    df = pd.read_csv(saved_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.")
+    return engine
+
 engine = create_or_load_sql_database(CSV_FILE_PATH, SQL_DB_PATH)
 db = SQLDatabase(engine=engine)
 
@@ -177,6 +197,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 label="Escolha o Modelo LLM para gerar a query SQL",
                 value="LLaMA 70B"
             )
+            csv_file = gr.File(label="📂 Enviar novo CSV", file_types=[".csv"])
 
         with gr.Column(scale=4):
             chatbot = gr.Chatbot(height=600)
@@ -195,5 +216,21 @@ 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, db, sql_agent
+            engine = load_uploaded_csv_and_create_db(file)
+            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
+            )
+            return gr.update(value=None)
+    
+        csv_file.change(handle_csv_upload, inputs=csv_file, outputs=csv_file)
+
 if __name__ == "__main__":
     demo.launch(share=False)