Update modules/rl.py
Browse files- modules/rl.py +14 -14
modules/rl.py
CHANGED
|
@@ -764,6 +764,7 @@ def exportar_para_excel(df_planilha, df_infos, df_original, df_escalado, df_outl
|
|
| 764 |
df_original.to_excel(writer, sheet_name='Dados Modelo', index=False)
|
| 765 |
df_outliers.to_excel(writer, sheet_name='Outliers', index=False)
|
| 766 |
|
|
|
|
| 767 |
resultados_lista = [linha.strip() for linha in resultados_gerais.split("\n") if linha.strip()]
|
| 768 |
resultados_df = pd.DataFrame({"Descrição": resultados_lista})
|
| 769 |
resultados_df.to_excel(writer, sheet_name='Resultados Gerais', index=False)
|
|
@@ -775,10 +776,11 @@ def exportar_para_excel(df_planilha, df_infos, df_original, df_escalado, df_outl
|
|
| 775 |
buffer.seek(0)
|
| 776 |
|
| 777 |
# Retorna o conteúdo como arquivo Gradio
|
| 778 |
-
return
|
| 779 |
-
|
| 780 |
except Exception as e:
|
| 781 |
return f"Erro ao criar o arquivo Excel: {str(e)}"
|
|
|
|
| 782 |
|
| 783 |
# Função para salvar o modelo
|
| 784 |
def exportar_modelo_completo_avse(nome_pacote, modelo, resultados_gerais, df_planilha, df_infos, df_original, df_outliers, resultados_vars, df_calc_obs):
|
|
@@ -805,9 +807,12 @@ def exportar_modelo_completo_avse(nome_pacote, modelo, resultados_gerais, df_pla
|
|
| 805 |
|
| 806 |
# Salvar o pacote usando joblib
|
| 807 |
dump(pacote, nome_pacote)
|
|
|
|
| 808 |
return f"Pacote '{nome_pacote}' criado com sucesso."
|
|
|
|
| 809 |
except Exception as e:
|
| 810 |
return f"Erro ao criar o pacote: {str(e)}"
|
|
|
|
| 811 |
|
| 812 |
# Função para carregar o modelo
|
| 813 |
def carregar_modelo(nome_pacote):
|
|
@@ -1413,24 +1418,19 @@ def rl_tab():
|
|
| 1413 |
|
| 1414 |
# Botão para conectar à função de exportar para o Excel
|
| 1415 |
botao_exportar_excel.click(
|
| 1416 |
-
lambda df_planilha, df_infos, df_original, df_escalado, df_out, resultados_gerais, resultados_vars, df_calc_obs, df_calc_obs_trans:
|
| 1417 |
-
|
| 1418 |
-
|
|
|
|
|
|
|
|
|
|
| 1419 |
),
|
| 1420 |
inputs=[
|
| 1421 |
dados_state, tabela_cabecalhos, tabela_original, tabela_escalada, df_out,
|
| 1422 |
resultado_geral, resultado_coef, calc_obs, calc_obs_trans
|
| 1423 |
],
|
| 1424 |
outputs=[arquivo_excel]
|
| 1425 |
-
|
| 1426 |
-
|
| 1427 |
-
# Atualizar visibilidade do status após clique no botão
|
| 1428 |
-
botao_exportar_excel.click(
|
| 1429 |
-
lambda: gr.update(visible=True),
|
| 1430 |
-
inputs=None,
|
| 1431 |
-
outputs=[arquivo_excel]
|
| 1432 |
-
)
|
| 1433 |
-
|
| 1434 |
|
| 1435 |
# 3.6. Botão para conectar à função de salvar o modelo (SALVAR MODELO)
|
| 1436 |
botao_salvar_modelo.click(
|
|
|
|
| 764 |
df_original.to_excel(writer, sheet_name='Dados Modelo', index=False)
|
| 765 |
df_outliers.to_excel(writer, sheet_name='Outliers', index=False)
|
| 766 |
|
| 767 |
+
# Processar resultados_gerais
|
| 768 |
resultados_lista = [linha.strip() for linha in resultados_gerais.split("\n") if linha.strip()]
|
| 769 |
resultados_df = pd.DataFrame({"Descrição": resultados_lista})
|
| 770 |
resultados_df.to_excel(writer, sheet_name='Resultados Gerais', index=False)
|
|
|
|
| 776 |
buffer.seek(0)
|
| 777 |
|
| 778 |
# Retorna o conteúdo como arquivo Gradio
|
| 779 |
+
return buffer.getvalue()
|
| 780 |
+
|
| 781 |
except Exception as e:
|
| 782 |
return f"Erro ao criar o arquivo Excel: {str(e)}"
|
| 783 |
+
|
| 784 |
|
| 785 |
# Função para salvar o modelo
|
| 786 |
def exportar_modelo_completo_avse(nome_pacote, modelo, resultados_gerais, df_planilha, df_infos, df_original, df_outliers, resultados_vars, df_calc_obs):
|
|
|
|
| 807 |
|
| 808 |
# Salvar o pacote usando joblib
|
| 809 |
dump(pacote, nome_pacote)
|
| 810 |
+
|
| 811 |
return f"Pacote '{nome_pacote}' criado com sucesso."
|
| 812 |
+
|
| 813 |
except Exception as e:
|
| 814 |
return f"Erro ao criar o pacote: {str(e)}"
|
| 815 |
+
|
| 816 |
|
| 817 |
# Função para carregar o modelo
|
| 818 |
def carregar_modelo(nome_pacote):
|
|
|
|
| 1418 |
|
| 1419 |
# Botão para conectar à função de exportar para o Excel
|
| 1420 |
botao_exportar_excel.click(
|
| 1421 |
+
lambda df_planilha, df_infos, df_original, df_escalado, df_out, resultados_gerais, resultados_vars, df_calc_obs, df_calc_obs_trans: (
|
| 1422 |
+
"resultado.xlsx",
|
| 1423 |
+
exportar_para_excel(
|
| 1424 |
+
df_planilha, df_infos, df_original, df_escalado, df_out,
|
| 1425 |
+
resultados_gerais, resultados_vars, df_calc_obs, df_calc_obs_trans
|
| 1426 |
+
)
|
| 1427 |
),
|
| 1428 |
inputs=[
|
| 1429 |
dados_state, tabela_cabecalhos, tabela_original, tabela_escalada, df_out,
|
| 1430 |
resultado_geral, resultado_coef, calc_obs, calc_obs_trans
|
| 1431 |
],
|
| 1432 |
outputs=[arquivo_excel]
|
| 1433 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1434 |
|
| 1435 |
# 3.6. Botão para conectar à função de salvar o modelo (SALVAR MODELO)
|
| 1436 |
botao_salvar_modelo.click(
|