cat-stack 1.0.3__tar.gz → 1.0.4__tar.gz

{cat_stack-1.0.3 → cat_stack-1.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cat-stack
-Version: 1.0.3
+Version: 1.0.4
 Summary: Domain-agnostic text, image, PDF, and DOCX classification engine powered by LLMs
 Project-URL: Documentation, https://github.com/chrissoria/cat-stack#readme
 Project-URL: Issues, https://github.com/chrissoria/cat-stack/issues

{cat_stack-1.0.3 → cat_stack-1.0.4}/src/cat_stack/__about__.py

@@ -1,7 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present Christopher Soria <chrissoria@berkeley.edu>
 #
 # SPDX-License-Identifier: GPL-3.0-or-later
-__version__ = "1.0.3"
+__version__ = "1.0.4"
 __author__ = "Chris Soria"
 __email__ = "chrissoria@berkeley.edu"
 __title__ = "cat-stack"

{cat_stack-1.0.3 → cat_stack-1.0.4}/src/cat_stack/text_functions.py

@@ -820,13 +820,17 @@ Output:
 
     print("\nTop categories:\n" + "\n".join(f"{i+1}. {c}" for i, c in enumerate(final[:max_categories])))
 
+    top = final[:max_categories]
+
     if filename:
-        result.to_csv(filename, index=False)
-        print(f"\nResults saved to {filename}")
+        import pandas as _pd
+        top_df = _pd.DataFrame({"rank": range(1, len(top) + 1), "category": top})
+        top_df.to_csv(filename, index=False)
+        print(f"\nTop {len(top)} categories saved to {filename}")
 
     return {
         "counts_df": result,
-        "top_categories": final[:max_categories],
+        "top_categories": top,
         "raw_top_text": top_categories_text
     }