genre counts

main.py

@@ -28,10 +28,16 @@ def fetch_titles_and_genres(engine: Engine) -> pd.DataFrame:
     query = """SELECT title, genres FROM audiobookshelf.ao3;"""
     return run_query(engine, query)
 
+def count_genre_occurrences(data: pd.DataFrame, column: str = "genres") -> pd.Series:
+    """Count how many times each genre appears across all titles."""
+    return data.explode(column)[column].value_counts()
+
 def main():
     engine = get_engine()
     data = fetch_titles_and_genres(engine)
-    print(data.head())
+    genre_counts = count_genre_occurrences(data)
+    print(genre_counts)
+    # print(data.head())
 
 
 if __name__ == "__main__":