convoviz 0.4.1__py3-none-any.whl

convoviz/__init__.py

@@ -0,0 +1,34 @@
+"""Convoviz - ChatGPT data visualization and export tool."""
+
+from convoviz import 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",
+]
+
+
+def __getattr__(name: str):
+    """Lazy import for optional submodules like analysis."""
+    if name == "analysis":
+        from convoviz import analysis
+
+        return analysis
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

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,22 @@
+"""Data analysis and visualization for convoviz.
+
+Requires the [viz] extra: uv tool install "convoviz[viz]"
+"""
+
+__all__ = [
+    "generate_week_barplot",
+    "generate_wordcloud",
+]
+
+
+def __getattr__(name: str):
+    """Lazy import for visualization functions requiring optional dependencies."""
+    if name == "generate_week_barplot":
+        from convoviz.analysis.graphs import generate_week_barplot
+
+        return generate_week_barplot
+    if name == "generate_wordcloud":
+        from convoviz.analysis.wordcloud import generate_wordcloud
+
+        return generate_wordcloud
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")