Update app.py

app.py

@@ -517,9 +517,6 @@ class RSM_BoxBehnken:
         """
         Guarda todas las tablas en un archivo Excel con múltiples hojas y retorna la ruta del archivo.
         """
-        if 'rsm' not in globals():
-            return None
-
         tables = self.get_all_tables()
         excel_buffer = io.BytesIO()
         with pd.ExcelWriter(excel_buffer, engine='xlsxwriter') as writer:
@@ -562,7 +559,7 @@ def load_data(x1_name, x2_name, x3_name, y_name, x1_levels_str, x2_levels_str, x
         rsm = RSM_BoxBehnken(data, x1_name, x2_name, x3_name, y_name, x1_levels, x2_levels, x3_levels)
 
         return data.round(3), x1_name, x2_name, x3_name, y_name, x1_levels, x2_levels, x3_levels, gr.update(visible=True)
-    
+
     except Exception as e:
         # Mostrar mensaje de error
         error_message = f"Error al cargar los datos: {str(e)}"
@@ -571,8 +568,8 @@ def load_data(x1_name, x2_name, x3_name, y_name, x1_levels_str, x2_levels_str, x
 
 def fit_and_optimize_model():
     if 'rsm' not in globals():
-        return [None]*10
-    
+        return [None]*11  # Ajustar el número de outputs
+
     # Ajustar modelos y optimizar
     model_completo, pareto_completo = rsm.fit_model()
     model_simplificado, pareto_simplificado = rsm.fit_simplified_model()
@@ -605,8 +602,8 @@ def fit_and_optimize_model():
         prediction_table,
         contribution_table,
         anova_table,
-        zip_path,  # Ruta del ZIP de gráficos
-        excel_path  # Ruta del Excel de tablas
+        zip_path,    # Ruta del ZIP de gráficos
+        excel_path   # Ruta del Excel de tablas
     )
 
 def show_plot(current_index, all_figures):
@@ -652,15 +649,18 @@ def download_current_plot(all_figures, current_index):
     
     return temp_path  # Retornar solo la ruta
 
-def download_all_plots_zip(all_figures):
+def download_all_plots_zip():
     """
     Descarga todas las figuras en un archivo ZIP.
     """
-    if not all_figures:
+    if 'rsm' not in globals():
         return None
     zip_path = rsm.save_figures_to_zip()
-    filename = f"Graficos_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip"
-    return zip_path  # Retornar solo la ruta
+    if zip_path:
+        filename = f"Graficos_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip"
+        # Renombrar el archivo temporal al nombre deseado
+        return (zip_path, filename)
+    return None
 
 def download_all_tables_excel():
     """
@@ -669,8 +669,11 @@ def download_all_tables_excel():
     if 'rsm' not in globals():
         return None
     excel_path = rsm.save_tables_to_excel()
-    filename = f"Tablas_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
-    return excel_path  # Retornar solo la ruta
+    if excel_path:
+        filename = f"Tablas_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
+        # Renombrar el archivo temporal al nombre deseado
+        return (excel_path, filename)
+    return None
 
 # --- Crear la interfaz de Gradio ---
 
@@ -742,7 +745,7 @@ with gr.Blocks() as demo:
                 download_all_plots_button = gr.DownloadButton("Descargar Todos los Gráficos (ZIP)")
             current_index_state = gr.State(0)  # Estado para el índice actual
             all_figures_state = gr.State([])  # Estado para todas las figuras
-    
+
     # Cargar datos
     load_button.click(
         load_data,
@@ -764,27 +767,32 @@ with gr.Blocks() as demo:
             prediction_table_output,
             contribution_table_output,
             anova_table_output,
-            download_all_plots_button,
-            download_excel_button
+            download_all_plots_button,  # Ruta del ZIP de gráficos
+            download_excel_button      # Ruta del Excel de tablas
         ]
     )
     
     # Generar y mostrar los gráficos
     plot_button.click(
-        lambda fixed_var, fixed_lvl: (rsm.plot_rsm_individual(fixed_var, fixed_lvl), "Gráfico 1 de " + str(len(rsm.all_figures)), 0),
+        lambda fixed_var, fixed_lvl: (
+            rsm.plot_rsm_individual(fixed_var, fixed_lvl),
+            f"Gráfico 1 de {len(rsm.all_figures)}" if rsm.all_figures else "No hay gráficos disponibles.",
+            0,
+            rsm.all_figures  # Actualizar el estado de todas las figuras
+        ),
         inputs=[fixed_variable_input, fixed_level_input],
-        outputs=[rsm_plot_output, plot_info, current_index_state]
+        outputs=[rsm_plot_output, plot_info, current_index_state, all_figures_state]
     )
     
     # Navegación de gráficos
     left_button.click(
-        navigate_plot,
-        inputs=[gr.Button.get_value(left_button), current_index_state, all_figures_state],
+        lambda current_index, all_figures: navigate_plot('left', current_index, all_figures),
+        inputs=[current_index_state, all_figures_state],
         outputs=[rsm_plot_output, plot_info, current_index_state]
     )
     right_button.click(
-        navigate_plot,
-        inputs=[gr.Button.get_value(right_button), current_index_state, all_figures_state],
+        lambda current_index, all_figures: navigate_plot('right', current_index, all_figures),
+        inputs=[current_index_state, all_figures_state],
         outputs=[rsm_plot_output, plot_info, current_index_state]
     )
     
@@ -798,7 +806,7 @@ with gr.Blocks() as demo:
     # Descargar todos los gráficos en ZIP
     download_all_plots_button.click(
         download_all_plots_zip,
-        inputs=[all_figures_state],
+        inputs=[],
         outputs=download_all_plots_button
     )