feat: add top 15 states with the most demographics interactions

Author: BMOit

charts/chart_01_daily_aadhaar_engagement_trends.png

@@ -45,6 +45,17 @@ def save(self, fig: Optional[plt.Figure] = None) -> Path:
         if fig is None:
             fig = self.generate()
 
+        fig.text(
+            0.99, 0.01,
+            "github.com/BMOit",
+            transform=fig.transFigure,
+            fontsize=9,
+            color="gray",
+            alpha=0.7,
+            ha="right",
+            va="bottom"
+        )
+
         config.CHARTS_OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
         fig.savefig(
             self.output_path,

charts/chart_02_top_15_states___demographic_interactions.png

@@ -0,0 +1 @@
+"""Top states charts."""

src/charts/__pycache__/base.cpython-313.pyc

@@ -0,0 +1,58 @@
+"""
+Chart 2: Top 15 States - Demographic Interactions
+Type of Chart: Horizontal bar chart
+Data Points:
+  - demographic.csv: state, demo_age_5_17, demo_age_17_
+  - Calculation: SUM(demo_age_5_17 + demo_age_17_) GROUP BY state, TOP 15
+"""
+import matplotlib.pyplot as plt
+
+from src.charts import register_chart
+from src.charts.base import BaseChart
+from src.processors import StateAggregator
+import config
+
+
+@register_chart
+class Chart02TopStatesDemographic(BaseChart):
+
+    @property
+    def chart_id(self) -> str:
+        return "02"
+
+    @property
+    def title(self) -> str:
+        return "Top 15 States - Demographic Interactions"
+
+    def generate(self) -> plt.Figure:
+        data = self.data_loader.get_all_data()
+        processor = StateAggregator()
+        state_data = processor.process(data, dataset="demographic", top_n=15)
+
+        fig, ax = plt.subplots(figsize=(12, 8))
+
+        bars = ax.barh(
+            state_data["state"],
+            state_data["total"],
+            color=config.COLORS["demographic"],
+            edgecolor="white",
+            linewidth=0.5
+        )
+
+        for bar, value in zip(bars, state_data["total"]):
+            ax.text(
+                value + state_data["total"].max() * 0.01,
+                bar.get_y() + bar.get_height() / 2,
+                f"{value:,.0f}",
+                va="center",
+                fontsize=9
+            )
+
+        self._apply_common_style(ax)
+        ax.set_xlabel("Total Demographic Interactions", fontsize=12)
+        ax.set_ylabel("State", fontsize=12)
+        ax.xaxis.set_major_formatter(plt.FuncFormatter(lambda x, p: f"{x/1e6:.1f}M"))
+        ax.set_xlim(0, state_data["total"].max() * 1.15)
+        fig.tight_layout()
+
+        return fig

src/charts/base.py

@@ -1,5 +1,6 @@
 """Data processors for transforming raw data into chart-ready formats."""
 from .base import BaseProcessor
 from .daily_aggregator import DailyAggregator
+from .state_aggregator import StateAggregator
 
-__all__ = ["BaseProcessor", "DailyAggregator"]
+__all__ = ["BaseProcessor", "DailyAggregator", "StateAggregator"]