From 6db0081fab08176192ce61c876758bd1c29922a3 Mon Sep 17 00:00:00 2001 From: WhatDidYouExpect <89535984+WhatDidYouExpect@users.noreply.github.com> Date: Mon, 7 Jul 2025 17:17:44 +0200 Subject: [PATCH] kill me --- modules/markovmemory.py | 10 ---------- modules/prestartchecks.py | 9 +++++++++ modules/sentenceprocessing.py | 8 +++++--- 3 files changed, 14 insertions(+), 13 deletions(-) diff --git a/modules/markovmemory.py b/modules/markovmemory.py index f13b52d..eae003e 100644 --- a/modules/markovmemory.py +++ b/modules/markovmemory.py @@ -26,16 +26,6 @@ def load_memory(): except FileNotFoundError: pass - # If MEMORY_LOADED_FILE does not exist, load default data and mark as loaded - if not os.path.exists(MEMORY_LOADED_FILE): - try: - with open(DEFAULT_DATASET_FILE, "r") as f: - default_data = json.load(f) - data.extend(default_data) - except FileNotFoundError: - pass - with open(MEMORY_LOADED_FILE, "w") as f: - f.write("Data loaded") return data # Save memory data to MEMORY_FILE diff --git a/modules/prestartchecks.py b/modules/prestartchecks.py index f5ce96a..171a39a 100644 --- a/modules/prestartchecks.py +++ b/modules/prestartchecks.py @@ -8,6 +8,7 @@ import sysconfig import ast import json import re +from spacy.util import is_package import importlib.metadata # import shutil @@ -19,6 +20,13 @@ except ImportError: psutilavaliable = False print(RED, _('missing_requests_psutil'), RESET) +def check_for_model(): + if is_package("en_core_web_sm"): + print("Model is installed.") + else: + print("Model is not installed.") + + def iscloned(): if os.path.exists(".git"): return True @@ -260,6 +268,7 @@ def start_checks(): print(f"{YELLOW}{(_('checks_disabled'))}{RESET}") return print(_('running_prestart_checks')) + check_for_model() iscloned() check_missing_translations() check_requirements() diff --git a/modules/sentenceprocessing.py b/modules/sentenceprocessing.py index f073210..c470aef 100644 --- a/modules/sentenceprocessing.py +++ b/modules/sentenceprocessing.py @@ -5,9 +5,7 @@ from modules.volta.main import _ import spacy from spacy.tokens import Doc from spacytextblob.spacytextblob import SpacyTextBlob -nlp = spacy.load("en_core_web_sm") -nlp.add_pipe("spacytextblob") -Doc.set_extension("polarity", getter=lambda doc: doc._.blob.polarity) + def check_resources(): try: @@ -22,6 +20,10 @@ def check_resources(): check_resources() +nlp = spacy.load("en_core_web_sm") +nlp.add_pipe("spacytextblob") +Doc.set_extension("polarity", getter=lambda doc: doc._.blob.polarity) + def is_positive(sentence): doc = nlp(sentence) sentiment_score = doc._.polarity # from spacytextblob