diff --git a/app/ui.py b/app/ui.py index 522bbd0..ed1b5fa 100644 --- a/app/ui.py +++ b/app/ui.py @@ -146,7 +146,7 @@ class MainWindow(QWidget): main_layout.addLayout(output_layout) self.setLayout(main_layout) - self.setWindowTitle('MollyAudit - created by yvling') + self.setWindowTitle('茉莉审计 - by yvling') screen = QGuiApplication.primaryScreen().geometry() window_width = 1000 window_height = 600 @@ -156,9 +156,13 @@ class MainWindow(QWidget): # 导出结果 export_button_layout = QHBoxLayout() + link_label = QLabel('联系作者:Github Email') + link_label.setOpenExternalLinks(True) + export_button_layout.addWidget(link_label) + export_button_layout.addStretch(1) + self.export_button = QPushButton('导出结果') self.export_button.clicked.connect(self.export_result) - export_button_layout.addStretch(1) # 添加伸缩项,使按钮靠右 export_button_layout.addWidget(self.export_button) main_layout.addLayout(export_button_layout) @@ -268,4 +272,4 @@ class MainWindow(QWidget): del os.environ['OPENAI_API_BASE'] if 'OPENAI_API_KEY' in os.environ: del os.environ['OPENAI_API_KEY'] - self.log.info('已终止代码审计流程') + self.log.critical('已终止代码审计流程') diff --git a/app/utils.py b/app/utils.py index 0a5dd21..ef71eca 100644 --- a/app/utils.py +++ b/app/utils.py @@ -3,5 +3,5 @@ import datetime def get_now_date(): now = datetime.datetime.now() - formatted = now.strftime("%Y-%m-%d %H:%M:%S") + formatted = now.strftime("%Y-%m-%d-%H-%M-%S") return formatted diff --git a/audit/__init__.py b/audit/__init__.py index d189872..9bc8d7d 100644 --- a/audit/__init__.py +++ b/audit/__init__.py @@ -26,7 +26,6 @@ class Audit: def __init__(self, base_url, api_key, reasoning_model, embedding_model, process_output_callback, result_output_callback): self.raw_chain = None self.source_files_list = [] - self.max_token = 4096 self.reasoning_model = reasoning_model self.embedding_model = embedding_model self.fortify_rules = FROTIFY_RULES diff --git a/build.bat b/build.bat new file mode 100644 index 0000000..d55635d --- /dev/null +++ b/build.bat @@ -0,0 +1 @@ +.\.venv\Scripts\pyinstaller.exe --hidden-import=tiktoken_ext.openai_public --hidden-import=tiktoken_ext --hidden-import=pydantic --hidden-import=pydantic-core --hidden-import=pydantic.deprecated.decorator --add-data "logo.ico;." -F -w -i logo.ico -n MollyAudit main.py \ No newline at end of file diff --git a/logo.ico b/logo.ico new file mode 100644 index 0000000..25e25d6 Binary files /dev/null and b/logo.ico differ diff --git a/main.py b/main.py index 6f8489a..0f86afa 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,6 @@ import sys +import os +from PyQt6.QtGui import QIcon from PyQt6.QtWidgets import QApplication from app import load_config from app.ui import MainWindow @@ -6,6 +8,13 @@ from app.ui import MainWindow app = QApplication(sys.argv) +def resource_path(relative_path): + try: + base_path = sys._MEIPASS + except AttributeError: + base_path = os.path.abspath(".") + return os.path.join(base_path, relative_path) + def main(): try: app.exec() @@ -19,6 +28,7 @@ if __name__ == '__main__': load_config() window = MainWindow() + window.setWindowIcon(QIcon(resource_path('logo.ico'))) window.show() sys.exit(main())