From 9af486520c6978b6eabaab1bbdd6507204902c02 Mon Sep 17 00:00:00 2001
From: Flameguy677 <Flameguy677@gmail.com>
Date: Sun, 13 Aug 2023 01:29:57 +0530
Subject: [PATCH] Added "Show Time Elapsed" option

Enables the User to choose whether or not they see the time elapsed after each generation by configuring it in lollms-settings under "Main settings"
---
 lollms/apps/console/__init__.py | 8 +++++---
 lollms/terminal.py              | 6 ++++++
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/lollms/apps/console/__init__.py b/lollms/apps/console/__init__.py
index 935e133..96aae84 100644
--- a/lollms/apps/console/__init__.py
+++ b/lollms/apps/console/__init__.py
@@ -179,9 +179,10 @@ Participating personalities:
                 ump = self.config.discussion_prompt_separator +self.config.user_name+": " if self.config.use_user_name_in_discussions else self.personality.user_message_prefix
                 if self.config.use_user_name_in_discussions:
                     prompt = input(f"{ASCIIColors.color_green}{self.config.user_name}: {ASCIIColors.color_reset}")
-                    t0 = time.time() #Time at start of request
                 else:
                     prompt = input(f"{ASCIIColors.color_green}You: {ASCIIColors.color_reset}")
+                if self.config.show_time_elapsed:
+                    t0 = time.time() #Time at start of request
                 if prompt == "exit":
                     return
                 if prompt == "menu":
@@ -273,8 +274,9 @@ Participating personalities:
 
                 self.log(full_discussion)
                 
-                t1 = time.time() # Time at end of defense
-                print(f"{ASCIIColors.color_cyan}Response Time: {ASCIIColors.color_reset}",str(int((t1-t0)*1000)),"ms\n") # Total time elapsed since t0 in ms
+                if self.config.show_time_elapsed:
+                    t1 = time.time() # Time at end of response
+                    print(f"{ASCIIColors.color_cyan}Time Elapsed: {ASCIIColors.color_reset}",str(int((t1-t0)*1000)),"ms\n") # Total time elapsed since t0 in ms
             except KeyboardInterrupt:
                 print("Keyboard interrupt detected.\nBye")
                 break
diff --git a/lollms/terminal.py b/lollms/terminal.py
index b79a0d7..4068fe2 100644
--- a/lollms/terminal.py
+++ b/lollms/terminal.py
@@ -144,6 +144,7 @@ class MainMenu(Menu):
         self.show([
             {'name': 'Set user name', 'fn': self.set_user_name, 'help': "Sets the user name."},
             {'name': 'Set use user name in discussion', 'fn': self.set_use_user_name_in_discussions, 'help': "Sets the user name."},
+            {'name': 'Set show time elapsed', 'fn':self.set_time_elapsed, 'help': "Sets showing time taken for each complete response."}
         ])
 
     def set_user_name(self):
@@ -156,6 +157,11 @@ class MainMenu(Menu):
         self.lollms_app.config.use_user_name_in_discussions = self.yes_no_question('Use user name in dicsussion')
         self.lollms_app.config.save_config()
 
+    def set_time_elapsed(self):
+
+        self.lollms_app.config.show_time_elapsed = self.yes_no_question('Show time elapsed for each complete response')
+        self.lollms_app.config.save_config()
+
     def show_logo(self):
         print(f"{ASCIIColors.color_bright_yellow}")
         print("      ___       ___           ___       ___       ___           ___      ")