Refine Lumi stats and detail UX

1 files changed, 20 insertions, 0 deletions

diff --git a/routers/stats.py b/routers/stats.py
index d600e5a..ad07e0c 100644
--- a/routers/stats.py
+++ b/routers/stats.py
@@ -65,7 +65,9 @@ def _build_stats_payload(films: list[Film]) -> dict:
             watched_with["solo"] += 1
 
     total_watched = len(films)
+    rewatches = sum(1 for film in films if film.rewatch or film.rewatch_count > 0)
     total_runtime_minutes = sum(film.runtime for film in films if film.runtime)
+    average_stars = round(sum(film.stars for film in films) / total_watched, 1) if total_watched else 0
 
     title_groups = defaultdict(list)
     for film in films:
@@ -90,6 +92,16 @@ def _build_stats_payload(films: list[Film]) -> dict:
 
     rewatch_details.sort(key=lambda x: (-x["watches"], x["title"]))
 
+    top_director = None
+    if directors:
+        top_director_name, top_director_count = sorted(directors.items(), key=lambda item: (-item[1], item[0]))[0]
+        top_director = {"director": top_director_name, "count": top_director_count}
+
+    top_month = None
+    if months:
+        top_month_key, top_month_count = sorted(months.items(), key=lambda item: (-item[1], item[0]))[0]
+        top_month = {"month": top_month_key, "count": top_month_count}
+
     today = date.today()
     start_day = today - timedelta(days=364)
     trailing_days = []
@@ -105,6 +117,7 @@ def _build_stats_payload(films: list[Film]) -> dict:
         },
         "total_watched": total_watched,
         "total_runtime_minutes": total_runtime_minutes,
+        "average_stars": average_stars,
         "films_per_country": [
             {"country": country, "count": count}
             for country, count in sorted(countries.items(), key=lambda item: (-item[1], item[0]))
@@ -143,6 +156,13 @@ def _build_stats_payload(films: list[Film]) -> dict:
             {"watched_with": watched_with_value, "count": count}
             for watched_with_value, count in sorted(watched_with.items(), key=lambda item: (-item[1], item[0]))
         ],
+        "rewatch_rate": {
+            "rewatched": rewatches,
+            "total_watched": total_watched,
+            "rate": round(rewatches / total_watched, 4) if total_watched else 0,
+        },
+        "top_director": top_director,
+        "top_month": top_month,
     }