Update app.py

app.py

@@ -95,7 +95,7 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, palette_puntos, estilo
     sns.set(style="whitegrid")
     plt.rcParams.update({'figure.autolayout': True})
 
-    fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(14, 6))
+    fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(10, 5))  # Tamaño ajustado a (10, 5)
 
     # Obtener colores de las paletas
     colors_puntos = sns.color_palette(palette_puntos, as_cmap=False)
@@ -267,6 +267,36 @@ Fecha: {datetime.now().strftime('%d/%m/%Y %H:%M')}
 """
     return informe, evaluacion['estado']
 
+def generar_graficos_actualizados(df_valid, n_replicas, unidad_medida,
+                                 palette_puntos, estilo_puntos,
+                                 palette_linea_ajuste, estilo_linea_ajuste,
+                                 palette_linea_ideal, estilo_linea_ideal,
+                                 palette_barras_error,
+                                 mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
+    # Función separada para evitar redundancia
+    return generar_graficos(df_valid, n_replicas, unidad_medida,
+                            palette_puntos, estilo_puntos,
+                            palette_linea_ajuste, estilo_linea_ajuste,
+                            palette_linea_ideal, estilo_linea_ideal,
+                            palette_barras_error,
+                            mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos)
+
+def evaluar_puntos_seleccionados(df, puntos_seleccionados):
+    if not puntos_seleccionados:
+        return df  # Si no se selecciona ningún punto, retornar el DataFrame completo
+
+    # Filtrar el DataFrame para incluir solo los puntos seleccionados
+    df_filtrado = df.iloc[puntos_seleccionados].reset_index(drop=True)
+    return df_filtrado
+
+def generar_lista_puntos(df):
+    """Genera una lista de descripciones de puntos para el CheckboxGroup"""
+    lista = []
+    for idx, row in df.iterrows():
+        descripcion = f"Punto {idx + 1}: Predicha={row['Concentración Predicha (UFC)']}, Real Promedio={row['Concentración Real Promedio (UFC)']}"
+        lista.append(descripcion)
+    return lista
+
 def actualizar_analisis(df, n_replicas, unidad_medida):
     if df is None or df.empty:
         return "Error en los datos", None, "No se pueden generar análisis", df
@@ -312,7 +342,7 @@ def actualizar_graficos(df, n_replicas, unidad_medida,
     if df is None or df.empty:
         return None
 
-    # Asegurarse de que los cálculos estén actualizados
+    # Calcular promedio y desviación estándar
     df = calcular_promedio_desviacion(df, n_replicas, unidad_medida)
 
     col_predicha_num = "Concentración Predicha Numérica"
@@ -339,6 +369,55 @@ def actualizar_graficos(df, n_replicas, unidad_medida,
 
     return fig
 
+def recalcular(df, n_replicas, unidad_medida, puntos_seleccionados,
+              palette_puntos, estilo_puntos,
+              palette_linea_ajuste, estilo_linea_ajuste,
+              palette_linea_ideal, estilo_linea_ideal,
+              palette_barras_error,
+              mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
+    if df is None or df.empty:
+        return "Error en los datos", None, "No se pueden generar análisis", df, []
+
+    # Calcular promedio y desviación estándar
+    df = calcular_promedio_desviacion(df, n_replicas, unidad_medida)
+
+    # Filtrar los puntos seleccionados
+    df_filtrado = evaluar_puntos_seleccionados(df, puntos_seleccionados)
+
+    col_predicha_num = "Concentración Predicha Numérica"
+    col_real_promedio = f"Concentración Real Promedio ({unidad_medida})"
+
+    # Convertir columnas a numérico
+    df_filtrado[col_predicha_num] = pd.to_numeric(df_filtrado[col_predicha_num], errors='coerce')
+    df_filtrado[col_real_promedio] = pd.to_numeric(df_filtrado[col_real_promedio], errors='coerce')
+
+    df_valid = df_filtrado.dropna(subset=[col_predicha_num, col_real_promedio])
+
+    if len(df_valid) < 2:
+        return "Se necesitan más datos", None, "Se requieren al menos dos valores reales para el análisis", df, []
+
+    # Calcular la regresión y agregar 'Ajuste Lineal'
+    slope, intercept, r_value, p_value, std_err = stats.linregress(df_valid[col_predicha_num], df_valid[col_real_promedio])
+    df_valid['Ajuste Lineal'] = intercept + slope * df_valid[col_predicha_num]
+
+    # Generar gráfico con opciones seleccionadas
+    fig = generar_graficos(
+        df_valid, n_replicas, unidad_medida,
+        palette_puntos, estilo_puntos,
+        palette_linea_ajuste, estilo_linea_ajuste,
+        palette_linea_ideal, estilo_linea_ideal,
+        palette_barras_error,
+        mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
+    )
+
+    # Generar informe
+    informe, estado = generar_informe_completo(df_valid, n_replicas, unidad_medida)
+
+    # Generar lista de puntos para selección
+    lista_puntos = generar_lista_puntos(df)
+
+    return estado, fig, informe, df, lista_puntos
+
 def exportar_informe_word(df_valid, informe_md, unidad_medida):
     # Crear documento Word
     doc = docx.Document()
@@ -593,9 +672,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
 
     with gr.Tab("📊 Análisis y Reporte"):
         estado_output = gr.Textbox(label="Estado", interactive=False)
-        graficos_output = gr.Plot(label="Gráficos de Análisis")
-
-        # Opciones y botones debajo del gráfico
+        with gr.Row():
+            with gr.Column(scale=3):
+                graficos_output = gr.Plot(label="Gráficos de Análisis")
+                with gr.Row():
+                    # Opciones y botones debajo del gráfico
+                    graficar_btn = gr.Button("📊 Graficar", variant="primary")
+            with gr.Column(scale=1):
+                puntos_seleccionados = gr.CheckboxGroup(
+                    choices=[],
+                    label="Seleccionar Puntos para Ajustar",
+                )
+                recalcular_btn = gr.Button("🔁 Recalcular", variant="primary")
+
+        # Opciones de personalización de gráficos
         with gr.Row():
             # Paletas de colores disponibles en Seaborn
             paletas_colores = ["deep", "muted", "pastel", "bright", "dark", "colorblind", "Set1", "Set2", "Set3"]
@@ -640,7 +730,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
             mostrar_linea_ajuste = gr.Checkbox(value=True, label="Mostrar Línea de Ajuste")
             mostrar_linea_ideal = gr.Checkbox(value=True, label="Mostrar Línea Ideal")
             mostrar_puntos = gr.Checkbox(value=True, label="Mostrar Puntos")
-            graficar_btn = gr.Button("📊 Graficar", variant="primary")
 
         with gr.Row():
             copiar_btn = gr.Button("📋 Copiar Informe", variant="secondary")
@@ -654,148 +743,165 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
         # Informe al final
         informe_output = gr.Markdown(elem_id="informe_output")
 
-    # Eventos
-    input_components = [tabla_output]
-    output_components = [estado_output, graficos_output, informe_output, tabla_output]
+        # Eventos
+        input_components = [tabla_output]
+        output_components = [estado_output, graficos_output, informe_output, tabla_output]
 
-    # Evento al presionar el botón Calcular
-    calcular_btn.click(
-        fn=actualizar_analisis,
-        inputs=[tabla_output, replicas_slider, unidad_input],
-        outputs=output_components
-    )
+        # Evento al presionar el botón Calcular
+        calcular_btn.click(
+            fn=actualizar_analisis,
+            inputs=[tabla_output, replicas_slider, unidad_input],
+            outputs=[estado_output, graficos_output, informe_output, tabla_output]
+        )
 
-    # Evento para graficar con opciones seleccionadas
-    graficar_btn.click(
-        fn=actualizar_graficos,
-        inputs=[
-            tabla_output, replicas_slider, unidad_input,
-            palette_puntos_dropdown, estilo_puntos_dropdown,
-            palette_linea_ajuste_dropdown, estilo_linea_ajuste_dropdown,
-            palette_linea_ideal_dropdown, estilo_linea_ideal_dropdown,
-            palette_barras_error_dropdown,
-            mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
-        ],
-        outputs=graficos_output
-    )
+        # Evento para graficar con opciones seleccionadas
+        graficar_btn.click(
+            fn=actualizar_graficos,
+            inputs=[
+                tabla_output, replicas_slider, unidad_input,
+                palette_puntos_dropdown, estilo_puntos_dropdown,
+                palette_linea_ajuste_dropdown, estilo_linea_ajuste_dropdown,
+                palette_linea_ideal_dropdown, estilo_linea_ideal_dropdown,
+                palette_barras_error_dropdown,
+                mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
+            ],
+            outputs=graficos_output
+        )
 
-    # Evento para limpiar datos
-    limpiar_btn.click(
-        fn=limpiar_datos,
-        inputs=[replicas_slider],
-        outputs=[concentracion_input, unidad_input, filas_slider, tabla_output, estado_output, graficos_output, informe_output]
-    )
+        # Evento para recalcular basado en puntos seleccionados
+        recalcular_btn.click(
+            fn=recalcular,
+            inputs=[
+                tabla_output, replicas_slider, unidad_input, 
+                puntos_seleccionados,
+                palette_puntos_dropdown, estilo_puntos_dropdown,
+                palette_linea_ajuste_dropdown, estilo_linea_ajuste_dropdown,
+                palette_linea_ideal_dropdown, estilo_linea_ideal_dropdown,
+                palette_barras_error_dropdown,
+                mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
+            ],
+            outputs=[estado_output, graficos_output, informe_output, tabla_output, puntos_seleccionados]
+        )
 
-    # Eventos de los botones de ejemplo
-    ejemplo_ufc_btn.click(
-        fn=cargar_ejemplo_ufc,
-        inputs=[replicas_slider],
-        outputs=[concentracion_input, unidad_input, filas_slider, tabla_output]
-    )
+        # Evento para limpiar datos
+        limpiar_btn.click(
+            fn=limpiar_datos,
+            inputs=[replicas_slider],
+            outputs=[concentracion_input, unidad_input, filas_slider, tabla_output, estado_output, graficos_output, informe_output]
+        )
 
-    ejemplo_od_btn.click(
-        fn=cargar_ejemplo_od,
-        inputs=[replicas_slider],
-        outputs=[concentracion_input, unidad_input, filas_slider, tabla_output]
-    )
+        # Eventos de los botones de ejemplo
+        ejemplo_ufc_btn.click(
+            fn=cargar_ejemplo_ufc,
+            inputs=[replicas_slider],
+            outputs=[concentracion_input, unidad_input, filas_slider, tabla_output]
+        )
 
-    # Evento para generar datos sintéticos
-    sinteticos_btn.click(
-        fn=generar_datos_sinteticos_evento,
-        inputs=[tabla_output, replicas_slider, unidad_input],
-        outputs=tabla_output
-    )
+        ejemplo_od_btn.click(
+            fn=cargar_ejemplo_od,
+            inputs=[replicas_slider],
+            outputs=[concentracion_input, unidad_input, filas_slider, tabla_output]
+        )
 
-    # Evento al presionar el botón Ajustar Decimales
-    ajustar_decimales_btn.click(
-        fn=ajustar_decimales_evento,
-        inputs=[tabla_output, decimales_slider],
-        outputs=tabla_output
-    )
+        # Evento para generar datos sintéticos
+        sinteticos_btn.click(
+            fn=generar_datos_sinteticos_evento,
+            inputs=[tabla_output, replicas_slider, unidad_input],
+            outputs=tabla_output
+        )
 
-    # Actualizar tabla al cambiar los parámetros (sin borrar "Concentración Real")
-    def actualizar_tabla_wrapper(df, filas, conc, unidad, replicas):
-        return actualizar_tabla_evento(df, filas, conc, unidad, replicas)
+        # Evento al presionar el botón Ajustar Decimales
+        ajustar_decimales_btn.click(
+            fn=ajustar_decimales_evento,
+            inputs=[tabla_output, decimales_slider],
+            outputs=tabla_output
+        )
 
-    concentracion_input.change(
-        fn=actualizar_tabla_wrapper,
-        inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
-        outputs=tabla_output
-    )
+        # Actualizar tabla al cambiar los parámetros (sin borrar "Concentración Real")
+        def actualizar_tabla_wrapper(df, filas, conc, unidad, replicas):
+            return actualizar_tabla_evento(df, filas, conc, unidad, replicas)
 
-    unidad_input.change(
-        fn=actualizar_tabla_wrapper,
-        inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
-        outputs=tabla_output
-    )
+        concentracion_input.change(
+            fn=actualizar_tabla_wrapper,
+            inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
+            outputs=tabla_output
+        )
 
-    filas_slider.change(
-        fn=actualizar_tabla_wrapper,
-        inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
-        outputs=tabla_output
-    )
+        unidad_input.change(
+            fn=actualizar_tabla_wrapper,
+            inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
+            outputs=tabla_output
+        )
 
-    replicas_slider.change(
-        fn=actualizar_tabla_wrapper,
-        inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
-        outputs=tabla_output
-    )
+        filas_slider.change(
+            fn=actualizar_tabla_wrapper,
+            inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
+            outputs=tabla_output
+        )
 
-    # No agregamos un evento para decimales_slider.change, para evitar borrar la columna "Concentración Real"
-
-    # Evento de copiar informe utilizando JavaScript
-    copiar_btn.click(
-        None,
-        [],
-        [],
-        js="""
-        function() {
-            const informeElement = document.querySelector('#informe_output');
-            const range = document.createRange();
-            range.selectNode(informeElement);
-            window.getSelection().removeAllRanges();
-            window.getSelection().addRange(range);
-            document.execCommand('copy');
-            window.getSelection().removeAllRanges();
-            alert('Informe copiado al portapapeles');
-        }
-        """
-    )
+        replicas_slider.change(
+            fn=actualizar_tabla_wrapper,
+            inputs=[tabla_output, filas_slider, concentracion_input, unidad_input, replicas_slider],
+            outputs=tabla_output
+        )
 
-    # Eventos de exportar informes
-    exportar_word_btn.click(
-        fn=exportar_word,
-        inputs=[tabla_output, informe_output, unidad_input],
-        outputs=exportar_word_file
-    )
+        # No agregamos un evento para decimales_slider.change, para evitar borrar la columna "Concentración Real"
+
+        # Evento de copiar informe utilizando JavaScript
+        copiar_btn.click(
+            None,
+            [],
+            [],
+            js="""
+            function() {
+                const informeElement = document.querySelector('#informe_output');
+                const range = document.createRange();
+                range.selectNode(informeElement);
+                window.getSelection().removeAllRanges();
+                window.getSelection().addRange(range);
+                document.execCommand('copy');
+                window.getSelection().removeAllRanges();
+                alert('Informe copiado al portapapeles');
+            }
+            """
+        )
 
-    exportar_latex_btn.click(
-        fn=exportar_latex,
-        inputs=[tabla_output, informe_output],
-        outputs=exportar_latex_file
-    )
+        # Eventos de exportar informes
+        exportar_word_btn.click(
+            fn=exportar_word,
+            inputs=[tabla_output, informe_output, unidad_input],
+            outputs=exportar_word_file
+        )
 
-    # Inicializar la interfaz con el ejemplo base
-    def iniciar_con_ejemplo():
-        n_replicas = 1
-        df = generar_tabla(7, 2000000, "UFC", n_replicas)
-        # Valores reales de ejemplo
-        df[f"Concentración Real 1 (UFC)"] = [2000000, 1600000, 1200000, 800000, 400000, 200000, 100000]
-        estado, fig, informe, df = actualizar_analisis(df, n_replicas, "UFC")
-        return (
-            2000000,
-            "UFC",
-            7,
-            df,
-            estado,
-            fig,
-            informe
+        exportar_latex_btn.click(
+            fn=exportar_latex,
+            inputs=[tabla_output, informe_output],
+            outputs=exportar_latex_file
         )
 
-    interfaz.load(
-        fn=iniciar_con_ejemplo,
-        outputs=[concentracion_input, unidad_input, filas_slider, tabla_output, estado_output, graficos_output, informe_output]
-    )
+        # Inicializar la interfaz con el ejemplo base
+        def iniciar_con_ejemplo():
+            n_replicas = 1
+            df = generar_tabla(7, 2000000, "UFC", n_replicas)
+            # Valores reales de ejemplo
+            df[f"Concentración Real 1 (UFC)"] = [2000000, 1600000, 1200000, 800000, 400000, 200000, 100000]
+            estado, fig, informe, df = actualizar_analisis(df, n_replicas, "UFC")
+            lista_puntos = generar_lista_puntos(df)
+            return (
+                2000000,
+                "UFC",
+                7,
+                df,
+                estado,
+                fig,
+                informe,
+                lista_puntos
+            )
+
+        interfaz.load(
+            fn=iniciar_con_ejemplo,
+            outputs=[concentracion_input, unidad_input, filas_slider, tabla_output, estado_output, graficos_output, informe_output, puntos_seleccionados]
+        )
 
 # Lanzar la interfaz
 if __name__ == "__main__":