Update app.py

app.py

@@ -111,117 +111,87 @@ else:
 # Procesar valores
 y_values_lists = [[float(i) for i in y_values.split(",") if i.strip()] for y_values in y_values_list]
 
-# Verificar si las listas tienen el mismo tamaño
-if any(len(x) != len(y) for y in y_values_lists):
-    st.error("Todos los valores de X y Y deben tener la misma cantidad de elementos.")
-else:
-    # Crear un DataFrame
-    data = pd.DataFrame({"X": x})
-    for idx, y_set in enumerate(y_values_lists):
-        data[y_names_list[idx]] = y_set
-
-    # Configuración común para todos los gráficos
-    common_layout = dict(
-        xaxis_title=x_label,
-        yaxis_title=y_label,
-        plot_bgcolor="white",
-        hovermode="x unified",
-        width=graph_width,
-        height=graph_height,
-        margin=dict(l=60, r=40, t=100, b=40),
-        xaxis=dict(showgrid=True, zeroline=True, gridcolor='rgba(211,211,211,0.5)', zerolinecolor='rgba(128,128,128,0.5)', autorange=True),
-        yaxis=dict(showgrid=True, zeroline=True, gridcolor='rgba(211,211,211,0.5)', zerolinecolor='rgba(128,128,128,0.5)', autorange=True),
-        font=dict(family=font_family, size=18, color="black"),
-        showlegend=show_legend,
-        legend=dict(orientation="v", yanchor="top", y=1, xanchor="left", x=1.02)
-    )
-    
-    # Personalizar el hovertemplate
-    hovertemplate = '<b>%{y}</b>'
-
-    # Generar el gráfico basado en el tipo seleccionado
-    if chart_type == "Línea":
-        fig = px.line(data, x="X", y=y_names_list, line_shape="spline")
-        fig.update_traces(hovertemplate=hovertemplate)
-        if use_multiple_colors:
-            for i, name in enumerate(y_names_list):
-                fig.update_traces(selector=dict(name=name), line_color=colors[i % len(colors)])
-        else:
-            fig.update_traces(line_color=color)
-    elif chart_type == "Área":
-        fig = px.area(data, x="X", y=y_names_list, line_shape="spline")
-        fig.update_traces(hovertemplate=hovertemplate)
-        if use_multiple_colors:
-            for i, name in enumerate(y_names_list):
-                fig.update_traces(selector=dict(name=name), line_color=colors[i % len(colors)], fillcolor=colors[i % len(colors)])
-        else:
-            fig.update_traces(line_color=color, fillcolor=color)
-    elif chart_type == "Dispersión":
-        fig = px.scatter(data, x="X", y=y_names_list)
-        fig.update_traces(hovertemplate=hovertemplate)
-        if use_multiple_colors:
-            for i, name in enumerate(y_names_list):
-                fig.update_traces(selector=dict(name=name), marker_color=colors[i % len(colors)], marker_line_color=border_colors[i % len(border_colors)], marker_line_width=border_width)
-        else:
-            fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=border_width)
-    elif chart_type == "Barras":
-        if horizontal_bars:
-            fig = px.bar(data, x=y_names_list[0], y="X", orientation='h', barmode='stack' if stacked_bars else 'group')
-        else:
-            fig = px.bar(data, x="X", y=y_names_list, barmode='stack' if stacked_bars else 'group')
-        fig.update_traces(hovertemplate=hovertemplate)
-        if use_multiple_colors:
-            if len(y_names_list) == 1:  # Solo una variable Y
-                fig.update_traces(marker_color=colors[:len(x)], marker_line_color=border_colors[:len(x)], marker_line_width=border_width)
-            else:
-                for i, trace in enumerate(fig.data):
-                    trace.marker.color = colors[i % len(colors)]
-                    trace.marker.line.color = border_colors[i % len(border_colors)]
-                    trace.marker.line.width = border_width
-        else:
-            fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=border_width)
-        fig.update_layout(bargap=0.2)
-    elif chart_type == "Donut":
-        figs = []
-        for i, y_name in enumerate(y_names_list):
-            fig = px.pie(data, values=y_name, names="X", hole=hole_size, color_discrete_sequence=colors[:len(x)])
-            fig.update_traces(hovertemplate='<b>%{label}</b>: %{value} (%{percent})', textinfo='percent+label')
-            fig.update_layout(
-                title=dict(
-                    text=f"{chart_title}<br><sub>{y_name}</sub>" if len(y_names_list) > 1 else chart_title,
-                    x=0.5,
-                    y=0.95,
-                    xanchor='center',
-                    yanchor='top',
-                    font=dict(
-                        family=font_family,
-                        size=18,
-                        color="#374151",
-                        weight="normal"
-                    )
-                ),
-                **common_layout
-            )
-            figs.append(fig)
-        for fig in figs:
-            st.plotly_chart(fig)
-    elif chart_type == "Línea temporal":
-        if horizontal_bars:
-            fig = px.bar(data, x="Y", y="X", orientation='h', barmode='stack')
-        else:
-            fig = px.bar(data, x="X", y="Y", barmode='stack')
-        fig.update_traces(hovertemplate=hovertemplate)
-        if use_multiple_colors:
+# Asegurar que todas las listas de Y tengan el mismo tamaño que X
+for y_values in y_values_lists:
+    while len(y_values) < len(x):
+        y_values.append(0)
+    while len(y_values) > len(x):
+        y_values.pop()
+
+# Crear un DataFrame
+data = pd.DataFrame({"X": x})
+for idx, y_set in enumerate(y_values_lists):
+    data[y_names_list[idx]] = y_set
+
+# Configuración común para todos los gráficos
+common_layout = dict(
+    xaxis_title=x_label,
+    yaxis_title=y_label,
+    plot_bgcolor="white",
+    hovermode="x unified",
+    width=graph_width,
+    height=graph_height,
+    margin=dict(l=60, r=40, t=100, b=40),
+    xaxis=dict(showgrid=True, zeroline=True, gridcolor='rgba(211,211,211,0.5)', zerolinecolor='rgba(128,128,128,0.5)', autorange=True),
+    yaxis=dict(showgrid=True, zeroline=True, gridcolor='rgba(211,211,211,0.5)', zerolinecolor='rgba(128,128,128,0.5)', autorange=True),
+    font=dict(family=font_family, size=18, color="black"),
+    showlegend=show_legend,
+    legend=dict(orientation="v", yanchor="top", y=1, xanchor="left", x=1.02)
+)
+
+# Personalizar el hovertemplate
+hovertemplate = '<b>%{y}</b>'
+
+# Generar el gráfico basado en el tipo seleccionado
+if chart_type == "Línea":
+    fig = px.line(data, x="X", y=y_names_list, line_shape="spline")
+    fig.update_traces(hovertemplate=hovertemplate)
+    if use_multiple_colors:
+        for i, name in enumerate(y_names_list):
+            fig.update_traces(selector=dict(name=name), line_color=colors[i % len(colors)])
+    else:
+        fig.update_traces(line_color=color)
+elif chart_type == "Área":
+    fig = px.area(data, x="X", y=y_names_list, line_shape="spline")
+    fig.update_traces(hovertemplate=hovertemplate)
+    if use_multiple_colors:
+        for i, name in enumerate(y_names_list):
+            fig.update_traces(selector=dict(name=name), line_color=colors[i % len(colors)], fillcolor=colors[i % len(colors)])
+    else:
+        fig.update_traces(line_color=color, fillcolor=color)
+elif chart_type == "Dispersión":
+    fig = px.scatter(data, x="X", y=y_names_list)
+    fig.update_traces(hovertemplate=hovertemplate)
+    if use_multiple_colors:
+        for i, name in enumerate(y_names_list):
+            fig.update_traces(selector=dict(name=name), marker_color=colors[i % len(colors)], marker_line_color=border_colors[i % len(border_colors)], marker_line_width=border_width)
+    else:
+        fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=border_width)
+elif chart_type == "Barras":
+    if horizontal_bars:
+        fig = px.bar(data, x=y_names_list[0], y="X", orientation='h', barmode='stack' if stacked_bars else 'group')
+    else:
+        fig = px.bar(data, x="X", y=y_names_list, barmode='stack' if stacked_bars else 'group')
+    fig.update_traces(hovertemplate=hovertemplate)
+    if use_multiple_colors:
+        if len(y_names_list) == 1:  # Solo una variable Y
             fig.update_traces(marker_color=colors[:len(x)], marker_line_color=border_colors[:len(x)], marker_line_width=border_width)
         else:
