""" 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 from lollms.client_session import Client from lollms.utilities import discussion_path_to_url from pathlib import Path lollmsElfServer:LOLLMSWebUI = LOLLMSWebUI.get_instance() def build_graphviz_output(code, ifram_name=None): """ 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() if ifram_name is not None: rendered = "\n".join([ '
', f'', '
' ] ) else: with open(Path(__file__).parent/"assets"/"graphviz_container.html","r",encoding="utf-8") as f: data = f.read() rendered = data.replace("{{svg_data}}","\n".join([c.replace("'","\'") for c in code.split("\n") if c.strip()!=""]) ) execution_time = time.time() - start_time return {"output": rendered, "execution_time": execution_time} def execute_graphviz(code, client:Client, message_id, build_file=False): if build_file: # Start the timer. start_time = time.time() if not "http" in lollmsElfServer.config.host and not "https" in lollmsElfServer.config.host: host = "http://"+lollmsElfServer.config.host else: host = lollmsElfServer.config.host # Create a temporary file. root_folder = client.discussion.discussion_folder root_folder.mkdir(parents=True,exist_ok=True) tmp_file = root_folder/f"ai_code_{message_id}.html" with open(tmp_file,"w",encoding="utf8") as f: f.write(build_graphviz_output(code)["output"]) link = f"{host}:{lollmsElfServer.config.port}/{discussion_path_to_url(tmp_file)}" execution_time = time.time() - start_time output_json = {"output": f'Page built successfully
Press here to view the page', "execution_time": execution_time} return output_json else: return build_graphviz_output(code, ifram_name="iframe")