Update app.py

app.py

@@ -334,7 +334,7 @@ def fit_and_optimize_model():
     equation_formatted = f"### Ecuación del Modelo Simplificado:<br>{equation_formatted}"
     excel_path = rsm.save_tables_to_excel()
     zip_path = rsm.save_figures_to_zip()
-    return (model_completo.summary().as_html(), pareto_completo, model_simplificado_output, pareto_simplificado, equation_formatted, optimization_table, prediction_table, contribution_table, anova_table, zip_path, excel_path)
+    return (model_completo_output, pareto_completo, model_simplificado_output, pareto_simplificado, equation_output, optimization_table, prediction_table, contribution_table, anova_table, zip_path, excel_path)
 
 def fit_custom_model(factor_checkboxes, interaction_checkboxes, model_personalized_output_component, pareto_personalized_output_component):
     if 'rsm' not in globals(): return [None]*2
@@ -498,11 +498,10 @@ def create_gradio_interface():
                 with gr.Row():
                     left_button = gr.Button("<")
                     right_button = gr.Button(">")
+                    download_plot_button_comp = gr.DownloadButton("Descargar Gráfico Actual (PNG)") # Defined HERE
+                    download_all_plots_button_comp = gr.DownloadButton("Descargar Todos los Gráficos (ZIP)")
                 rsm_plot_output_comp = rsm_plot_output
                 plot_info_comp = plot_info
-                with gr.Row():
-                    download_plot_button_comp = download_plot_button # Correctly assigned here now
-                    download_all_plots_button_comp = download_all_plots_button
                 current_index_state_comp = current_index_state
                 all_figures_state_comp = all_figures_state
                 current_model_type_state_comp = current_model_type_state