- Colorized intro prompt. interactive

Author: Simatwa

docs/CHANGELOG.md

@@ -150,4 +150,12 @@
 **What's new?**
 
 - Busy bar disabled when `--quiet` issued in *generate* mode. #12 Thanks to @johnd0e
-- `interactive` takes action on `$ pytgpt` otherwise one has to explictly declare the action. #11
+- `interactive` takes action on `$ pytgpt` otherwise one has to explictly declare the action. #11
+
+## v0.2.8
+
+**What's new?**
+
+- Auto-quiet on output redirection. Thanks to @johnd0e
+- Dropped support for output redirection. #12
+- Colorized command prompt. <kbd>interactive</kbd>

docs/README.md

@@ -8,7 +8,7 @@
 <a href="https://github.com/Simatwa/python-tgpt/actions/workflows/python-test.yml"><img src="https://github.com/Simatwa/python-tgpt/actions/workflows/python-test.yml/badge.svg" alt="Python Test"/></a>
 -->
 <a href="https://github.com/Simatwa/python-tgpt/blob/main/LICENSE"><img alt="License" src="https://img.shields.io/static/v1?logo=GPL&color=Blue&message=MIT&label=License"/></a>
-<a href="https://pypi.org/project/python-tgpt"><img alt="PyPi" src="https://img.shields.io/static/v1?logo=pypi&label=Pypi&message=0.2.7&color=green"/></a>
+<a href="https://pypi.org/project/python-tgpt"><img alt="PyPi" src="https://img.shields.io/static/v1?logo=pypi&label=Pypi&message=0.2.8&color=green"/></a>
 <a href="https://github.com/psf/black"><img alt="Black" src="https://img.shields.io/static/v1?logo=Black&label=Code-style&message=Black"/></a>
 <a href="#"><img alt="Passing" src="https://img.shields.io/static/v1?logo=Docs&label=Docs&message=Passing&color=green"/></a>
 <a href="https://github.com/Simatwa/python-tgpt/actions/workflows/python-package.yml"><img src="https://github.com/Simatwa/python-tgpt/actions/workflows/python-package.yml/badge.svg"/></a>

requirements.txt

@@ -5,4 +5,5 @@ clipman==3.1.0
 pyperclip==1.8.2
 appdirs==1.4.4
 webchatgpt==0.2.6
-GoogleBard==1.4.0
+GoogleBard==1.4.0
+colorama==0.4.6

setup.py

@@ -14,7 +14,7 @@
 
 setup(
     name="python-tgpt",
-    version="0.2.7",
+    version="0.2.8",
     license="MIT",
     author="Smartwa",
     maintainer="Smartwa",
@@ -45,6 +45,7 @@
         "appdirs==1.4.4",
         "webchatgpt==0.2.6",
         "GoogleBard==1.4.0",
+        "colorama==0.4.6",
     ],
     python_requires=">=3.9",
     keywords=[

src/pytgpt/__init__.py

@@ -1,6 +1,6 @@
 from .utils import appdir
 
-__version__ = "0.2.7"
+__version__ = "0.2.8"
 __author__ = "Smartwa"
 __repo__ = "https://github.com/Simatwa/python-tgpt"
 

src/pytgpt/console.py

@@ -12,6 +12,8 @@
 import re
 import sys
 import platform
+import datetime
+import time
 from time import sleep
 from threading import Thread as thr
 from functools import wraps
@@ -27,6 +29,10 @@
 from pytgpt.utils import default_path
 from pytgpt.utils import AwesomePrompts
 from WebChatGPT.console import chat as webchatgpt
+from colorama import Fore
+from colorama import init as init_colorama
+
+init_colorama(autoreset=True)
 
 getExc = lambda e: e.args[1] if len(e.args) > 1 else str(e)
 
@@ -207,7 +213,6 @@ def main(*args, **kwargs):
 
 class Main(cmd.Cmd):
     intro = f"Welcome to AI Chat in terminal. Type 'help' or 'h' for usage info \n Submit any bug at {pytgpt.__repo__}/issues/new"
-    prompt = f"╭─[{getpass.getuser().capitalize()}@TGPT](v{pytgpt.__version__})\n╰─>"
 
     def __init__(
         self,
@@ -360,6 +365,33 @@ def __init__(
         self.quiet = quiet
         self.vertical_overflow = "ellipsis"
         self.disable_stream = False
+        self.provider = provider
+        self.disable_coloring = False
+        self.__init_time = time.time()
+        self.__start_time = time.time()
+        self.__end_time = time.time()
+
+    @property
+    def prompt(self):
+        current_time = datetime.datetime.now().strftime("%H:%M:%S")
+        find_range = lambda start, end: round(end - start, 1)
+
+        if not self.disable_coloring:
+            return (
+                f"{Fore.LIGHTGREEN_EX}╭─[{Fore.CYAN}{getpass.getuser().capitalize()}@pyTGPT]{Fore.MAGENTA}({self.provider})"
+                f"{Fore.BLUE}~[{current_time}-"
+                f"{Fore.RED}T'{find_range(self.__init_time, time.time())}s,"
+                f"{Fore.YELLOW}L'{find_range(self.__start_time, self.__end_time)}s]"
+                f"\n╰─>{Fore.RESET}"
+            )
+        else:
+            return (
+                f"╭─[{getpass.getuser().capitalize()}@pyTGPT]({self.provider})"
+                f"~[{current_time}-"
+                f"T'{find_range(self.__init_time, time.time())}s,"
+                f"L'{find_range(self.__start_time, self.__end_time)}s]"
+                "\n╰─>"
+            )
 
     def output_bond(
         self,
@@ -625,6 +657,7 @@ def default(self, line, exit_on_error: bool = False):
         if line.startswith("./"):
             os.system(line[2:])
         else:
+            self.__start_time = time.time()
             try:
 
                 def generate_response():
@@ -677,6 +710,8 @@ def for_non_stream():
                 logging.error(getExc(e))
                 if exit_on_error:
                     sys.exit(1)
+            finally:
+                self.__end_time = time.time()
 
     def do_sys(self, line):
         """Execute system commands
@@ -854,6 +889,9 @@ def tgpt2_():
     is_flag=True,
     help="Postfix prompt with last copied text",
 )
+@click.option(
+    "-nc", "--no-coloring", is_flag=True, help="Disable intro prompt font-coloring"
+)
 @click.help_option("-h", "--help")
 def interactive(
     model,
@@ -881,6 +919,7 @@ def interactive(
     quiet,
     new,
     with_copied,
+    no_coloring,
 ):
     """Chat with AI interactively"""
     clear_history_file(filepath, new)
@@ -905,6 +944,7 @@ def interactive(
     busy_bar.spin_index = busy_bar_index
     bot.code_theme = code_theme
     bot.color = font_color
+    bot.disable_coloring = no_coloring
     bot.prettify = prettify
     bot.vertical_overflow = vertical_overflow
     bot.disable_stream = whole