Spaces:
Running
Running
| # 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 | |
| ###################################################################################################################### | |