Update app.py

app.py

@@ -40,6 +40,7 @@ class RSM_ExperimentalDesign:
         self.optimized_results = None
         self.optimal_levels = None
         self.all_figures = []
+        self.selected_factors = []
 
     def generate_design(self):
         """
@@ -96,6 +97,7 @@ class RSM_ExperimentalDesign:
         Args:
             selected_factors (list): Lista de factores a incluir en el modelo.
         """
+        self.selected_factors = selected_factors
         terms = selected_factors.copy()
         # Términos cuadráticos
         terms += [f'I({var}**2)' for var in selected_factors]
@@ -263,9 +265,6 @@ class RSM_ExperimentalDesign:
         self.all_figures = []  # Resetear la lista de figuras
 
         # Niveles naturales para graficar
-        levels_to_plot_natural = self.factor_levels
-
-        # Generar y almacenar gráficos individuales
         for fixed_variable in self.selected_factors:
             for level in self.factor_levels[fixed_variable]:
                 fig = self.plot_rsm_individual(fixed_variable, level)
@@ -382,7 +381,7 @@ class RSM_ExperimentalDesign:
                 f_stat = ms_factor / ms_error
                 p_value = f.sf(f_stat, df_factor, anova_table.loc['Residual', 'df'])
                 contribution_percentage = (ss_factor / ss_total) * 100
