Refactor file management and flag handling; update requirements for sentence-transformers

Signed-off-by: Shahm Najeeb <[email protected]>

Author: DefinetlyNotAI

CODE/logicytics/FileManagement.py

@@ -30,6 +30,7 @@ def open_file(file: str, use_full_path: bool = False) -> str | None:
                 subprocess.run(["start", file_path], shell=False)
             except Exception as e:
                 return f"Error opening file: {e}"
+        return None
 
     @staticmethod
     def mkdir():
@@ -145,6 +146,7 @@ def __remove_files(path: str, files: list) -> str | None:
                     os.remove(os.path.join(path, file))
                 except Exception as e:
                     return f"Error: {e}"
+            return None
 
         @staticmethod
         def __generate_sha256_hash(filename: str) -> str:

CODE/logicytics/Flag.py

@@ -53,7 +53,6 @@ def __get_sim(user_input: str, all_descriptions: list[str]) -> list[float]:
         """
         # Encode the current user input and historical inputs
         from sentence_transformers import SentenceTransformer, util
-
         import logging  # Suppress logging messages from Sentence Transformer due to verbosity
         # Set the logging level based on the debug mode, either DEBUG or ERROR (aka only important messages)
         if DEBUG_MODE:
@@ -211,14 +210,14 @@ def _generate_summary_and_graph(cls):
                 log.info("\nFlag Usage Summary Graph saved in current working directory as 'Flag_usage_summary.png'")
 
     @staticmethod
-    def load_history() -> dict[str, any]:
+    def load_history() -> dict:
         """
         Load user interaction history from a gzipped JSON file.
         
         This method attempts to read and parse historical interaction data from a compressed JSON file. If the file is not found, it returns an empty history structure with an empty interactions dictionary and a zero-initialized flags usage counter.
         
         Returns:
-            dict[str, any]: A dictionary containing:
+            dict: A dictionary containing:
                 - 'interactions': A dictionary of past user interactions
                 - 'flags_usage': A Counter object tracking flag usage frequencies
         
@@ -233,7 +232,7 @@ def load_history() -> dict[str, any]:
             return {'interactions': {}, 'flags_usage': Counter()}
 
     @staticmethod
-    def save_history(history_data: dict[str, any]):
+    def save_history(history_data: dict):
         """
         Save user interaction history to a gzipped JSON file.
         

CODE/packet_sniffer.py

@@ -112,8 +112,8 @@ def sniff_packets(self, iface: str, count: int, timeout: int, retry_max: int):
                 )
                 log.info("Sniff complete.")
                 break
-            except Exception as e:
-                log.warning(f"Sniff failed on {iface}: {e}")
+            except Exception as err:
+                log.warning(f"Sniff failed on {iface}: {err}")
                 iface = self._correct_interface(iface)
         else:
             log.error("Max retry time exceeded.")

requirements.txt

@@ -16,4 +16,5 @@ matplotlib~=3.10.1
 networkx~=3.2.1
 pandas~=2.2.3
 cryptography~=44.0.2
-scapy~=2.5.0
+scapy~=2.5.0
+sentence-transformers~=5.0.0