File size: 3,924 Bytes
c5b0bb7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
# Mind_Map_tab.py
# Description: File contains functions for generation of PlantUML mindmaps for the gradio tab
#
# Imports
import re
#
# External Libraries
import gradio as gr
#
######################################################################################################################
#
# Functions:

def parse_plantuml_mindmap(plantuml_text: str) -> dict:
    """Parse PlantUML mindmap syntax into a nested dictionary structure"""
    lines = [line.strip() for line in plantuml_text.split('\n')
             if line.strip() and not line.strip().startswith('@')]

    root = None
    nodes = []
    stack = []

    for line in lines:
        level_match = re.match(r'^([+\-*]+|\*+)', line)
        if not level_match:
            continue
        level = len(level_match.group(0))
        text = re.sub(r'^([+\-*]+|\*+)\s*', '', line).strip('[]').strip('()')
        node = {'text': text, 'children': []}

        while stack and stack[-1][0] >= level:
            stack.pop()

        if stack:
            stack[-1][1]['children'].append(node)
        else:
            root = node

        stack.append((level, node))

    return root

def create_mindmap_html(plantuml_text: str) -> str:
    """Convert PlantUML mindmap to HTML visualization with collapsible nodes using CSS only"""
    # Parse the mindmap text into a nested structure
    root_node = parse_plantuml_mindmap(plantuml_text)
    if not root_node:
        return "<p>No valid mindmap content provided.</p>"

    html = "<style>"
    html += """

    details {

        margin-left: 20px;

    }

    summary {

        cursor: pointer;

        padding: 5px;

        border: 1px solid #333;

        border-radius: 3px;

        background-color: #e6f3ff;

    }

    .mindmap-node {

        margin-left: 20px;

        padding: 5px;

        border: 1px solid #333;

        border-radius: 3px;

    }

    """
    html += "</style>"

    colors = ['#e6f3ff', '#f0f7ff', '#f5f5f5', '#fff0f0', '#f0fff0']

    def create_node_html(node, level):
        bg_color = colors[(level - 1) % len(colors)]
        if node['children']:
            children_html = ''.join(create_node_html(child, level + 1) for child in node['children'])
            return f"""

            <details open>

                <summary style="background-color: {bg_color};">{node['text']}</summary>

                {children_html}

            </details>

            """
        else:
            return f"""

            <div class="mindmap-node" style="background-color: {bg_color}; margin-left: {level * 20}px;">

                {node['text']}

            </div>

            """

    html += create_node_html(root_node, level=1)
    return html

# Create Gradio interface
def create_mindmap_tab():
    with gr.TabItem("PlantUML Mindmap"):
        gr.Markdown("# Collapsible PlantUML Mindmap Visualizer")
        gr.Markdown("Convert PlantUML mindmap syntax to a visual mindmap with collapsible nodes.")
        plantuml_input = gr.Textbox(
            lines=15,
            label="Enter PlantUML mindmap",
            placeholder="""@startmindmap

    * Project Planning

    ** Requirements

    *** Functional Requirements

    **** User Interface

    **** Backend Services

    *** Technical Requirements

    **** Performance

    **** Security

    ** Timeline

    *** Phase 1

    *** Phase 2

    ** Resources

    *** Team

    *** Budget

    @endmindmap"""
        )
        submit_btn = gr.Button("Generate Mindmap")
        mindmap_output = gr.HTML(label="Mindmap Output")
        submit_btn.click(
            fn=create_mindmap_html,
            inputs=plantuml_input,
            outputs=mindmap_output
        )

#
# End of Mind_Map_tab.py
######################################################################################################################