Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import plotly.graph_objects as go
+import plotly.express as px
 import pandas as pd
 from datetime import datetime, timedelta
 
@@ -24,31 +24,31 @@ st.sidebar.header("Configuración del Gráfico")
 chart_title = st.sidebar.text_input("Título del Gráfico", "Gráfico de Franjas de Tiempo Apiladas")
 
 # Seleccionar formato de tiempo
-time_format = st.sidebar.selectbox("Formato de Tiempo", ["Año", "Mes", "Día"])
+time_format = st.sidebar.selectbox("Formato de Tiempo", ["Año"])
 
 # Ingresar valores para franjas de tiempo
 num_y_vars = st.sidebar.number_input("Número de Variables Y", min_value=1, max_value=10, value=3, step=1)
+
 y_values = {}
 for i in range(num_y_vars):
     y_start = st.sidebar.text_input(f"Fecha de Inicio para Y-{i+1} (YYYY-MM-DD)", key=f"y_start_{i}", value=f"202{i}-01-01")
     y_end = st.sidebar.text_input(f"Fecha de Fin para Y-{i+1} (YYYY-MM-DD)", key=f"y_end_{i}", value=f"202{i+1}-12-31")
     y_name = st.sidebar.text_input(f"Nombre de la Variable Y-{i+1}", key=f"y_name_{i}", value=f"Y-{i+1}")
-    y_color = st.sidebar.color_picker(f"Color para Y-{i+1}", f"#{hash(y_name)&0xFFFFFF:06x}", key=f"y_color_{i}")
     if y_start and y_end and y_name:
-        y_values[y_name] = (y_start, y_end, y_color)
+        y_values[y_name] = (y_start, y_end)
 
 # Opción para apilar franjas de tiempo
 stacked = st.sidebar.checkbox("Apilar Franjas de Tiempo", value=True)
 
 # Crear el DataFrame con franjas de tiempo
 data_list = []
-for y_name, (start_date, end_date, color) in y_values.items():
+for y_name, (start_date, end_date) in y_values.items():
     data_list.append({
         'Variable': y_name,
         'Start': datetime.strptime(start_date, '%Y-%m-%d'),
-        'End': datetime.strptime(end_date, '%Y-%m-%d'),
-        'Color': color
+        'End': datetime.strptime(end_date, '%Y-%m-%d')
     })
+
 data = pd.DataFrame(data_list)
 
 # Ajustar fechas para simular apilamiento si está activado
@@ -58,45 +58,35 @@ if stacked:
     data['End'] = data['End'] + data.groupby('Variable').cumcount() * offset
 
 # Crear gráfico de franjas de tiempo
-fig = go.Figure()
-
-for _, row in data.iterrows():
-    fig.add_trace(go.Bar(
-        x=[(row['End'] - row['Start']).days],  # Convertir timedelta a días
-        y=[row['Variable']],
-        orientation='h',
-        base=row['Start'],
-        marker_color=row['Color'],
-        name=row['Variable'],
-        hoverinfo='text',
-        text=f"{row['Variable']}<br>Inicio: {row['Start'].strftime('%Y-%m-%d')}<br>Fin: {row['End'].strftime('%Y-%m-%d')}",
-    ))
+fig = px.timeline(data, x_start="Start", x_end="End", y="Variable", color="Variable", title=chart_title)
 
 # Actualizar el diseño del gráfico
+fig.update_yaxes(categoryorder="total ascending")
+
+# Personalizar el formato de tiempo en el eje X
+if time_format == "Año":
+    tickformat = "%Y"
+
 fig.update_layout(
-    title=chart_title,
     xaxis_title="Tiempo",
     yaxis_title="Variables",
     xaxis=dict(
-        type='date',
-        tickformat='%Y' if time_format == "Año" else ('%b %Y' if time_format == "Mes" else '%Y-%m-%d'),
+        tickformat=tickformat,
         titlefont=dict(size=14),
         tickfont=dict(size=12),
     ),
     yaxis=dict(
-        categoryorder="total ascending",
         titlefont=dict(size=14),
         tickfont=dict(size=12),
     ),
     width=800,
     height=600,
     margin=dict(l=60, r=40, t=100, b=40),
-    font=dict(family="Times New Roman", size=18, color="black"),
-    barmode='stack' if stacked else 'overlay'
+    font=dict(family="Times New Roman", size=18, color="black")
 )
 
 # Mostrar el gráfico
 st.plotly_chart(fig)
 
 # Información adicional
-st.write("Aquí puede añadir información adicional sobre los gráficos generados.")
+st.write("Aquí puede añadir información adicional sobre los gráficos generados.")