avalia-se commited on
Commit
e930af0
·
verified ·
1 Parent(s): 3d95093

Update modules/rl.py

Browse files
Files changed (1) hide show
  1. 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 (buffer, "resultado.xlsx")
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: exportar_para_excel(
1417
- df_planilha, df_infos, df_original, df_escalado, df_out,
1418
- resultados_gerais, resultados_vars, df_calc_obs, df_calc_obs_trans
 
 
 
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(