convoviz 0.2.6__tar.gz → 0.2.7__tar.gz

{convoviz-0.2.6 → convoviz-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: convoviz
-Version: 0.2.6
+Version: 0.2.7
 Summary: Get analytics and visualizations on your ChatGPT data!
 Keywords: markdown,chatgpt,openai,visualization,analytics,json,export,data-analysis,obsidian
 Author: Mohamed Cheikh Sidiya
@@ -84,11 +84,6 @@ You can provide arguments directly to skip the prompts:
 convoviz --input path/to/your/export.zip --output path/to/output/folder
 ```
 
-Inputs can be any of:
-- A ChatGPT export ZIP (downloaded from OpenAI)
-- An extracted export directory containing `conversations.json`
-- A `conversations.json` file directly
-
 Notes:
 - `--zip` / `-z` is kept as an alias for `--input` for convenience.
 - You can force non-interactive mode with `--no-interactive`.

{convoviz-0.2.6 → convoviz-0.2.7}/README.md

@@ -58,11 +58,6 @@ You can provide arguments directly to skip the prompts:
 convoviz --input path/to/your/export.zip --output path/to/output/folder
 ```
 
-Inputs can be any of:
-- A ChatGPT export ZIP (downloaded from OpenAI)
-- An extracted export directory containing `conversations.json`
-- A `conversations.json` file directly
-
 Notes:
 - `--zip` / `-z` is kept as an alias for `--input` for convenience.
 - You can force non-interactive mode with `--no-interactive`.

{convoviz-0.2.6 → convoviz-0.2.7}/convoviz/cli.py

@@ -5,7 +5,7 @@ from pathlib import Path
 import typer
 from rich.console import Console
 
-from convoviz.config import get_default_config
+from convoviz.config import FolderOrganization, get_default_config
 from convoviz.exceptions import ConfigurationError, InvalidZipError
 from convoviz.interactive import run_interactive_config
 from convoviz.io.loaders import find_latest_zip
@@ -38,6 +38,12 @@ def run(
         "-o",
         help="Path to the output directory.",
     ),
+    flat: bool = typer.Option(
+        False,
+        "--flat",
+        "-f",
+        help="Put all markdown files in a single folder (disables date organization).",
+    ),
     interactive: bool | None = typer.Option(
         None,
         "--interactive/--no-interactive",
@@ -57,6 +63,8 @@ def run(
         config.input_path = input_path
     if output_dir:
         config.output_folder = output_dir
+    if flat:
+        config.folder_organization = FolderOrganization.FLAT
 
     # Determine mode: interactive if explicitly requested or no input provided
     use_interactive = interactive if interactive is not None else (input_path is None)

{convoviz-0.2.6 → convoviz-0.2.7}/convoviz/config.py

@@ -1,11 +1,19 @@
 """Configuration models using Pydantic v2."""
 
+from enum import Enum
 from pathlib import Path
 from typing import Literal
 
 from pydantic import BaseModel, Field
 
 
+class FolderOrganization(str, Enum):
+    """How to organize markdown output files in folders."""
+
+    FLAT = "flat"  # All files in one directory (default)
+    DATE = "date"  # Nested by year/month/week
+
+
 class AuthorHeaders(BaseModel):
     """Headers for different message authors in markdown output."""
 
@@ -83,6 +91,7 @@ class ConvovizConfig(BaseModel):
 
     input_path: Path | None = None
     output_folder: Path = Field(default_factory=lambda: Path.home() / "Documents" / "ChatGPT-Data")
+    folder_organization: FolderOrganization = FolderOrganization.DATE
     message: MessageConfig = Field(default_factory=MessageConfig)
     conversation: ConversationConfig = Field(default_factory=ConversationConfig)
     wordcloud: WordCloudConfig = Field(default_factory=WordCloudConfig)

{convoviz-0.2.6 → convoviz-0.2.7}/convoviz/io/writers.py

@@ -6,12 +6,59 @@ from pathlib import Path
 from orjson import OPT_INDENT_2, dumps
 from tqdm import tqdm
 
-from convoviz.config import AuthorHeaders, ConversationConfig
+from convoviz.config import AuthorHeaders, ConversationConfig, FolderOrganization
 from convoviz.io.assets import copy_asset, resolve_asset_path
 from convoviz.models import Conversation, ConversationCollection
 from convoviz.renderers import render_conversation
 from convoviz.utils import sanitize
 
+# Month names for folder naming
+_MONTH_NAMES = [
+    "January",
+    "February",
+    "March",
+    "April",
+    "May",
+    "June",
+    "July",
+    "August",
+    "September",
+    "October",
+    "November",
+    "December",
+]
+
+
+def get_date_folder_path(conversation: Conversation) -> Path:
+    """Get the date-based folder path for a conversation.
+
+    Creates a nested structure: year/month/week
+    Example: 2024/03-March/Week-02/
+
+    Args:
+        conversation: The conversation to get the path for
+
+    Returns:
+        Relative path for the date-based folder structure
+    """
+    create_time = conversation.create_time
+
+    # Year folder: "2024"
+    year = str(create_time.year)
+
+    # Month folder: "03-March"
+    month_num = create_time.month
+    month_name = _MONTH_NAMES[month_num - 1]
+    month = f"{month_num:02d}-{month_name}"
+
+    # Week folder: "Week-01" through "Week-05" (week of the month)
+    # Calculate which week of the month this date falls into
+    day = create_time.day
+    week_of_month = (day - 1) // 7 + 1
+    week = f"Week-{week_of_month:02d}"
+
+    return Path(year) / month / week
+
 
 def save_conversation(
     conversation: Conversation,
@@ -74,6 +121,7 @@ def save_collection(
     config: ConversationConfig,
     headers: AuthorHeaders,
     *,
+    folder_organization: FolderOrganization = FolderOrganization.FLAT,
     progress_bar: bool = False,
 ) -> None:
     """Save all conversations in a collection to markdown files.
@@ -83,6 +131,7 @@ def save_collection(
         directory: Target directory
         config: Conversation rendering configuration
         headers: Author header configuration
+        folder_organization: How to organize files in folders (flat or by date)
         progress_bar: Whether to show a progress bar
     """
     directory.mkdir(parents=True, exist_ok=True)
@@ -92,7 +141,14 @@ def save_collection(
         desc="Writing Markdown 📄 files",
         disable=not progress_bar,
     ):
-        filepath = directory / f"{sanitize(conv.title)}.md"
+        # Determine target directory based on organization setting
+        if folder_organization == FolderOrganization.DATE:
+            target_dir = directory / get_date_folder_path(conv)
+            target_dir.mkdir(parents=True, exist_ok=True)
+        else:
+            target_dir = directory
+
+        filepath = target_dir / f"{sanitize(conv.title)}.md"
         save_conversation(conv, filepath, config, headers, source_path=collection.source_path)
 
 

{convoviz-0.2.6 → convoviz-0.2.7}/convoviz/pipeline.py

@@ -113,6 +113,7 @@ def run_pipeline(config: ConvovizConfig) -> None:
         markdown_folder,
         config.conversation,
         config.message.author_headers,
+        folder_organization=config.folder_organization,
         progress_bar=True,
     )
     console.print(

{convoviz-0.2.6 → convoviz-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "convoviz"
-version = "0.2.6"
+version = "0.2.7"
 description = "Get analytics and visualizations on your ChatGPT data!"
 license = "MIT"
 keywords = [