Update modules/planilha.py

modules/planilha.py

@@ -6,7 +6,6 @@ from .shared_state import state  # Importa o estado compartilhado
 # Variável global para armazenar o DataFrame original
 original_df = None
 
-# PLANILHA
 def list_sheets(file):
     if file is None:
         return "Nenhum arquivo carregado.", []
@@ -47,7 +46,6 @@ def add_new_variable(file, sheet_name, first_var, operation, second_var, new_var
     df = pd.read_excel(file.name, sheet_name=sheet_name)
     if first_var not in df.columns or second_var not in df.columns:
         return pd.DataFrame({"Erro": ["Variáveis selecionadas não existem no DataFrame."]})
-
     try:
         if operation == "Adição":
             df[new_var_name] = df[first_var] + df[second_var]
@@ -61,7 +59,6 @@ def add_new_variable(file, sheet_name, first_var, operation, second_var, new_var
         return pd.DataFrame({"Erro": ["Divisão por zero detectada."]})
     except Exception as e:
         return pd.DataFrame({"Erro": [f"Erro ao realizar a operação: {str(e)}"]})
-
     return df
 
 def update_variable_choices(file, sheet_name):
@@ -83,19 +80,17 @@ def update_columns(file, sheet_name):
         gr.update(choices=columns, value=[], interactive=True)
     )
 
-# Função para restaurar o DataFrame ao estado inicial
 def restore_dataframe():
     global original_df
     if original_df is not None:
-        return original_df, None  # Retorna o DataFrame e um valor nulo para limpar o download
-    return pd.DataFrame(), None  # Retorna DataFrame vazio e nenhum arquivo
+        return original_df, None
+    return pd.DataFrame(), None
 
 def finalize_dataframe(file, sheet_name, selected_columns, first_var, operation, second_var, new_var_name, add_index):
     if file is None or not sheet_name:
         return pd.DataFrame({"Erro": ["Carregue um arquivo e selecione uma aba."]})
     df = pd.read_excel(file.name, sheet_name=sheet_name)
 
-    # Adiciona a nova variável, se necessário
     if new_var_name and first_var and second_var:
         try:
             if operation == "Adição":
@@ -111,74 +106,56 @@ def finalize_dataframe(file, sheet_name, selected_columns, first_var, operation,
         except Exception as e:
             return pd.DataFrame({"Erro": [f"Erro ao realizar a operação: {str(e)}"]})
 
-    # Adiciona a nova variável às colunas selecionadas, se existir
     if new_var_name and new_var_name in df.columns:
         selected_columns.append(new_var_name)
-
     if selected_columns:
         df = df[selected_columns]
 
-    # Adiciona índice na primeira coluna, se necessário
     if add_index:
         df.insert(0, 'Índice', range(1, len(df) + 1))
 
-    # Salvando DataFrame filtrado em arquivo CSV
     file_path = "Planilha_final.xlsx"
     df.to_excel(file_path)
-
     return df, file_path
 
 def planilha_tab(filtered_df_output):
-    # Criação da aba
     with gr.Tab("Carregar Planilha"):
-        # Checkbox para escolher entre usar o arquivo da aba anterior ou carregar um novo arquivo
-        use_filtered_df = gr.Checkbox(label="Usar arquivo da aba anterior", value=True)
-
-        # Upload do arquivo Excel
         with gr.Row():
-            excel_file = gr.File(label="Carregue sua planilha Excel", file_types=[".xls", ".xlsx"], elem_classes=["small-file-upload"], visible=False)
+            excel_file = gr.File(label="Carregue sua planilha Excel", file_types=[".xls", ".xlsx"], elem_classes=["small-file-upload"])
 
         with gr.Row():
             list_button = gr.Button("Listar Abas")
 
-        # Exibição das abas disponíveis e seleção de aba
         with gr.Row():
             sheet_output = gr.Textbox(label="Abas disponíveis", interactive=False)
             sheet_dropdown = gr.Dropdown(label="Selecione uma aba")
 
-        # Botão para carregar aba e exibir colunas
         with gr.Row():
             load_button = gr.Button("Carregar Aba")
 
-        # Informações e exibição de colunas da aba carregada
         with gr.Row():
             sheet_info = gr.Text(label="Informação de Linhas e Colunas")
             sheet_columns = gr.JSON(label="Lista de Colunas", visible=False)
             view_sheet_checkbox = gr.Checkbox(label="Visualizar a planilha", value=False)
 
-        # Exibição do conteúdo da aba selecionada
         with gr.Row():
             sheet_content = gr.Dataframe(
                 label="Conteúdo da Aba Selecionada",
                 visible=False
             )
 
-        # CheckboxGroup para seleção de colunas
         with gr.Row():
             column_selector = gr.CheckboxGroup(
                 label="Selecione colunas para análise",
-                choices=[],  # Inicia vazio
+                choices=[],
                 interactive=True
             )
 
-        # Checkbox para adicionar índice na primeira coluna
         with gr.Row():
             add_index_checkbox = gr.Checkbox(label="Adicionar índice na primeira coluna", value=False)
 
-        # Checkbox para habilitar operações com variáveis
         operations_checkbox = gr.Checkbox(label="Operações com variáveis", value=False)
 
-        # Elementos relacionados às operações, inicialmente invisíveis
         with gr.Row(visible=False) as operations_inputs:
             variable_name_textbox = gr.Textbox(
                 label="Nome da nova variável",
@@ -201,47 +178,31 @@ def planilha_tab(filtered_df_output):
                 interactive=True
             )
 
-        # Associações de eventos
-        use_filtered_df.change(lambda x: gr.update(visible=not x), inputs=[use_filtered_df], outputs=[excel_file])
         list_button.click(update_dropdown, inputs=[excel_file], outputs=[sheet_output, sheet_dropdown])
-        load_button.click(update_columns, inputs=[excel_file, sheet_dropdown], outputs=[sheet_content, sheet_info, sheet_columns])
-        load_button.click(update_column_selector, inputs=[excel_file, sheet_dropdown], outputs=[column_selector])
+        load_button.click(update_columns, inputs=[excel_file, sheet_dropdown], outputs=[sheet_content, sheet_info, sheet_columns, column_selector])
         view_sheet_checkbox.change(toggle_sheet_visibility, inputs=[view_sheet_checkbox], outputs=[sheet_content])
         operations_checkbox.change(toggle_operations_inputs, inputs=[operations_checkbox], outputs=[operations_inputs])
 
         apply_operations_button = gr.Button("Criar Nova Variável")
         new_df_output = gr.Dataframe(label="Novo DataFrame", interactive=True)
+
         apply_operations_button.click(add_new_variable, inputs=[excel_file, sheet_dropdown, first_variable_dropdown, operation_dropdown, second_variable_dropdown, variable_name_textbox], outputs=[new_df_output])
         load_button.click(update_variable_choices, inputs=[excel_file, sheet_dropdown], outputs=[first_variable_dropdown, second_variable_dropdown])
 
-        # Botão para finalizar o DataFrame
         finalize_button = gr.Button("Finalizar DataFrame")
         download_output = gr.File(label="Baixar Novo Dataframe")
+
         finalize_button.click(finalize_dataframe, inputs=[excel_file, sheet_dropdown, column_selector, first_variable_dropdown, operation_dropdown, second_variable_dropdown, variable_name_textbox, add_index_checkbox], outputs=[new_df_output, download_output])
 
-        # Adicionar botões "Restaurar" e "Limpar"
         with gr.Row():
             restore_button = gr.Button("Restaurar")
-            clear_button = gr.ClearButton(components=[excel_file, sheet_output, sheet_dropdown, sheet_info, sheet_columns, view_sheet_checkbox, sheet_content, column_selector, add_index_checkbox, operations_checkbox, variable_name_textbox, first_variable_dropdown, operation_dropdown, second_variable_dropdown, new_df_output, download_output],
-                                          value="Limpar")
+            clear_button = gr.ClearButton(components=[excel_file, sheet_output, sheet_dropdown, sheet_info, sheet_columns, view_sheet_checkbox, sheet_content, column_selector, add_index_checkbox, operations_checkbox, variable_name_textbox, first_variable_dropdown, operation_dropdown, second_variable_dropdown, new_df_output, download_output], value="Limpar")
 
-        # Associações de eventos para os botões "Restaurar"
         restore_button.click(restore_dataframe, outputs=[new_df_output, download_output])
 
-        # Se a opção "Usar arquivo da aba anterior" estiver marcada, carregue o filtered_df_output
-        def load_filtered_df(use_filtered_df, filtered_df_output):
-            if use_filtered_df:
-                df = pd.DataFrame(filtered_df_output)
-                return df, f"{df.shape[0]} linhas e {df.shape[1]} colunas", df.columns.tolist(), df.columns.tolist()
-            return None, "", [], []
-
-        use_filtered_df.change(load_filtered_df, inputs=[use_filtered_df, filtered_df_output], outputs=[sheet_content, sheet_info, column_selector, column_selector])
-
     return locals(), new_df_output
 
 
 
-
-
 ### Manipulação de colunas
 ### Receber planilha da aba dados