diff --git a/lollms/code_parser.py b/lollms/code_parser.py index 6d60d8a..ccd696f 100644 --- a/lollms/code_parser.py +++ b/lollms/code_parser.py @@ -1,6 +1,7 @@ import re from lollms.utilities import PackageManager - +from pipmaster import PackageManager +from bs4 import BeautifulSoup def compress_js(js_code): # Patterns to match function, class, and variable declarations function_pattern = r"function\s+(\w+)\s*\(([^)]*)\)" diff --git a/lollms/personality.py b/lollms/personality.py index 022845c..072bdbb 100644 --- a/lollms/personality.py +++ b/lollms/personality.py @@ -38,7 +38,7 @@ from lollms.com import LoLLMsCom from lollms.helpers import trace_exception from lollms.utilities import PackageManager -from lollms.code_parser import compress_js, compress_python +from lollms.code_parser import compress_js, compress_python, compress_html import requests @@ -3274,6 +3274,8 @@ fetch('/open_file', { return compress_js(code) def compress_python(self, code): return compress_python(code) + def compress_html(self, code): + return compress_html(code) class AIPersonalityInstaller: def __init__(self, personality:AIPersonality) -> None: diff --git a/requirements.txt b/requirements.txt index 14dd05f..456cb68 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,8 +5,11 @@ wget setuptools requests + safe_store +pipmaster ascii_colors>=0.1.3 +beautifulsoup4 autopep8