Update app.py

app.py

@@ -3,7 +3,6 @@ import pandas as pd
 import statsmodels.formula.api as smf
 import statsmodels.api as sm
 import plotly.graph_objects as go
-from plotly.subplots import make_subplots
 from scipy.optimize import minimize
 import plotly.express as px
 from scipy.stats import t, f
@@ -568,11 +567,13 @@ def load_data(x1_name, x2_name, x3_name, y_name, x1_levels_str, x2_levels_str, x
         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:
-        return None, "", "", "", "", [], [], [], gr.update(visible=False), f"Error: {e}"
+        # Nota: Para mostrar mensajes de error, Gradio necesita manejarlo de manera diferente
+        # Aquí, simplemente retornamos nada y ocultamos la sección de análisis
+        return None, "", "", "", "", [], [], [], gr.update(visible=False)
 
 def fit_and_optimize_model():
     if 'rsm' not in globals():
-        return None, None, None, None, None, None, None, None, None, gr.State(), gr.State()
+        return [None]*10, 0
     
     # Ajustar modelos y optimizar
     model_completo, pareto_completo = rsm.fit_model()
@@ -601,26 +602,23 @@ def fit_and_optimize_model():
         contribution_table,
         anova_table,
         rsm.all_figures,  # Devuelve todas las figuras generadas
-        gr.State()  # Inicializa el índice actual
+        0  # Inicializar el índice actual
     )
 
-def generate_rsm_plot(all_figures, current_index):
+def show_plot(all_figures, current_index):
     if not all_figures:
         return None, "No hay gráficos disponibles.", current_index
-    
     selected_fig = all_figures[current_index]
     plot_info_text = f"Gráfico {current_index + 1} de {len(all_figures)}"
-    
     return selected_fig, plot_info_text, current_index
 
-def navigate_plot(direction, current_index, total_plots, all_figures):
+def navigate_plot(direction, current_index, all_figures):
     """
     Navega entre los gráficos.
 
     Args:
         direction (str): 'left' o 'right'.
         current_index (int): Índice actual.
-        total_plots (int): Total de gráficos.
         all_figures (list): Lista de todas las figuras.
 
     Returns:
@@ -630,14 +628,14 @@ def navigate_plot(direction, current_index, total_plots, all_figures):
         return None, "No hay gráficos disponibles.", current_index
     
     if direction == 'left':
-        new_index = (current_index - 1) % total_plots
+        new_index = (current_index - 1) % len(all_figures)
     elif direction == 'right':
-        new_index = (current_index + 1) % total_plots
+        new_index = (current_index + 1) % len(all_figures)
     else:
         new_index = current_index
     
     selected_fig = all_figures[new_index]
-    plot_info_text = f"Gráfico {new_index + 1} de {total_plots}"
+    plot_info_text = f"Gráfico {new_index + 1} de {len(all_figures)}"
     
     return selected_fig, plot_info_text, new_index
 
@@ -762,13 +760,15 @@ with gr.Blocks() as demo:
                 download_plot_button = gr.DownloadButton("Descargar Gráfico Actual (PNG)")
                 download_all_plots_button = gr.DownloadButton("Descargar Todos los Gráficos (ZIP)")
             current_index_state = gr.State(0)  # Estado para el índice actual
-    
+
+    # Cargar datos
     load_button.click(
         load_data,
         inputs=[x1_name_input, x2_name_input, x3_name_input, y_name_input, x1_levels_input, x2_levels_input, x3_levels_input, data_input],
         outputs=[data_output, x1_name_input, x2_name_input, x3_name_input, y_name_input, x1_levels_input, x2_levels_input, x3_levels_input, analysis_row]
     )
     
+    # Ajustar modelo y optimizar
     fit_button.click(
         fit_and_optimize_model,
         inputs=[],
@@ -783,75 +783,48 @@ with gr.Blocks() as demo:
             contribution_table_output,
             anova_table_output,
             gr.State(),  # all_figures
-            gr.State()   # current_index (reset)
+            gr.State()   # current_index
         ]
     )
     
+    # Generar y mostrar el primer gráfico
     plot_button.click(
-        lambda: (None, "Gráfico 1 de 9", 0),  # Inicializar plot y índice
-        inputs=[],
+        lambda all_figures: show_plot(all_figures, 0),
+        inputs=[gr.State()],
         outputs=[rsm_plot_output, plot_info, current_index_state]
     )
     
     # Navegación de gráficos
     left_button.click(
-        navigate_plot,
-        inputs=["left", current_index_state, gr.Number(value=9), gr.State()],
+        lambda current_index, all_figures: navigate_plot('left', current_index, all_figures),
+        inputs=[current_index_state, gr.State()],
         outputs=[rsm_plot_output, plot_info, current_index_state]
     )
     right_button.click(
-        navigate_plot,
-        inputs=["right", current_index_state, gr.Number(value=9), gr.State()],
+        lambda current_index, all_figures: navigate_plot('right', current_index, all_figures),
+        inputs=[current_index_state, gr.State()],
         outputs=[rsm_plot_output, plot_info, current_index_state]
     )
     
-    # Función para actualizar y mostrar el gráfico actual
-    def update_plot(direction, current_index, total_plots, all_figures):
-        return navigate_plot(direction, current_index, int(total_plots), all_figures)
-    
     # Descargar gráfico actual
     download_plot_button.click(
         download_current_plot,
         inputs=[gr.State(), current_index_state],
-        outputs=[download_plot_button]
+        outputs=[gr.File()]
     )
     
     # Descargar todos los gráficos en ZIP
     download_all_plots_button.click(
         download_all_plots_zip,
         inputs=[gr.State()],
-        outputs=[download_all_plots_button]
+        outputs=[gr.File()]
     )
     
     # Descargar todas las tablas en Excel
     download_excel_button.click(
         download_all_tables_excel,
         inputs=[],
-        outputs=[download_excel_button]
-    )
-    
-    # Mostrar el primer gráfico después de generar todos
-    def show_first_plot(all_figures):
-        if not all_figures:
-            return None, "No hay gráficos disponibles.", 0
-        return all_figures[0], "Gráfico 1 de 9", 0
-    
-    plot_button.click(
-        lambda: (None, "Gráfico 1 de 9", 0),  # Placeholder para reiniciar el índice
-        inputs=[],
-        outputs=[rsm_plot_output, plot_info, current_index_state]
-    )
-    
-    # Actualizar gráficos al navegar
-    left_button.click(
-        update_plot,
-        inputs=["left", current_index_state, gr.Number(value=9), gr.State()],
-        outputs=[rsm_plot_output, plot_info, current_index_state]
-    )
-    right_button.click(
-        update_plot,
-        inputs=["right", current_index_state, gr.Number(value=9), gr.State()],
-        outputs=[rsm_plot_output, plot_info, current_index_state]
+        outputs=[gr.File()]
     )
     
     # Ejemplo de uso