Update app.py

app.py

@@ -74,7 +74,10 @@ def calcular_promedio_desviacion(df, n_replicas, unidad_medida):
 
     return df
 
-def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
+def generar_graficos(df_valid, n_replicas, unidad_medida, color_puntos, estilo_puntos,
+                     color_linea_ajuste, estilo_linea_ajuste,
+                     color_linea_ideal, estilo_linea_ideal,
+                     color_barras_error, mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
     col_predicha_num = "Concentración Predicha Numérica"
     col_real_promedio = f"Concentración Real Promedio ({unidad_medida})"
     col_desviacion = f"Desviación Estándar ({unidad_medida})"
@@ -92,16 +95,6 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
     sns.set(style="whitegrid")
     plt.rcParams.update({'figure.autolayout': True})
 
-    # Definir estilos de gráficos
-    estilos = {
-        "Estilo 1": {"color_puntos": "blue", "color_linea": "green", "color_error": "lightgray", "marker": "o", "line_style": "-"},
-        "Estilo 2": {"color_puntos": "red", "color_linea": "orange", "color_error": "pink", "marker": "s", "line_style": "--"},
-        "Estilo 3": {"color_puntos": "purple", "color_linea": "cyan", "color_error": "gray", "marker": "^", "line_style": "-."},
-        "Estilo 4": {"color_puntos": "black", "color_linea": "yellow", "color_error": "darkgray", "marker": "D", "line_style": ":"}
-    }
-
-    estilo = estilos.get(estilo_grafico, estilos["Estilo 1"])
-
     fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(14, 6))
 
     # Gráfico de dispersión con línea de regresión
@@ -112,9 +105,9 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
                 df_valid[col_predicha_num],
                 df_valid[col_real_promedio],
                 yerr=df_valid[col_desviacion],
-                fmt=estilo["marker"],
-                color=estilo["color_puntos"],
-                ecolor=estilo["color_error"],
+                fmt=estilo_puntos,
+                color=color_puntos,
+                ecolor=color_barras_error,
                 elinewidth=2,
                 capsize=3,
                 label='Datos Reales'
@@ -123,10 +116,10 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
             ax1.scatter(
                 df_valid[col_predicha_num],
                 df_valid[col_real_promedio],
-                color=estilo["color_puntos"],
+                color=color_puntos,
                 s=100,
                 label='Datos Reales',
-                marker=estilo["marker"]
+                marker=estilo_puntos
             )
 
     # Línea de ajuste
@@ -134,10 +127,10 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
         ax1.plot(
             df_valid[col_predicha_num],
             df_valid['Ajuste Lineal'],
-            color=estilo["color_linea"],
+            color=color_linea_ajuste,
             label='Ajuste Lineal',
             linewidth=2,
-            linestyle=estilo["line_style"]
+            linestyle=estilo_linea_ajuste
         )
 
     # Línea ideal
@@ -147,8 +140,8 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
         ax1.plot(
             [min_predicha, max_predicha],
             [min_predicha, max_predicha],
-            color='red',
-            linestyle='--',
+            color=color_linea_ideal,
+            linestyle=estilo_linea_ideal,
             label='Ideal'
         )
 
@@ -174,9 +167,9 @@ def generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostra
     ax2.scatter(
         df_valid[col_predicha_num],
         residuos,
-        color=estilo["color_puntos"],
+        color=color_puntos,
         s=100,
-        marker=estilo["marker"],
+        marker=estilo_puntos,
         label='Residuos'
     )
 
@@ -286,12 +279,24 @@ def actualizar_analisis(df, n_replicas, unidad_medida):
     df_valid['Ajuste Lineal'] = intercept + slope * df_valid[col_predicha_num]
 
     # Generar gráfico con opciones predeterminadas
