Update app.py

app.py

@@ -145,14 +145,19 @@ def preparar_sugestoes(df_falhas, evt: gr.SelectData):
         sugestoes = [int(x.strip()) for x in sugestoes.split(",") if x.strip().isdigit()]
 
     return idx, lado, intervalo, gr.update(choices=sugestoes, value=None)
+    
 
-
-def aplicar_sugestao(df_original, idx, numero_escolhido, coluna_num):
+def aplicar_sugestao(df_original, idx, numero_escolhido, coluna_num, df_falhas):
     if numero_escolhido is None:
-        return df_original
+        return df_original, df_falhas
     df_original = df_original.copy()
     df_original.loc[df_original["_idx"] == idx, coluna_num] = int(numero_escolhido)
-    return df_original
+
+    # Remove a linha corrigida do DataFrame de falhas
+    df_falhas = df_falhas[df_falhas["_idx"] != idx]
+
+    return df_original, df_falhas
+
 
 # Interface
 with gr.Blocks() as app:
@@ -184,8 +189,15 @@ with gr.Blocks() as app:
 
     btn_interpolar.click(fn=interpolar, inputs=[tabela, dropdown_cdlog, dropdown_num], outputs=[tabela_resultado, arquivo_saida, falhas])
     falhas.select(fn=preparar_sugestoes, inputs=[falhas], outputs=[idx_selecionado, info_lado, info_intervalo, radio_sugestoes])
-    btn_aplicar_sugestao.click(fn=aplicar_sugestao, inputs=[estado_tabela, idx_selecionado, radio_sugestoes, dropdown_num], outputs=estado_tabela).then(
-        fn=lambda df: df, inputs=estado_tabela, outputs=tabela
-    ).then(fn=interpolar, inputs=[estado_tabela, dropdown_cdlog, dropdown_num], outputs=[tabela_resultado, arquivo_saida, falhas])
+    btn_aplicar_sugestao.click(
+        fn=aplicar_sugestao,
+        inputs=[estado_tabela, idx_selecionado, radio_sugestoes, dropdown_num, falhas],
+        outputs=[estado_tabela, falhas]
+    ).then(
+        fn=interpolar,
+        inputs=[estado_tabela, dropdown_cdlog, dropdown_num],
+        outputs=[tabela_resultado, arquivo_saida, falhas]
+    )
+
 
 app.launch(server_name="0.0.0.0", server_port=7860)