Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import plotly.express as px
 import pandas as pd
 import numpy as np
+from datetime import datetime
 
 # Configuración de la página
 st.set_page_config(page_title="Generador de Gráficos Personalizado", layout="wide")
@@ -29,11 +30,31 @@ st.sidebar.header("Configuración del Gráfico")
 chart_title = st.sidebar.text_input("Título del Gráfico", "Generador de Gráfico")
 
 # Tipo de gráfico
-chart_type = st.sidebar.selectbox("Tipo de Gráfico", ["Línea", "Área", "Dispersión", "Barras", "Donut"])
+chart_type = st.sidebar.selectbox("Tipo de Gráfico", ["Línea", "Área", "Dispersión", "Barras", "Donut", "Línea temporal"])
 
-# Ingresar valores para los ejes
-x_values = st.sidebar.text_area("Valores para X (separados por comas)", "2013,2014,2015,2016,2017,2018")
-x = x_values.split(",")
+# Función para generar rangos de fechas
+def generate_date_range(start_date, end_date, freq):
+    return pd.date_range(start=start_date, end=end_date, freq=freq).strftime('%Y-%m-%d').tolist()
+
+if chart_type == "Línea temporal":
+    time_unit = st.sidebar.selectbox("Unidad de tiempo", ["Días", "Meses", "Años"])
+    
+    if time_unit == "Días":
+        start_date = st.sidebar.date_input("Fecha de inicio", datetime(2023, 1, 1))
+        end_date = st.sidebar.date_input("Fecha de fin", datetime(2023, 12, 31))
+        x = generate_date_range(start_date, end_date, 'D')
+    elif time_unit == "Meses":
+        start_date = st.sidebar.date_input("Fecha de inicio", datetime(2023, 1, 1))
+        end_date = st.sidebar.date_input("Fecha de fin", datetime(2023, 12, 31))
+        x = generate_date_range(start_date, end_date, 'M')
+    elif time_unit == "Años":
+        start_year = st.sidebar.number_input("Año de inicio", min_value=2000, max_value=2100, value=2020)
+        end_year = st.sidebar.number_input("Año de fin", min_value=2000, max_value=2100, value=2023)
+        x = [str(year) for year in range(start_year, end_year + 1)]
+else:
+    # Ingresar valores para los ejes
+    x_values = st.sidebar.text_area("Valores para X (separados por comas)", "2013,2014,2015,2016,2017,2018")
+    x = x_values.split(",")
 
 # Selector de número de variables Y
 num_y_vars = st.sidebar.number_input("Número de variables Y", min_value=1, max_value=100, value=1, step=1, key="num_y_vars")
@@ -76,11 +97,11 @@ use_multiple_colors = st.sidebar.checkbox("Usar múltiples colores", value=True
 selected_color = st.sidebar.color_picker("Color", "#24CBA0", key="single_color")
 
 # Definir colores
-if use_multiple_colors or chart_type == "Donut":
-    num_colors = max(len(y_values_list), len(x))
-    colors = [hex_to_rgba(selected_color if i == 0 else st.sidebar.color_picker(f"Color {i+1}", predefined_colors[i % len(predefined_colors)], key=f"color_{i}"), alpha=opacity)
+if use_multiple_colors:
+    num_colors = len(x)
+    colors = [hex_to_rgba(st.sidebar.color_picker(f"Color {i+1}", predefined_colors[i % len(predefined_colors)], key=f"color_{i}"), alpha=opacity)
               for i in range(num_colors)]
-    border_colors = [hex_to_rgba(selected_color if i == 0 else predefined_colors[i % len(predefined_colors)], alpha=border_opacity)
+    border_colors = [hex_to_rgba(st.sidebar.color_picker(f"Color de Borde {i+1}", predefined_colors[i % len(predefined_colors)], key=f"border_color_{i}"), alpha=border_opacity)
                      for i in range(num_colors)]
 else:
     color = hex_to_rgba(selected_color, alpha=opacity)
@@ -145,9 +166,9 @@ 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')
+            fig = px.bar(data, x=y_names_list, y="X", orientation='h', barmode='stack')
         else:
-            fig = px.bar(data, x="X", y=y_names_list, barmode='stack' if stacked_bars else 'group')
+            fig = px.bar(data, x="X", y=y_names_list, barmode='stack')
         fig.update_traces(hovertemplate=hovertemplate)
         if use_multiple_colors:
             if len(y_names_list) == 1:  # Solo una variable Y
@@ -184,6 +205,23 @@ else:
             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_names_list, y="X", orientation='h', barmode='stack')
+        else:
+            fig = px.bar(data, x="X", y=y_names_list, barmode='stack')
+        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)
 
     # Añadir anotación para el título
     if chart_type != "Donut":
@@ -218,4 +256,4 @@ else:
         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.")
+st.write("")