Update app.py

app.py

@@ -431,17 +431,17 @@ def download_excel():
     if 'rsm' not in globals():
         return None, "Error: Carga los datos primero."
 
-    output = io.BytesIO()
-    with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+    # Create a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.xlsx')
+    
+    with pd.ExcelWriter(temp_file.name, engine='xlsxwriter') as writer:
         rsm.data.to_excel(writer, sheet_name='Datos', index=False)
         rsm.generate_prediction_table().to_excel(writer, sheet_name='Predicciones', index=False)
         rsm.optimize().to_excel(writer, sheet_name='Optimizacion', index=False)
         rsm.calculate_contribution_percentage().to_excel(writer, sheet_name='Contribucion', index=False)
         rsm.calculate_detailed_anova().to_excel(writer, sheet_name='ANOVA', index=False)
-
-    output.seek(0)
     
-    return output, "Descargar Excel"  # Return BytesIO and label
+    return temp_file.name
 
 def download_images():
     global plot_images
@@ -451,16 +451,37 @@ def download_images():
     if not plot_images:
         return None, "Error: No se han generado gráficos."
 
-    # Crear un archivo zip en memoria
-    zip_buffer = io.BytesIO()
-    with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zipf:
+    # Create a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.zip')
+    
+    with zipfile.ZipFile(temp_file.name, 'w', zipfile.ZIP_DEFLATED) as zipf:
         for i, fig in enumerate(plot_images):
             img_bytes = fig.to_image(format="png")
             zipf.writestr(f"plot_{i}.png", img_bytes)
+    
+    return temp_file.name
+
+def generate_rsm_plot(fixed_variable, fixed_level):
+    global current_plot_index, plot_images
+
+    if 'rsm' not in globals():
+        return None, "Error: Carga los datos primero.", None
+
+    if not plot_images:
+        plot_images = rsm.generate_all_plots()
 
-    zip_buffer.seek(0)
-    return zip_buffer, "Descargar ZIP"  # Return BytesIO and label
+    if not plot_images:
+        return None, "Error: No se pudieron generar los gráficos.", None
 
+    current_plot_index = (current_plot_index) % len(plot_images)
+    fig = plot_images[current_plot_index]
+
+    # Create a temporary file for the image
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+    fig.write_image(temp_file.name)
+    
+    return fig, "", temp_file.name
+    
 def next_plot():
     global current_plot_index
     current_plot_index += 1
@@ -545,12 +566,14 @@ with gr.Blocks() as demo:
     )
 
     fit_button.click(fit_and_optimize_model, outputs=[model_completo_output, pareto_completo_output, model_simplificado_output, pareto_simplificado_output, equation_output, optimization_table_output, prediction_table_output, contribution_table_output, anova_table_output])
+
+    plot_button.click(generate_rsm_plot, 
+                      inputs=[fixed_variable_input, fixed_level_input], 
+                      outputs=[rsm_plot_output, equation_output, plot_image_output])
     
-    plot_button.click(generate_rsm_plot, inputs=[fixed_variable_input, fixed_level_input], outputs=[rsm_plot_output, equation_output, plot_image_output])
-    
-    download_excel_button.click(download_excel, outputs=[download_excel_button, download_excel_button])
+    download_excel_button.click(download_excel, outputs=[download_excel_button])
 
-    download_images_button.click(download_images, outputs=[download_images_button, download_images_button])
+    download_images_button.click(download_images, outputs=[download_images_button])
     
     prev_plot_button.click(prev_plot, outputs=prev_plot_button).then(generate_rsm_plot, inputs=[fixed_variable_input, fixed_level_input], outputs=[rsm_plot_output, equation_output, plot_image_output])