Update utils.py

utils.py

@@ -1,18 +1,104 @@
-from pathlib import Path
-#import pandas as pd
-import faicons as fa
-from shiny import ui
-
-app_dir = Path(__file__).parent
-#tips = pd.read_csv(app_dir / "tips.csv")
-
-
-def create_nav_button(button_icon, button_label_text, button_link, button_cls = "btn btn-primary"):
-    return ui.tags.a(
-        ui.HTML(f"{fa.icon_svg(button_icon)} {button_label_text}"),
-        href=button_link, 
-        class_=button_cls
-    )
-
-
-
+from pathlib import Path
+#import pandas as pd
+import faicons as fa
+from shiny import ui
+from shared import app_dir
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+#app_dir = Path(__file__).parent
+#tips = pd.read_csv(app_dir / "tips.csv")
+
+
+def create_nav_button(button_icon, button_label_text, button_link, button_cls = "btn btn-primary"):
+    return ui.tags.a(
+        ui.HTML(f"{fa.icon_svg(button_icon)} {button_label_text}"),
+        href=button_link, 
+        class_=button_cls
+    )
+
+def conlogo():
+    img = {
+        "src": app_dir / "images" / "logo.png",
+        "style": "width: 80%; height: auto; max-height: 150px; margin-bottom: 0px;"
+        }
+    return img
+
+def vendorlogo():
+    img = {
+        "src": app_dir / "images" / "logo2.png",
+        "style": "width: 80%; height: auto; max-height: 150px; margin-bottom: 0px;"
+        }
+    return img
+
+def create_bar_chart(categories, values, plottitle = "Bar Chart Example"):
+
+    fig, ax = plt.subplots()
+    #ax.bar(categories, values, color=["#1f77b4", "#ff7f0e", "#2ca02c"])
+    ax.bar(categories, values)
+    ax.set_title(plottitle)
+    ax.set_ylabel("Values")
+    ax.set_xlabel("Categories")
+
+    return fig
+
+def create_pie_chart(labels, sizes, plottitle):
+    #sizes = [22, 41, 37]
+    #colors = ["#00b3b3", "#70d281", "#ff7f0e"]
+    #explode = (0.1, 0, 0)  # Explode the first slice for emphasis
+
+    fig, ax = plt.subplots()
+    wedges, texts, autotexts = ax.pie(
+        sizes,
+        #explode=explode,
+        labels=labels,
+        autopct="%1.0f%%",
+        startangle=90,
+        #colors=colors,
+        textprops=dict(color="black"),
+        wedgeprops=dict(width=0.4)  # Adjust width for the donut effect
+    )
+
+    # Customizing the labels
+    for text, label in zip(autotexts, labels):
+        text.set_color("black")
+        text.set_fontsize(12)
+
+    #ax.set_title("What Types of Trips are Occuring within Napa County on a Weekday?", fontsize=16, fontweight="bold")
+    ax.set_title(plottitle, fontsize=16, fontweight="bold")
+    return fig
+
+def create_stacked_bar_chart(categories, intra_napa, into_napa, out_napa):
+    # Data for the stacked bar chart
+    #categories = ["Early AM", "AM Peak", "Mid-Day", "PM Peak", "Evening"]
+    #intra_napa = [2000, 16000, 14000, 12000, 5000]
+    #into_napa = [1000, 8000, 2000, 3000, 2000]
+    #out_napa = [2000, 4000, 2000, 5000, 1000]
+
+    bar_width = 0.5  # Width of the bars
+    x = np.arange(len(categories))  # x-axis positions
+
+    fig, ax = plt.subplots(figsize=(8, 6))
+
+    # Stacking the bars
+    ax.bar(x, intra_napa, bar_width, label="Intra-Napa County", color="#70d281")
+    ax.bar(x, into_napa, bar_width, bottom=intra_napa, label="Into Napa County", color="#ff7f0e")
+    ax.bar(x, out_napa, bar_width, bottom=np.array(intra_napa) + np.array(into_napa), label="Out of Napa County", color="#00b3b3")
+
+    # Customizing the plot
+    ax.set_xticks(x)
+    ax.set_xticklabels(categories)
+    ax.set_ylabel("Trips")
+    ax.set_title("Weekday Work Trip Types")
+    ax.legend(loc="upper right")
+
+    # Add value annotations
+    for i in range(len(categories)):
+        ax.text(x[i], intra_napa[i] / 2, f"{intra_napa[i]}", ha="center", va="center", color="white")
+        ax.text(x[i], intra_napa[i] + into_napa[i] / 2, f"{into_napa[i]}", ha="center", va="center", color="white")
+        ax.text(x[i], intra_napa[i] + into_napa[i] + out_napa[i] / 2, f"{out_napa[i]}", ha="center", va="center", color="white")
+
+    plt.tight_layout()
+    return fig
+