Update app.py

app.py

@@ -47,19 +47,21 @@ chart_type = st.sidebar.selectbox("Tipo de Gráfico", ["Línea", "Barras", "Disp
 
 # Ingresar valores para los ejes
 x_values = st.sidebar.text_area("Valores para X (separados por comas)", "2013,2014,2015,2016,2017,2018")
-y_values = st.sidebar.text_area("Valores para Y (separados por comas)", "10,18,5,9,12,21")
+
+# Selector de número de variables Y
+if chart_type == "Línea":
+    num_y_vars = st.sidebar.selectbox("Número de variables Y", list(range(1, 11)), index=0)
+    y_values_list = []
+    for i in range(num_y_vars):
+        y_values = st.sidebar.text_area(f"Valores para Y-{i+1} (separados por comas)", f"10,18,5,9,12,21")
+        y_values_list.append(y_values)
+else:
+    y_values = st.sidebar.text_area("Valores para Y (separados por comas)", "10,18,5,9,12,21")
+    y_values_list = [y_values]
 
 # Ingresar nombres para los valores (opcional)
 names = st.sidebar.text_area("Nombres para los valores (opcional, separados por comas)", "")
 
-# Opciones específicas para gráficos de línea
-if chart_type == "Línea":
-    multiple_y = st.sidebar.checkbox("Múltiple")
-    if multiple_y:
-        y2_values = st.sidebar.text_area("Valores para Y-2 (separados por comas)", "15,23,7,14,19,25")
-    else:
-        y2_values = None
-
 # Etiquetas personalizadas para los ejes
 x_label = st.sidebar.text_input("Etiqueta para el eje X", "X")
 y_label = st.sidebar.text_input("Etiqueta para el eje Y", "Y")
@@ -76,7 +78,7 @@ use_multiple_colors = st.sidebar.checkbox("Usar múltiples colores")
 
 # Seleccionar color(es) para el gráfico
 if use_multiple_colors:
-    num_colors = len(y_values.split(","))
+    num_colors = len(y_values_list)
     colors = [hex_to_rgba(st.sidebar.color_picker(f"Color {i+1}", predefined_colors[i % len(predefined_colors)], key=f"color_{i}"), alpha=0.3)
               for i in range(num_colors)]
     border_colors = [hex_to_rgba(predefined_colors[i % len(predefined_colors)], alpha=0.6)
@@ -88,34 +90,25 @@ else:
 
 # Procesar valores
 x = x_values.split(",")
-y = [float(i) for i in y_values.split(",") if i.strip()]
-if y2_values:
-    y2 = [float(i) for i in y2_values.split(",") if i.strip()]
-else:
-    y2 = None
+y_values_lists = [[float(i) for i in y_values.split(",") if i.strip()] for y_values in y_values_list]
 names = names.split(",") if names else None
 
 # Verificar si las listas tienen el mismo tamaño
-if len(x) != len(y):
-    st.error("Los valores de X y Y deben tener la misma cantidad de elementos.")
-elif y2 and len(x) != len(y2):
-    st.error("Los valores de X y Y-2 deben tener la misma cantidad de elementos.")
-elif names and len(names) != len(y):
+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.")
+elif names and len(names) != len(x):
     st.error("La cantidad de nombres debe ser igual a la cantidad de valores Y.")
 else:
     # Crear un DataFrame
-    data = pd.DataFrame({"X": x, "Y": y})
-    if y2:
-        data["Y2"] = y2
+    data = pd.DataFrame({"X": x})
+    for idx, y_set in enumerate(y_values_lists):
+        data[f"Y{idx+1}"] = y_set
     if names:
         data["Nombre"] = names
 
     # Calcular el rango del eje Y
-    y_min, y_max = calculate_y_range(y)
-    if y2:
-        y2_min, y2_max = calculate_y_range(y2)
-        y_min = min(y_min, y2_min)
-        y_max = max(y_max, y2_max)
+    y_min, y_max = min(min(y_set) for y_set in y_values_lists), max(max(y_set) for y_set in y_values_lists)
+    y_min, y_max = calculate_y_range([y_min, y_max])
     
     # Determinar si mostrar la leyenda
     show_legend = show_legend and bool(names)
@@ -138,26 +131,24 @@ else:
     
     # Generar el gráfico basado en el tipo seleccionado
     if chart_type == "Línea":
-        fig = px.line(data, x="X", y=["Y", "Y2"] if y2 else "Y", line_shape="spline")
+        fig = px.line(data, x="X", y=[f"Y{idx+1}" for idx in range(len(y_values_lists))], line_shape="spline")
         if not use_multiple_colors:
             fig.update_traces(line_color=color)
-        if y2:
-            fig.data[1].line.color = "orange"  # Puedes ajustar este color según tus preferencias
     elif chart_type == "Barras":
-        fig = px.bar(data, x="X", y="Y", hover_name="Nombre" if names else None)
+        fig = px.bar(data, x="X", y="Y1", hover_name="Nombre" if names else None)
         if not use_multiple_colors:
             fig.update_traces(marker_color=color, marker_line_color=border_color, marker_line_width=1.5)
         fig.update_layout(bargap=0.2)  # Ajustar el espacio entre las barras
     elif chart_type == "Dispersión":
-        fig = px.scatter(data, x="X", y="Y", hover_name="Nombre" if names else None)
+        fig = px.scatter(data, x="X", y="Y1", hover_name="Nombre" if names else None)
         if not use_multiple_colors:
             fig.update_traces(marker_color=color)
     elif chart_type == "Área":
-        fig = px.area(data, x="X", y="Y", line_shape="spline", hover_name="Nombre" if names else None)
+        fig = px.area(data, x="X", y="Y1", line_shape="spline", hover_name="Nombre" if names else None)
         if not use_multiple_colors:
             fig.update_traces(fillcolor=color, line_color=border_color)
     elif chart_type == "Pastel":
-        fig = px.pie(data, values="Y", names="X", hover_name="Nombre" if names else None)
+        fig = px.pie(data, values="Y1", names="X", hover_name="Nombre" if names else None)
         if not use_multiple_colors:
             fig.update_traces(marker=dict(colors=[color]*len(x)))