-    fig = generar_graficos(df_valid, n_replicas, unidad_medida, "Estilo 1", True, True, True)
+    fig = generar_graficos(
+        df_valid, n_replicas, unidad_medida,
+        color_puntos='blue', estilo_puntos='o',
+        color_linea_ajuste='green', estilo_linea_ajuste='-',
+        color_linea_ideal='red', estilo_linea_ideal='--',
+        color_barras_error='lightgray',
+        mostrar_linea_ajuste=True, mostrar_linea_ideal=True, mostrar_puntos=True
+    )
     informe, estado = generar_informe_completo(df_valid, n_replicas, unidad_medida)
 
     return estado, fig, informe, df
 
-def actualizar_graficos(df, n_replicas, unidad_medida, estilo_grafico, mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
+def actualizar_graficos(df, n_replicas, unidad_medida,
+                        color_puntos, estilo_puntos,
+                        color_linea_ajuste, estilo_linea_ajuste,
+                        color_linea_ideal, estilo_linea_ideal,
+                        color_barras_error,
+                        mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos):
     if df is None or df.empty:
         return None
 
@@ -311,7 +316,14 @@ def actualizar_graficos(df, n_replicas, unidad_medida, estilo_grafico, mostrar_l
         return None
 
     # Generar gráfico con opciones seleccionadas
-    fig = generar_graficos(df_valid, n_replicas, unidad_medida, estilo_grafico, mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos)
+    fig = generar_graficos(
+        df_valid, n_replicas, unidad_medida,
+        color_puntos, estilo_puntos,
+        color_linea_ajuste, estilo_linea_ajuste,
+        color_linea_ideal, estilo_linea_ideal,
+        color_barras_error,
+        mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
+    )
 
     return fig
 
@@ -573,10 +585,41 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
 
         # Opciones y botones debajo del gráfico
         with gr.Row():
-            estilo_grafico_dropdown = gr.Dropdown(
-                choices=["Estilo 1", "Estilo 2", "Estilo 3", "Estilo 4"],
-                value="Estilo 1",
-                label="Estilo del Gráfico"
+            color_puntos_dropdown = gr.Dropdown(
+                choices=["blue", "red", "green", "purple", "orange", "black"],
+                value="blue",
+                label="Color de Puntos"
+            )
+            estilo_puntos_dropdown = gr.Dropdown(
+                choices=["o", "s", "^", "D", "v", "<", ">", "h", "H", "p", "*", "X", "d"],
+                value="o",
+                label="Estilo de Puntos"
+            )
+            color_linea_ajuste_dropdown = gr.Dropdown(
+                choices=["green", "blue", "red", "purple", "orange", "black"],
+                value="green",
+                label="Color Línea de Ajuste"
+            )
+            estilo_linea_ajuste_dropdown = gr.Dropdown(
+                choices=["-", "--", "-.", ":"],
+                value="-",
+                label="Estilo Línea de Ajuste"
+            )
+        with gr.Row():
+            color_linea_ideal_dropdown = gr.Dropdown(
+                choices=["red", "green", "blue", "purple", "orange", "black"],
+                value="red",
+                label="Color Línea Ideal"
+            )
+            estilo_linea_ideal_dropdown = gr.Dropdown(
+                choices=["--", "-", "-.", ":"],
+                value="--",
+                label="Estilo Línea Ideal"
+            )
+            color_barras_error_dropdown = gr.Dropdown(
+                choices=["lightgray", "gray", "darkgray", "pink", "lightblue", "lightgreen"],
+                value="lightgray",
+                label="Color Barras de Error"
             )
             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")
@@ -609,7 +652,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as interfaz:
     # Evento para graficar con opciones seleccionadas
     graficar_btn.click(
         fn=actualizar_graficos,
-        inputs=[tabla_output, replicas_slider, unidad_input, estilo_grafico_dropdown, mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos],
+        inputs=[
+            tabla_output, replicas_slider, unidad_input,
+            color_puntos_dropdown, estilo_puntos_dropdown,
+            color_linea_ajuste_dropdown, estilo_linea_ajuste_dropdown,
+            color_linea_ideal_dropdown, estilo_linea_ideal_dropdown,
+            color_barras_error_dropdown,
+            mostrar_linea_ajuste, mostrar_linea_ideal, mostrar_puntos
+        ],
         outputs=graficos_output
     )