Themes

​

ThemeTree

• label (str): The name of the theme or sub-theme.
• node (int): A unique identifier for the node.
• summary (str, optional): A brief explanation of the node’s relevance.
• children (Optional[List[ThemeTree]]): A list of child nodes representing sub-themes.
• keywords (Optional[List[str]]): A list of keywords summarizing the main theme.

• from_dict(tree_dict): Create a ThemeTree object from a dictionary.
• as_string(prefix=""): Convert the tree into a string.
• get_label_summaries(): Extract all label summaries from the tree.
• get_summaries(): Extract all node summaries from the tree.
• get_terminal_label_summaries(): Extract label/summary pairs from terminal nodes.
• get_terminal_labels(): Extract terminal node labels.
• get_terminal_summaries(): Extract summaries from terminal nodes.
• print(prefix=""): Print the tree.
• visualize(engine="graphviz"): Visualize the tree as a mind map (requires graphviz or plotly).
• get_label_to_parent_mapping(): Map each leaf node label to its parent.
• save_json(filepath): Save the ThemeTree as a JSON file.

from bigdata_research_tools.themes import ThemeTree

tree = ThemeTree(
    label="AI",
    node=0,
    summary="Artificial Intelligence and its applications.",
    children=[],
    keywords=["machine learning", "automation"]
)
print(tree.as_string())

​

generate_theme_tree

• main_theme (str): The primary theme to analyze.
• focus (str, optional): Specific aspect(s) to guide sub-theme generation.
• llm_model_config (dict, optional): Configuration for the LLM used to generate themes.

• ThemeTree: The generated theme tree.

from bigdata_research_tools.themes import generate_theme_tree

tree = generate_theme_tree(main_theme="AI", focus="healthcare")
tree.print()

​

stringify_label_summaries

• label_summaries (dict): Dictionary of label summaries from a ThemeTree.

• List[str]: List of strings, each containing a label and its summary.

from bigdata_research_tools.themes import stringify_label_summaries

summaries = tree.get_label_summaries()
summary_strings = stringify_label_summaries(summaries)
for s in summary_strings:
    print(s)