initial commit
commit
924b1e593c
@ -0,0 +1,43 @@
|
||||
# Awesome ChatGPT Prompts CLI
|
||||
|
||||
This is a command line tool that allows you to select a prompt to use with
|
||||
ChatGPT from the [Awesome ChatGPT
|
||||
Prompts](https://raw.githubusercontent.com/f/awesome-chatgpt-prompts)
|
||||
|
||||
|
||||
## Installation
|
||||
|
||||
To install the dependencies needed to run this program, run:
|
||||
|
||||
`pip install -r requirements.txt`
|
||||
|
||||
You will also need to have `fzf` locally installed.
|
||||
|
||||
## Usage
|
||||
|
||||
To use this program, run:
|
||||
|
||||
`cgpt.py [COMMAND]`
|
||||
|
||||
If no command is provided, it will present you with a list of prompts that you
|
||||
can choose from. Use the up and down arrow keys to navigate the list and the
|
||||
Enter key to select a prompt.
|
||||
|
||||
You can also use the following commands:
|
||||
|
||||
- `add`: Add a custom prompt to the list.
|
||||
- `delete`: Delete a custom prompt from the list.
|
||||
- `update`: Update the list of prompts from the remote repository.
|
||||
|
||||
## Features/TODO
|
||||
|
||||
- [x] Update the list of prompts from the remote repository.
|
||||
- [x] Add support for custom prompts
|
||||
- [ ] Create a pull request to add custom prompts to the
|
||||
remote repository.
|
||||
|
||||
## Credits
|
||||
The prompts in this tool are taken from the [awesome-chatgpt-prompts](https://github.com/f/awesome-chatgpt-prompts) repository.
|
||||
|
||||
|
||||
|
@ -0,0 +1,143 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
|
||||
import os
|
||||
import sys
|
||||
import requests
|
||||
import csv
|
||||
|
||||
# We will need to import the `click` package to build the CLI.
|
||||
import click
|
||||
|
||||
# We will also need to import the `fzf` package to use the `fzf` command.
|
||||
from fzf import Fzf, fzf
|
||||
|
||||
XDG_DATA_HOME = os.environ.get('XDG_DATA_HOME', os.path.expanduser('~/.local/share'))
|
||||
PROMPTS_URL = "https://raw.githubusercontent.com/f/awesome-chatgpt-prompts/main/prompts.csv"
|
||||
CACHE_LOCATION = os.path.join(XDG_DATA_HOME, "chatgpt-prompts", "prompts.csv")
|
||||
PROMPTS_REPO_URL = "https://github.com/f/awesome-chatgpt-prompts"
|
||||
REPO_PROMPTS_RELPATH = "prompts.csv"
|
||||
|
||||
# save custom prompts to my-prompts.csv
|
||||
CUSTOM_PROMPTS = os.path.join(XDG_DATA_HOME, "chatgpt-prompts", "my-prompts.csv")
|
||||
|
||||
def cache_prompts(update=False):
|
||||
'''
|
||||
if the prompts are not cached, download them and cache them locally
|
||||
'''
|
||||
if not os.path.exists(CACHE_LOCATION) or update:
|
||||
print("Downloading remote prompts...")
|
||||
os.makedirs(os.path.dirname(CACHE_LOCATION), exist_ok=True)
|
||||
with open(CACHE_LOCATION, "w") as f:
|
||||
f.write(requests.get(PROMPTS_URL).text)
|
||||
f.close()
|
||||
|
||||
def load_prompts(custom_only=False):
|
||||
'''
|
||||
Loads the prompts from the cache. if the cache is empty, download the prompts.
|
||||
The prompts csv contains two columns separated by a comma in the form "act", "prompt"
|
||||
Returns the list of prompts as tuples
|
||||
'''
|
||||
|
||||
cache_prompts()
|
||||
prompts = {}
|
||||
|
||||
if not custom_only:
|
||||
with open(CACHE_LOCATION, "r") as f:
|
||||
prompts = f.readlines()
|
||||
|
||||
# load the prompts into a list of tuples
|
||||
prompts = [tuple(prompt.strip().split(",")) for prompt in prompts]
|
||||
|
||||
# transform into a dict of acts keys and prompt values
|
||||
prompts = {prompt[0]: prompt[1] for prompt in prompts[1:]}
|
||||
|
||||
f.close()
|
||||
|
||||
with open(CUSTOM_PROMPTS, "r") as f:
|
||||
custom_prompts = f.readlines()
|
||||
custom_prompts = [tuple(prompt.strip().split(",")) for prompt in custom_prompts]
|
||||
custom_prompts = {prompt[0]: prompt[1] for prompt in custom_prompts[1:]}
|
||||
prompts.update(custom_prompts)
|
||||
f.close()
|
||||
|
||||
|
||||
return prompts
|
||||
|
||||
|
||||
|
||||
# save custom prompts to CUSTOM_PROMPTS
|
||||
def save_prompt(act, prompt):
|
||||
# if custom prompts is empty initialize it with a the csv header
|
||||
if not os.path.exists(CUSTOM_PROMPTS):
|
||||
with open(CUSTOM_PROMPTS, "w") as f:
|
||||
header = '"act","prompt"\n'
|
||||
f.write(header)
|
||||
f.close()
|
||||
|
||||
with open(CUSTOM_PROMPTS, "a") as f:
|
||||
f.write(f'"{act}","{prompt}"\n')
|
||||
f.close()
|
||||
|
||||
|
||||
@click.group()
|
||||
def cli():
|
||||
pass
|
||||
|
||||
# select and return a tuple (act, prompt)
|
||||
def choose_prompt(custom_only=False):
|
||||
prompts = load_prompts(custom_only)
|
||||
|
||||
# load custom prompts
|
||||
|
||||
chooser = Fzf()
|
||||
header = "Select an act"
|
||||
preview = "echo {}"
|
||||
preview_window = "right:50%"
|
||||
act = chooser.prompt(prompts.keys(), header=header, preview=preview, preview_window=preview_window)
|
||||
prompt = prompts.get(act[0])
|
||||
|
||||
return (act[0], prompt)
|
||||
|
||||
def show_prompts():
|
||||
act, prompt = choose_prompt()
|
||||
print(prompt)
|
||||
|
||||
|
||||
# update refreshes the prompts from the remote repo
|
||||
@click.command()
|
||||
def update():
|
||||
cache_prompts(update=True)
|
||||
|
||||
# We will create a function called `pr` that will be executed when the `pr`
|
||||
# command is provided.
|
||||
@click.command()
|
||||
def pr():
|
||||
print("[TODO] Opening a PR to add your prompt to the prompts repo")
|
||||
|
||||
|
||||
|
||||
# The add command allows the user to define a new prompt by providing the act and the prompt
|
||||
@click.command()
|
||||
def add():
|
||||
act = input("Enter the act title: ")
|
||||
prompt = input("Enter the prompt: ")
|
||||
save_prompt(act, prompt)
|
||||
|
||||
|
||||
# We will add the `add`, `update`, and `pr` functions to the `show_prompts`
|
||||
# function.
|
||||
cli.add_command(add)
|
||||
cli.add_command(update)
|
||||
cli.add_command(pr)
|
||||
|
||||
# We will execute the `show_prompts` function if the script is executed directly
|
||||
# without any command provided
|
||||
if __name__ == "__main__":
|
||||
# if no command provided call show_prompts
|
||||
if len(sys.argv) == 1:
|
||||
show_prompts()
|
||||
|
||||
else:
|
||||
cli()
|
||||
|
Loading…
Reference in New Issue