feat: add delete function

Início.py

@@ -1,9 +1,10 @@
 import streamlit as st
 import pandas as pd
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, MetaData, Table, delete
+from sqlalchemy.orm import sessionmaker
 from filter import filter_dataframe
-import toml
 import os
+import time
 
 # Dados de login
 USERNAME = os.getenv("username")
@@ -52,12 +53,25 @@ def load_data() -> pd.DataFrame:
     connection_url = f"{dialect}+{driver}://{username}:{password}@{host}:{port}/{database}"
 
     engine = create_engine(connection_url)
+    Session = sessionmaker(bind=engine)
+    session = Session()
     
     with open('query.sql', 'r', encoding='utf-8') as file:
         query = file.read()
     
     df = pd.read_sql(query, engine)
-    return df
+    
+    return df, engine, session
+
+# Função para deletar linhas selecionadas
+def delete_selected_rows(selected_ids, engine, session):
+    meta = MetaData()
+    table = Table('Extraction', meta, autoload_with=engine)
+    with session.begin():
+        for id_value in selected_ids:
+            stmt = delete(table).where(table.c.id == id_value)
+            session.execute(stmt)
+        session.commit()
 
 # Função para download dos dados como CSV
 def convert_df_to_csv(df):
@@ -73,23 +87,32 @@ if not st.session_state['logged_in']:
 else:
     st.header('⛰️ Banco de Dados - GeoCosmos')
 
-    df = load_data()
+    df, engine, session = load_data()
 
+    # Aplica o filtro ao dataframe
     filtered_df = filter_dataframe(df)
 
-    all_columns = filtered_df.columns.tolist()
-    selected_columns = st.multiselect(
-        'Escolha as colunas para exibir:',
-        options=all_columns,
-        placeholder="Selecione uma ou mais colunas",
-        default=[] 
-    )
+    # Adiciona a coluna de seleção
+    filtered_df.insert(0, 'Seleção', False)
 
-    if selected_columns:
-        displayed_df = filtered_df[selected_columns]
-    else:
-        displayed_df = filtered_df
+    # Exibe o dataframe filtrado com checkboxes para seleção
+    edited_df = st.data_editor(filtered_df, disabled=[col for col in filtered_df.columns if col != 'Seleção'], hide_index=True, column_order = [col for col in filtered_df.columns if col != 'id'])
+
+    # Verifica se há pelo menos um checkbox selecionado
+    selected_ids = edited_df.loc[edited_df['Seleção'], 'id'].tolist()
+
+    if selected_ids:
+        with st.popover("APAGAR"):
+            st.warning("Você está prestes a apagar dados. Esta ação não pode ser desfeita.")
+            confirmation = st.text_input("Digite 'APAGAR' e pressione ENTER para confirmar a deleção:")
 
-    st.write('Tabela Atualizada')
-    st.dataframe(displayed_df, column_order=selected_columns if selected_columns else None)
+            if confirmation == "APAGAR":
+                delete_selected_rows(selected_ids, engine, session)
+                st.success(f'Linhas com IDs {selected_ids} foram deletadas.')
 
+                time.sleep(3)
+                st.rerun()
+            else:
+                st.write("")
+    else:
+        st.write("")