Update app.py

app.py

@@ -11,6 +11,11 @@ def hex_to_rgba(hex_color, alpha=1.0):
     hex_color = hex_color.lstrip('#')
     return f'rgba({int(hex_color[0:2], 16)},{int(hex_color[2:4], 16)},{int(hex_color[4:6], 16)},{alpha})'
 
+# Colores predefinidos
+predefined_colors = [
+    "#FF5C5C", "#5CCFFF", "#FFA500", "#90EE90", "#9370DB", "#FFD700"
+]
+
 # Lista de fuentes más utilizadas
 font_options = [
     "Times New Roman", "Arial", "Helvetica", "Calibri", "Verdana", 
@@ -50,15 +55,39 @@ show_grid = st.sidebar.checkbox("Mostrar Líneas de Rejilla", value=True)
 x_axis_label = st.sidebar.text_input("Etiqueta para el Eje X", "Tiempo")
 y_axis_label = st.sidebar.text_input("Etiqueta para el Eje Y", "Variables")
 
+# Opción para múltiples colores
+use_multiple_colors = st.sidebar.checkbox("Usar múltiples colores", value=True, key="use_multiple_colors")
+
+# Seleccionar color(es) para el gráfico
+selected_color = st.sidebar.color_picker("Color", "#24CBA0", key="single_color")
+
+# Definir colores
+if use_multiple_colors:
+    colors = [hex_to_rgba(st.sidebar.color_picker(f"Color {i+1}", predefined_colors[i % len(predefined_colors)], key=f"color_{i}"))
+              for i in range(num_y_vars)]
+else:
+    color = hex_to_rgba(selected_color)
+    colors = [color] * num_y_vars  # Definir colors para casos donde no se usa múltiple colores
+
+# Opciones adicionales
+with st.sidebar.expander("Opciones Adicionales"):
+    opacity = st.slider("Opacidad (%)", min_value=0, max_value=100, value=100, step=1) / 100
+    border_width = st.slider("Grosor del Borde", min_value=0.0, max_value=3.0, value=1.0, step=0.1)
+    border_opacity = st.slider("Opacidad del Borde (%)", min_value=0, max_value=100, value=100, step=1) / 100
+
 # Crear el DataFrame con franjas de tiempo
 data_list = []
-for y_name, (start_date, end_date) in y_values.items():
+for idx, (y_name, (start_date, end_date)) in enumerate(y_values.items()):
     if start_date and end_date and y_name:
         data_list.append({
             'Variable': y_name,
             'Start': datetime.strptime(start_date, '%Y-%m-%d'),
             'End': datetime.strptime(end_date, '%Y-%m-%d'),
-            'Category': '' if stacked else y_name  # Eliminar "Apilado" cuando se apilen las franjas
+            'Category': '' if stacked else y_name,  # Eliminar "Apilado" cuando se apilen las franjas
+            'Color': colors[idx],  # Asignar color a cada variable
+            'Opacity': opacity,
+            'Border Width': border_width,
+            'Border Opacity': border_opacity
         })
 
 data = pd.DataFrame(data_list)
@@ -73,6 +102,11 @@ if stacked:
 # Crear gráfico de franjas de tiempo
 fig = px.timeline(data, x_start="Start", x_end="End", y="Category", color="Variable", title=chart_title)
 
+# Actualizar los colores y estilos de las franjas
+for idx, row in data.iterrows():
+    fig.data[idx].update(marker_color=row['Color'], opacity=row['Opacity'])
+    fig.data[idx].update(marker_line=dict(width=row['Border Width'], color=f'rgba(0,0,0,{row["Border Opacity"]})'))
+
 # Actualizar el diseño del gráfico
 fig.update_yaxes(categoryorder="total ascending")