-    
+
                 contribution_table = pd.concat([contribution_table, pd.DataFrame({
                     'Fuente de Variación': [factor_name],
                     'Suma de Cuadrados': [ss_factor],
@@ -512,19 +511,6 @@ class RSM_ExperimentalDesign:
         """
         return fig.to_image(format="png")
 
-    def save_all_figures_png(self):
-        """
-        Guarda todas las figuras en archivos PNG temporales y retorna las rutas.
-        """
-        png_paths = []
-        for idx, fig in enumerate(self.all_figures, start=1):
-            img_bytes = fig.to_image(format="png")
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
-                temp_file.write(img_bytes)
-                temp_path = temp_file.name
-                png_paths.append(temp_path)
-        return png_paths
-
     def save_tables_to_excel(self):
         """
         Guarda todas las tablas en un archivo Excel con múltiples hojas y retorna la ruta del archivo.
@@ -705,12 +691,7 @@ def download_current_plot(all_figures, current_index):
     img_bytes = rsm.save_fig_to_bytes(fig)
     filename = f"Grafico_RSM_{current_index + 1}.png"
 
-    # Crear un archivo temporal
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
-        temp_file.write(img_bytes)
-        temp_path = temp_file.name
-
-    return temp_path  # Retornar solo la ruta
+    return (filename, img_bytes)
 
 def download_all_plots_zip():
     """
@@ -720,7 +701,10 @@ def download_all_plots_zip():
         return None
     zip_path = rsm.save_figures_to_zip()
     if zip_path:
-        return zip_path
+        with open(zip_path, 'rb') as f:
+            zip_bytes = f.read()
+        filename = f"Graficos_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.zip"
+        return (filename, zip_bytes)
     return None
 
 def download_all_tables_excel():
@@ -731,18 +715,24 @@ def download_all_tables_excel():
         return None
     excel_path = rsm.save_tables_to_excel()
     if excel_path:
-        return excel_path
+        with open(excel_path, 'rb') as f:
+            excel_bytes = f.read()
+        filename = f"Tablas_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
+        return (filename, excel_bytes)
     return None
 
 def exportar_word(tables_dict):
     """
     Función para exportar las tablas a un documento de Word.
     """
-    if 'rsm' not in globals():
+    if not tables_dict:
         return None
     word_path = rsm.export_tables_to_word(tables_dict)
     if word_path and os.path.exists(word_path):
-        return word_path
+        with open(word_path, 'rb') as f:
+            word_bytes = f.read()
+        filename = f"Tablas_RSM_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
+        return (filename, word_bytes)
     return None
 
 # --- Crear la interfaz de Gradio ---
@@ -817,8 +807,8 @@ def create_gradio_interface():
                 gr.Markdown("**Tabla ANOVA Detallada**")
                 anova_table_output = gr.Dataframe(label="Tabla ANOVA Detallada", interactive=False)
                 gr.Markdown("## Descargar Todas las Tablas")
-                download_excel_button = gr.File(label="Descargar Tablas en Excel", visible=False)
-                download_word_button = gr.File(label="Descargar Tablas en Word", visible=False)
+                download_excel_button = gr.DownloadButton(label="Descargar Tablas en Excel")
+                download_word_button = gr.DownloadButton(label="Descargar Tablas en Word")
 
             with gr.Column():
                 gr.Markdown("## Selección de Factores para el Modelo Simplificado")
@@ -828,8 +818,6 @@ def create_gradio_interface():
                     value=[]
                 )
                 fit_button_2 = gr.Button("Aplicar Selección de Factores")
-                gr.Markdown("## Resultados de Optimización")
-                optimization_table_output_2 = gr.Dataframe(label="Tabla de Optimización", interactive=False)
                 gr.Markdown("## Generar Gráficos de Superficie de Respuesta")
                 plot_button = gr.Button("Generar Gráficos")
                 with gr.Row():
@@ -838,14 +826,23 @@ def create_gradio_interface():
                 rsm_plot_output = gr.Plot()
                 plot_info = gr.Textbox(label="Información del Gráfico", value="Gráfico 1 de 0", interactive=False)
                 with gr.Row():
-                    download_plot_button = gr.File(label="Descargar Gráfico Actual (PNG)", visible=False)
-                    download_all_plots_button = gr.File(label="Descargar Todos los Gráficos (ZIP)", visible=False)
+                    download_plot_button = gr.DownloadButton(label="Descargar Gráfico Actual (PNG)")
+                    download_all_plots_button = gr.DownloadButton(label="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
 
         # Funciones de carga y ajuste
+        def load_data_interface(design_type, factor_names, factor_levels, y_name, data_str):
+            data, analysis_visibility, factor_names_list = load_data(design_type, factor_names, factor_levels, y_name, data_str)
+            if data is not None:
+                # Actualizar las opciones de selección de factores
+                selected_factors = factor_names_list
+                return data, analysis_visibility, selected_factors
+            else:
+                return data, analysis_visibility, []
+
         load_button.click(
-            load_data,
+            load_data_interface,
             inputs=[design_type_input, factor_names_input, factor_levels_input, y_name_input, data_input],
             outputs=[data_output, analysis_row, selected_factors_input]
         )
@@ -869,40 +866,55 @@ def create_gradio_interface():
             ]
         )
 
-        # Descargar todas las tablas en Excel y Word
+        # Conectar los botones de descarga de tablas
         download_excel_button.click(
-            fn=lambda: download_all_tables_excel(),
+            fn=download_all_tables_excel,
             inputs=[],
             outputs=download_excel_button
         )
-        
+
         download_word_button.click(
-            fn=lambda: exportar_word(rsm.get_all_tables()),
-            inputs=[],
+            fn=exportar_word,
+            inputs=[gr.State()],
             outputs=download_word_button
         )
 
         # Generar y mostrar los gráficos
+        def generate_plots():
+            if 'rsm' not in globals() or not rsm.all_figures:
+                return None, "No hay gráficos disponibles.", 0, []
+            # Mostrar el primer gráfico
+            return rsm.all_figures[0], f"Gráfico 1 de {len(rsm.all_figures)}", 0, rsm.all_figures
+
         plot_button.click(
-            lambda: (
-                None,  # Placeholder, se actualizará después
-                "No hay gráficos disponibles.",
-                0,
-                []
-            ),
+            generate_plots,
             inputs=[],
             outputs=[rsm_plot_output, plot_info, current_index_state, all_figures_state]
         )
 
         # Navegación de gráficos
+        def navigate(direction, current_index, all_figures):
+            if not all_figures:
+                return None, "No hay gráficos disponibles.", current_index
+            if direction == 'left':
+                new_index = (current_index - 1) % len(all_figures)
+            elif direction == 'right':
+                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 {len(all_figures)}"
+            return selected_fig, plot_info_text, new_index
+
         left_button.click(
-            lambda current_index, all_figures: navigate_plot('left', current_index, all_figures),
-            inputs=[current_index_state, all_figures_state],
+            navigate,
+            inputs=[gr.Button("left"), current_index_state, all_figures_state],
             outputs=[rsm_plot_output, plot_info, current_index_state]
         )
+
         right_button.click(
-            lambda current_index, all_figures: navigate_plot('right', current_index, all_figures),
-            inputs=[current_index_state, all_figures_state],
+            navigate,
+            inputs=[gr.Button("right"), current_index_state, all_figures_state],
             outputs=[rsm_plot_output, plot_info, current_index_state]
         )
 
@@ -926,95 +938,25 @@ def create_gradio_interface():
         1. **Configura el Diseño:**
             - Selecciona el tipo de diseño (Box-Behnken o Central Compuesto).
             - Ingresa los nombres de los factores separados por comas.
-            - Ingresa los niveles de cada factor separados por comas y cada conjunto de niveles por ';'.
+            - Ingresa los niveles para cada factor separados por comas y cada conjunto de niveles por ';'.
             - Especifica el nombre de la variable dependiente.
             - Proporciona los datos del experimento en formato CSV.
         2. **Cargar Datos:**
             - Haz clic en 'Cargar Datos' para cargar y visualizar los datos.
-        3. **Ajustar Modelo y Optimizar:**
-            - Selecciona los factores que deseas incluir en el modelo simplificado.
+        3. **Seleccionar Factores para el Modelo Simplificado:**
+            - Marca los factores que deseas incluir en el modelo simplificado.
+        4. **Ajustar Modelo y Optimizar:**
             - Haz clic en 'Ajustar Modelo y Optimizar' para ajustar los modelos y obtener los resultados.
-        4. **Generar Gráficos:**
+        5. **Generar y Navegar Gráficos:**
             - Haz clic en 'Generar Gráficos' para crear las superficies de respuesta.
             - Navega entre los gráficos usando los botones '<' y '>'.
-            - Descarga el gráfico actual en PNG o descarga todos los gráficos en un ZIP.
-        5. **Descargar Tablas:**
+            - Descarga el gráfico actual en PNG o todos los gráficos en un archivo ZIP.
+        6. **Descargar Tablas:**
             - Descarga todas las tablas generadas en un archivo Excel o Word.
         """)
 
     return demo
 
-def navigate_plot(direction, current_index, all_figures):
-    """
-    Navega entre los gráficos.
-    """
-    if not all_figures:
-        return None, "No hay gráficos disponibles.", current_index
-
-    if direction == 'left':
-        new_index = (current_index - 1) % len(all_figures)
-    elif direction == 'right':
-        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 {len(all_figures)}"
-
-    return selected_fig, plot_info_text, new_index
-
-# --- Funciones de descarga y exportación ---
-
-def download_current_plot(all_figures, current_index):
-    """
-    Descarga la figura actual como PNG.
-    """
-    if not all_figures:
-        return None
-    fig = all_figures[current_index]
-    img_bytes = rsm.save_fig_to_bytes(fig)
-    filename = f"Grafico_RSM_{current_index + 1}.png"
-
-    # Crear un archivo temporal
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as temp_file:
-        temp_file.write(img_bytes)
-        temp_path = temp_file.name
-
-    return temp_path  # Retornar solo la ruta
-
-def download_all_plots_zip():
-    """
-    Descarga todas las figuras en un archivo ZIP.
-    """
-    if 'rsm' not in globals():
-        return None
-    zip_path = rsm.save_figures_to_zip()
-    if zip_path:
-        return zip_path
-    return None
-
-def download_all_tables_excel():
-    """
-    Descarga todas las tablas en un archivo Excel con múltiples hojas.
-    """
-    if 'rsm' not in globals():
-        return None
-    excel_path = rsm.save_tables_to_excel()
-    if excel_path:
-        return excel_path
-    return None
-
-def exportar_word(tables_dict):
-    """
-    Función para exportar las tablas a un documento de Word.
-    """
-    if not tables_dict:
-        return None
-    word_path = rsm.export_tables_to_word(tables_dict)
-    if word_path and os.path.exists(word_path):
-        return word_path
-    return None
-
 # --- Función Principal ---
 
 def main():