-            fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=border_width)
-        fig.update_layout(bargap=0.2)
-
-    # Añadir anotación para el título
-    if chart_type != "Donut":
+            for i, trace in enumerate(fig.data):
+                trace.marker.color = colors[i % len(colors)]
+                trace.marker.line.color = border_colors[i % len(border_colors)]
+                trace.marker.line.width = border_width
+    else:
+        fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=border_width)
+    fig.update_layout(bargap=0.2)
+elif chart_type == "Donut":
+    figs = []
+    for i, y_name in enumerate(y_names_list):
+        fig = px.pie(data, values=y_name, names="X", hole=hole_size, color_discrete_sequence=colors[:len(x)])
+        fig.update_traces(hovertemplate='<b>%{label}</b>: %{value} (%{percent})', textinfo='percent+label')
         fig.update_layout(
             title=dict(
-                text=f"{chart_title}",
+                text=f"{chart_title}<br><sub>{y_name}</sub>" if len(y_names_list) > 1 else chart_title,
                 x=0.5,
                 y=0.95,
                 xanchor='center',
@@ -232,22 +202,44 @@ else:
                     color="#374151",
                     weight="normal"
                 )
-            )
+            ),
+            **common_layout
         )
-        
-        # Aplicar configuración común
-        fig.update_layout(**common_layout)
-        
-        # Aplicar múltiples colores si se seleccionó la opción
-        if use_multiple_colors and chart_type != "Donut":
-            for i, trace in enumerate(fig.data):
-                trace.update(marker_color=colors[i % len(colors)], marker_line_color=border_colors[i % len(border_colors)], marker_line_width=border_width)
-        
-        # Mostrar el gráfico
+        figs.append(fig)
+    for fig in figs:
         st.plotly_chart(fig)
 
-        # Forzar el autoescale
-        fig.update_layout(xaxis_autorange=True, yaxis_autorange=True)
+# Añadir anotación para el título
+if chart_type != "Donut":
+    fig.update_layout(
+        title=dict(
+            text=f"{chart_title}",
+            x=0.5,
+            y=0.95,
+            xanchor='center',
+            yanchor='top',
+            font=dict(
+                family=font_family,
+                size=18,
+                color="#374151",
+                weight="normal"
+            )
+        )
+    )
+    
+    # Aplicar configuración común
+    fig.update_layout(**common_layout)
+    
+    # Aplicar múltiples colores si se seleccionó la opción
+    if use_multiple_colors and chart_type != "Donut":
+        for i, trace in enumerate(fig.data):
+            trace.update(marker_color=colors[i % len(colors)], marker_line_color=border_colors[i % len(border_colors)], marker_line_width=border_width)
+    
+    # Mostrar el gráfico
+    st.plotly_chart(fig)
+
+    # Forzar el autoescale
+    fig.update_layout(xaxis_autorange=True, yaxis_autorange=True)
 
 # Información adicional
 st.write("Aquí puede añadir información adicional sobre los gráficos generados.")