Add awesome command to display all-awesome-prompts

Author: Simatwa

src/pytgpt/console.py

@@ -1109,6 +1109,11 @@ def generate(
         provider,
         quiet,
     )
+    if with_copied:
+        sep = "\n" if prompt else ""
+        prompt = prompt + sep + clipman.get()
+        assert prompt.strip() != sep, "No copied text found, issue prompt"
+
     if not prompt:
         # Let's try to read piped input
         import signal
@@ -1137,8 +1142,6 @@ def timeout_handler(signum, frame):
             # Just incase the previous timeout was not 0
 
     clear_history_file(filepath, new)
-    if with_copied:
-        prompt = prompt + "\n" + clipman.get()
     prompt = Optimizers.code(prompt) if code else prompt
     prompt = Optimizers.shell_command(prompt) if shell else prompt
     busy_bar.spin_index = busy_bar_index
@@ -1263,6 +1266,57 @@ def delete(name, case_sensitive, file):
     return AwesomePrompts().delete_prompt(name, case_sensitive)
 
 
+@awesome.command()
+@click.option(
+    "-j",
+    "--json",
+    is_flag=True,
+    help="Display prompts in json format",
+)
+@click.option(
+    "-i",
+    "--indent",
+    type=click.IntRange(1, 20),
+    help="Json format indentation level",
+    default=4,
+)
+@click.option(
+    "-x",
+    "--index",
+    is_flag=True,
+    help="Display prompts with their corresponding indexes",
+)
+@click.option("-c", "--color", help="Prompts stdout font color")
+@click.option("-o", "--output", type=click.Path(), help="Path to save the prompts")
+@click.help_option("-h", "--help")
+def all(json, indent, index, color, output):
+    """Stdout all awesome prompts"""
+    ap = AwesomePrompts()
+    awesome_prompts = ap.all_acts if index else ap.get_acts()
+    from json import dumps
+
+    formatted_awesome_prompts = dumps(awesome_prompts)
+    if json:
+        # click.secho(formatted_awesome_prompts, fg=color)
+        rich.print_json(formatted_awesome_prompts,indent=indent)
+
+    else:
+        awesome_table = Table(show_lines=True, title="All Awesome-Prompts")
+        awesome_table.add_column("index", justify="center", style="yellow")
+        awesome_table.add_column("Act Name/Index", justify="left", style="cyan")
+        awesome_table.add_column(
+            "Prompt",
+            style=color,
+        )
+        for index, key_value in enumerate(awesome_prompts.items()):
+            awesome_table.add_row(str(index), str(key_value[0]), key_value[1])
+        rich.print(awesome_table)
+
+    if output:
+        with open(output, "w") as fh:
+            fh.write(formatted_awesome_prompts)
+
+
 tgpt2_.add_command(
     webchatgpt, "webchatgpt"
 )  # Intergration with WebChatGPT https://github.com/Simatwa/WebChatGPT

src/pytgpt/utils.py

@@ -205,7 +205,7 @@ def __search_key(self, key: str, raise_not_found: bool = False) -> str:
         if raise_not_found:
             raise KeyError(f"Zero awesome prompt found with key - `{key}`")
 
-    def __get_acts(self):
+    def get_acts(self):
         """Retrieves all awesome-prompts"""
         with open(self.awesome_prompt_path) as fh:
             prompt_dict = json.load(fh)
@@ -225,7 +225,7 @@ def update_prompts_from_online(self, override: bool = False):
             response.raise_for_status
             resp.update(response.json())
             if os.path.isfile(self.awesome_prompt_path) and not override:
-                resp.update(self.__get_acts())
+                resp.update(self.get_acts())
             self.__is_prompt_updated = True
             with open(self.awesome_prompt_path, "w") as fh:
                 json.dump(resp, fh, indent=4)
@@ -243,9 +243,9 @@ def all_acts(self) -> dict:
         resp = {}
         if not os.path.isfile(self.awesome_prompt_path):
             self.update_prompts_from_online()
-        resp.update(self.__get_acts())
+        resp.update(self.get_acts())
 
-        for count, key_value in enumerate(self.__get_acts().items()):
+        for count, key_value in enumerate(self.get_acts().items()):
             # Lets map also index to the value
             resp.update({count: key_value[1]})
 
@@ -289,7 +289,7 @@ def add_prompt(self, name: str, prompt: str) -> bool:
         Returns:
             bool: is_successful report
         """
-        current_prompts = self.__get_acts()
+        current_prompts = self.get_acts()
         with open(self.awesome_prompt_path, "w") as fh:
             current_prompts[name] = prompt
             json.dump(current_prompts, fh, indent=4)
@@ -308,7 +308,7 @@ def delete_prompt(
             bool: is_successful report
         """
         name = self.__search_key(name, raise_not_found) if case_insensitive else name
-        current_prompts = self.__get_acts()
+        current_prompts = self.get_acts()
         is_name_available = (
             current_prompts[name] if raise_not_found else current_prompts.get(name)
         )