convoviz 0.2.12__py3-none-any.whl

convoviz/__init__.py

@@ -0,0 +1,25 @@
+"""Convoviz - ChatGPT data visualization and export tool."""
+
+from convoviz import analysis, config, io, models, renderers, utils
+from convoviz.config import ConvovizConfig, get_default_config
+from convoviz.models import Conversation, ConversationCollection, Message, Node
+from convoviz.pipeline import run_pipeline
+
+__all__ = [
+    # Submodules
+    "analysis",
+    "config",
+    "io",
+    "models",
+    "renderers",
+    "utils",
+    # Main classes
+    "Conversation",
+    "ConversationCollection",
+    "ConvovizConfig",
+    "Message",
+    "Node",
+    # Functions
+    "get_default_config",
+    "run_pipeline",
+]

convoviz/__main__.py

@@ -0,0 +1,6 @@
+"""Allow running convoviz as a module: python -m convoviz"""
+
+from convoviz.cli import main_entry
+
+if __name__ == "__main__":
+    main_entry()

convoviz/analysis/__init__.py

@@ -0,0 +1,9 @@
+"""Data analysis and visualization for convoviz."""
+
+from convoviz.analysis.graphs import generate_week_barplot
+from convoviz.analysis.wordcloud import generate_wordcloud
+
+__all__ = [
+    "generate_week_barplot",
+    "generate_wordcloud",
+]