Merge pull request #99 from SaintsSec/vincent/history2

[ADD] Command history

install/requirements.txt

@@ -13,3 +13,4 @@ image
 hashcat
 mido
 psutil
+prompt_toolkit==3.0.48

navi.py

@@ -63,4 +63,4 @@ def get_parameters(input_str: str) -> list[str]:
     # Flatten the list and filter out empty strings
     parameters = [item for sublist in matches for item in sublist if item]
 
-    return parameters
+    return parameters

navi_internal.py

@@ -9,6 +9,9 @@
 import spacy
 import platform
 
+from prompt_toolkit import PromptSession
+from prompt_toolkit.history import FileHistory
+
 from mods import mods
 
 
@@ -21,6 +24,9 @@ class NaviApp:
     server: str = config.server
     port: int = config.port
 
+    script_dir = os.path.dirname(os.path.abspath(__file__))
+    hist_file = os.path.join(script_dir, ".navi_history")
+
     # NLP setup
     nlp: spacy.language.Language = spacy.load("en_core_web_sm")
     ruler: spacy.pipeline.EntityRuler = nlp.add_pipe("entity_ruler")
@@ -34,6 +40,9 @@ def __new__(cls, *args, **kwargs):
             cls._instance = super(NaviApp, cls).__new__(cls, *args, **kwargs)
         return cls._instance
 
+    def setup_history(self) -> None:
+        self.session = PromptSession(history=FileHistory(self.hist_file))
+
     def get_user(self) -> str:
         return self.user
 
@@ -145,7 +154,7 @@ def chat_with_navi(self) -> None:
         while True:
             # Get user input
             try:
-                user_message = input(f"\n{self.user}> ")
+                user_message = self.session.prompt(f"\n{self.user}> ")
             except EOFError:
                 self.print_message("Encountered an unexpected end of input.")
                 break

navi_shell.py

@@ -77,6 +77,7 @@ def main() -> None:
         navi_instance.setup_navi_vocab()
         navi_instance.set_user(user)
         navi_instance.clear_terminal()
+        navi_instance.setup_history()
         navi_instance.chat_with_navi()
         navi_instance.print_message(f"How can I help you {user}")
     except KeyboardInterrupt: