ExternalVendorCode/lollms-webui
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms-webui.git synced 2025-06-15 22:08:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

enhanced mermaid

Browse Source
This commit is contained in:
Saifeddine ALOUI
2024-02-08 23:28:50 +01:00
parent 9d90c8c921
commit 175a343c60
3 changed files with 23 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
utilities/execution_engines/graphviz_execution_engine.py
View File

@ -31,6 +31,11 @@ def build_graphviz_output(code, ifram_name="unnamed"):
'<div style="width: 100%; margin: 0 auto;">',
f'<iframe id="{ifram_name}" srcdoc=\'',
'<style>',
'iframe {',
'width: 100%;',
'height: 100%;',
'border: none;',
'}',
'.graph {',
'background-color: transparent;',
'padding: 20px;',

7
utilities/execution_engines/javascript_execution_engine.py
View File

@ -30,6 +30,13 @@ def build_javascript_output(code, ifram_name="unnamed"):
rendered = "\n".join([
'<div style="width: 100%; margin: 0 auto;">',
f'<iframe id="{ifram_name}" srcdoc="',
'<style>',
'iframe {',
'width: 100%;',
'height: 100%;',
'border: none;',
'}',
'</style>',
'<script>',
code,
'</script>',

14
utilities/execution_engines/mermaid_execution_engine.py
View File

@ -29,8 +29,13 @@ def build_mermaid_output(code, ifram_name="unnamed"):
start_time = time.time()
rendered = "\n".join([
'<div style="width: 100%; margin: 0 auto;">',
f'<iframe id="{ifram_name}" srcdoc="',
f'<iframe id="{ifram_name}" style="width: 100%" srcdoc="',
'<style>',
'iframe {',
'width: 100%;',
'height: 100%;',
'border: none;',
'}',
'.mermaid {',
'background-color: transparent;',
'padding: 20px;',
@ -44,7 +49,6 @@ def build_mermaid_output(code, ifram_name="unnamed"):
'<div class=\'mermaid\'>',
"\n".join([c for c in code.split("\n") if c.strip()!=""]),
'</div>',
'<button onclick="saveSVG()">Save SVG</button>',
'<script src=\'https://cdn.jsdelivr.net/npm/mermaid/dist/mermaid.min.js\'></script>',
'<script>',
'// Initialize the mermaid library and render our diagram',
@ -60,10 +64,11 @@ def build_mermaid_output(code, ifram_name="unnamed"):
'a.href = url;',
'a.download = "diagram.svg";',
'a.click();',
'}',
'}',
'</script>',
'<div style=\'text-align: center;\'>',
'</div>',
'<button onclick="saveSVG()">Save SVG</button>',
'" style="width: 100%; height: 600px; border: none;"></iframe>',
'</div>'
]
@ -72,6 +77,9 @@ def build_mermaid_output(code, ifram_name="unnamed"):
return {"output": rendered, "execution_time": execution_time}
def execute_mermaid(code, discussion_id, message_id):
return build_mermaid_output(code)