diff --git a/endpoints/lollms_advanced.py b/endpoints/lollms_advanced.py
index 7664a04a..3282ad41 100644
--- a/endpoints/lollms_advanced.py
+++ b/endpoints/lollms_advanced.py
@@ -27,6 +27,11 @@ import platform
from utilities.execution_engines.python_execution_engine import execute_python
from utilities.execution_engines.latex_execution_engine import execute_latex
from utilities.execution_engines.shell_execution_engine import execute_bash
+from utilities.execution_engines.javascript_execution_engine import execute_javascript
+from utilities.execution_engines.mermaid_execution_engine import execute_mermaid
+from utilities.execution_engines.graphviz_execution_engine import execute_graphviz
+
+
# ----------------------- Defining router and main class ------------------------------
@@ -50,15 +55,31 @@ async def execute_code(request: Request):
language = data.get("language","python")
- ASCIIColors.info("Executing python code:")
- ASCIIColors.yellow(code)
if language=="python":
+ ASCIIColors.info("Executing python code:")
+ ASCIIColors.yellow(code)
return execute_python(code, discussion_id, message_id)
+ if language=="javascript":
+ ASCIIColors.info("Executing javascript code:")
+ ASCIIColors.yellow(code)
+ return execute_javascript(code, discussion_id, message_id)
elif language=="latex":
+ ASCIIColors.info("Executing latex code:")
+ ASCIIColors.yellow(code)
return execute_latex(code, discussion_id, message_id)
elif language in ["bash","shell","cmd","powershell"]:
+ ASCIIColors.info("Executing shell code:")
+ ASCIIColors.yellow(code)
return execute_bash(code, discussion_id, message_id)
+ elif language in ["mermaid"]:
+ ASCIIColors.info("Executing mermaid code:")
+ ASCIIColors.yellow(code)
+ return execute_mermaid(code, discussion_id, message_id)
+ elif language in ["graphviz","dot"]:
+ ASCIIColors.info("Executing graphviz code:")
+ ASCIIColors.yellow(code)
+ return execute_graphviz(code, discussion_id, message_id)
return {"output": "Unsupported language", "execution_time": 0}
except Exception as ex:
trace_exception(ex)
@@ -252,3 +273,4 @@ def stop_recording():
lollmsElfServer.info("Stopping audio capture")
text = lollmsElfServer.audio_cap.stop_recording()
return text
+
diff --git a/lollms_core b/lollms_core
index d0975b0c..33ec4861 160000
--- a/lollms_core
+++ b/lollms_core
@@ -1 +1 @@
-Subproject commit d0975b0c6ad01b4385512755af32fa505c4f3bdb
+Subproject commit 33ec48614462d8e3c3be21afe8dd7b70cec7c3f3
diff --git a/utilities/execution_engines/graphviz_execution_engine.py b/utilities/execution_engines/graphviz_execution_engine.py
new file mode 100644
index 00000000..072070ae
--- /dev/null
+++ b/utilities/execution_engines/graphviz_execution_engine.py
@@ -0,0 +1,66 @@
+"""
+project: lollms_webui
+file: shell_execution_engine.py
+author: ParisNeo
+description:
+ This is a utility for executing python code
+
+"""
+from lollms_webui import LOLLMSWebUI
+from ascii_colors import get_trace_exception, trace_exception
+import time
+import subprocess
+import json
+
+lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
+
+def build_graphviz_output(code, ifram_name="unnamed"):
+ """
+ This function creates an HTML5 iframe with the given HTML content and iframe name.
+
+ Args:
+ html (str): The HTML content to be displayed in the iframe.
+ ifram_name (str, optional): The name of the iframe. Defaults to "unnamed".
+
+ Returns:
+ str: The HTML string for the iframe.
+ """
+ # Start the timer.
+ start_time = time.time()
+ rendered = "\n".join([
+ '
',
+ f'',
+ '
'
+ ]
+ )
+ execution_time = time.time() - start_time
+ return {"output": rendered, "execution_time": execution_time}
+
+def execute_graphviz(code, discussion_id, message_id):
+
+ return build_graphviz_output(code)
\ No newline at end of file
diff --git a/utilities/execution_engines/javascript_execution_engine.py b/utilities/execution_engines/javascript_execution_engine.py
new file mode 100644
index 00000000..3b81b6ff
--- /dev/null
+++ b/utilities/execution_engines/javascript_execution_engine.py
@@ -0,0 +1,46 @@
+"""
+project: lollms_webui
+file: shell_execution_engine.py
+author: ParisNeo
+description:
+ This is a utility for executing python code
+
+"""
+from lollms_webui import LOLLMSWebUI
+from ascii_colors import get_trace_exception, trace_exception
+import time
+import subprocess
+import json
+
+lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
+
+def build_javascript_output(code, ifram_name="unnamed"):
+ """
+ This function creates an HTML5 iframe with the given HTML content and iframe name.
+
+ Args:
+ html (str): The HTML content to be displayed in the iframe.
+ ifram_name (str, optional): The name of the iframe. Defaults to "unnamed".
+
+ Returns:
+ str: The HTML string for the iframe.
+ """
+ # Start the timer.
+ start_time = time.time()
+ rendered = "\n".join([
+ '
',
+ f'',
+ '
'
+ ]
+ )
+ execution_time = time.time() - start_time
+ return {"output": rendered, "execution_time": execution_time}
+
+def execute_javascript(code, discussion_id, message_id):
+ return build_javascript_output(code)
\ No newline at end of file
diff --git a/utilities/execution_engines/mermaid_execution_engine.py b/utilities/execution_engines/mermaid_execution_engine.py
new file mode 100644
index 00000000..49b8ed05
--- /dev/null
+++ b/utilities/execution_engines/mermaid_execution_engine.py
@@ -0,0 +1,63 @@
+"""
+project: lollms_webui
+file: shell_execution_engine.py
+author: ParisNeo
+description:
+ This is a utility for executing python code
+
+"""
+from lollms_webui import LOLLMSWebUI
+from ascii_colors import get_trace_exception, trace_exception
+import time
+import subprocess
+import json
+
+lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance()
+
+def build_mermaid_output(code, ifram_name="unnamed"):
+ """
+ This function creates an HTML5 iframe with the given HTML content and iframe name.
+
+ Args:
+ html (str): The HTML content to be displayed in the iframe.
+ ifram_name (str, optional): The name of the iframe. Defaults to "unnamed".
+
+ Returns:
+ str: The HTML string for the iframe.
+ """
+ # Start the timer.
+ start_time = time.time()
+ rendered = "\n".join([
+ '