Function main

examples/generate_documentation.py:153–193

()

def main():
    parser = argparse.ArgumentParser(description="Generate example MDX files from Jupyter notebooks")
    parser.add_argument("notebook_path", help="Path to the Jupyter notebook")

    notebook_path = parser.parse_args().notebook_path

    # Validate notebook path
    notebook_file = Path(notebook_path)
    if not notebook_file.exists():
        sys.exit(f"Error: Notebook file not found: {notebook_path}")

    if notebook_file.suffix != ".ipynb":
        sys.exit(f"Error: File must be a Jupyter notebook (.ipynb): {notebook_path}")

    if not str(notebook_file).startswith("examples/"):
        sys.exit(f"Error: Notebook must be in the examples/ directory: {notebook_path}")

    # Generate output path
    folder_name = notebook_file.parent.name
    output_path = Path(f"docs/v2/examples/{folder_name}.mdx")

    print(f"Processing: {notebook_path} -> {output_path}")

    # Convert notebook to markdown and process
    markdown_content = convert_notebook_to_markdown(notebook_path)
    processed_content = process_pip_installations(markdown_content)

    # Check for existing special frontmatter
    existing_frontmatter = get_existing_frontmatter(output_path)

    if existing_frontmatter:
        final_content = generate_mdx_content(notebook_path, processed_content, existing_frontmatter)
    else:
        print("Generating new frontmatter and content")
        final_content = generate_mdx_content(notebook_path, processed_content)

    # Write the file
    output_path.parent.mkdir(parents=True, exist_ok=True)
    output_path.write_text(final_content, encoding="utf-8")

    print(f"Successfully generated: {output_path}")


if __name__ == "__main__":