cli-gpt.sh

#!/bin/sh

# ======================================================== #
# ======================== cli gpt ======================= #
# ======================================================== #

DEFAULT_SAVE_FILEPATH="$HOME/chat-export.md"
PROMPTS_DIR="$HOME/.config/cli-gpt/system-prompts"

# check if prompts dir exists
if [ ! -d "$PROMPTS_DIR" ]; then
    mkdir -p "$PROMPTS_DIR"
    touch "$PROMPTS_DIR/default.txt"
    echo "You are a helpful assistant. The user is interacting with you through the command line." > "$PROMPTS_DIR/default.txt"
fi

PROMPTS=()

# load all prompts (= filename without extension)
for file in "$PROMPTS_DIR"/*.txt; do
    filename=$(basename -- "$file")
    filename="${filename%.*}"
    PROMPTS+=("$filename")
done

messages="[]"
num_messages=0
last_message_type=""

# ~~~~~~~~~~~~~~~ get flags ~~~~~~~~~~~~~~~ #

SYS_PROMPT_FLAG="-s"
SYS_PROMPT_FLAG_LONG="--system-prompt"
SYS_PROMPT_FLAG_VALUE="default"

LIST_SYS_PROMPTS_FLAG="-ls"
LIST_SYS_PROMPTS_FLAG_LONG="--list-system-prompts"

START_PROMPT_FLAG="-p"
START_PROMPT_FLAG_LONG="--prompt"
START_PROMPT_FLAG_VALUE=""

MODEL_FLAG="-m"
MODEL_FLAG_LONG="--model"
MODEL_FLAG_VALUE="gpt-3.5-turbo"

HELP_FLAG="-h"
HELP_FLAG_LONG="--help"

get_readme() {
    curl -s https://raw.githubusercontent.com/AlexW00/cli-gpt/master/commands.md
}

show_help() {
    get_readme | gum pager --soft-wrap 
}

for arg in "$@"
do
    case $arg in
        $SYS_PROMPT_FLAG=*|$SYS_PROMPT_FLAG_LONG=*)
            if [[ ! " ${PROMPTS[*]} " =~ " ${arg#*=} " ]]; then
                echo "Invalid system prompt name: ${arg#*=} (valid prompts: ${PROMPTS[*]})"
                exit 1
            fi
            SYS_PROMPT_FLAG_VALUE="${arg#*=}"
            shift # Remove --prompt= from processing
        ;;

        "$LIST_SYS_PROMPTS_FLAG"|"$LIST_SYS_PROMPTS_FLAG_LONG")
            echo "Available system prompts: ${PROMPTS[*]}"
            exit 0
        ;;

        "$MODEL_FLAG"=*|"$MODEL_FLAG_LONG"=*)
            MODEL_FLAG_VALUE="${arg#*=}"
            shift # Remove --model= from processing
        ;;

        "$START_PROMPT_FLAG"=*|"$START_PROMPT_FLAG_LONG"=*)
            START_PROMPT_FLAG_VALUE="${arg#*=}"
            shift # Remove --prompt= from processing
        ;;

        "$HELP_FLAG"|"$HELP_FLAG_LONG")
            show_help
            exit 0
        ;;

        *)
            # unknown option
            echo "Unknown option: '$arg'; see help with -h or --help for available options."
            exit 1
        ;;
    esac    
done


# ~~~~~~~~~~~ validate env vars ~~~~~~~~~~~ #

if [ -z "$OPENAI_API_KEY" ]; then
    echo "Please set OPENAI_API_KEY environment variable."
    exit 1
fi

# ~~~~~~~~~ validate dependencies ~~~~~~~~~ #

if ! command -v gum &> /dev/null
then
    echo "gum could not be found. Please install gum from https://github.com/charmbracelet/gum"
    exit
fi

# if linux
if [ "$(uname)" = "Linux" ]; then
    # if xclip is not installed
    if ! command -v xclip &> /dev/null
    then
        echo "xclip could not be found. Please install xclip."
        exit
    fi
fi

# ~~~~~~~~~~~~~~~~~~ code ~~~~~~~~~~~~~~~~~ #

push_message() {
    role="$1"
    content="$2"
    # escape content for json format
    content=$(echo "$content" | sed -e 's/"/'\''/g')
    messages=$(echo "$messages" | jq ". += [{\"role\": \"$role\", \"content\": \"$content\"}]")
    num_messages=$((num_messages+1))
}

get_placeholder() {
    # if the chat is empty, return "Your message"
    if [ "$num_messages" -eq 1 ]; then
        echo "Your message (:h for help)"
    else
        echo "Another message (:q to quit, :h for help)"
    fi
}


prompt_user_message() {
    content=""

    if [ -n "$START_PROMPT_FLAG_VALUE" ]; then
        content="$START_PROMPT_FLAG_VALUE"
        START_PROMPT_FLAG_VALUE=""
    else
        content=$(gum input --placeholder "$(get_placeholder)")
    fi

    # if message is empty, q, exit, or ctrl-c
    if [ -z "$content" ] || [ "$content" = ":q" ] || [ "$content" = "exit" ]; then
        last_message_type="exit"
    elif [ "$content" = ":h" ]; then
        last_message_type="help"
    elif [ "$content" = ":s" ]; then
        last_message_type="save"
    elif [ "$content" = ":c" ]; then
        last_message_type="copy"
    else
        push_message "user" "$content"
        last_message_type="message"
    fi
}

parse_chatgpt_response(){
    # return content of first choice and replace newlines with spaces, double quotes with single quotes
    echo "$1" | jq -r '.choices[0].message.content' | sed -e 's/\\n/ /g' | sed -e 's/"/'\''/g'
}

get_chatgpt_response () {
    # only show the result of the curl command
    gum spin --spinner points --show-output --title "Responding..." -- curl -s https://api.openai.com/v1/chat/completions \
    -H "Authorization: Bearer $OPENAI_API_KEY" \
    -H "Content-Type: application/json" \
    -d "{
    \"model\": \"$MODEL_FLAG_VALUE\",
    \"messages\": $messages
    }"
}

respond(){
    # get response from chatgpt
    response=$(get_chatgpt_response)
    # parse response
    content=$(parse_chatgpt_response "$response")
    # push message
    push_message "assistant" "$content"
}

messages_to_md() {
    md=""
    # loop through messages
    for row in $(echo "${messages}" | jq -r '.[] | @base64'); do
        _jq() {
            echo "${row}" | base64 --decode | jq -r "${1}"
        }
        role=$(_jq '.role')
        # if role != system
        if [ "$role" != "system" ]; then
            content=$(_jq '.content')
            md+="[$role]:\n"
            md+="$content\n\n"
        fi

    done

    echo -e "$md"
}

save_chat() {
    # get filepath
    filepath=$(gum input --placeholder "$DEFAULT_SAVE_FILEPATH")
    # if filepath is empty, use default
    if [ -z "$filepath" ]; then
        filepath="$DEFAULT_SAVE_FILEPATH"
    fi
    # save chat
    messages_to_md > "$filepath"
    # notify user
    echo "Chat saved to $filepath"
}

copy_to_clipboard() {
    str="$1"
    # on linux, use xclip
    if [ "$(uname)" = "Linux" ]; then
        echo "$str" | xclip -selection clipboard
    # on mac, use pbcopy
    elif [ "$(uname)" = "Darwin" ]; then
        echo "$str" | pbcopy
    fi

    echo "Last message copied to clipboard"
}

copy_last_message() {
    # get last message
    last_message=$(echo "$messages" | jq -r '.[-1].content')
    # copy last message to clipboard using xclip
    copy_to_clipboard "$last_message"
}

show_messages() {
    messages_to_md | gum pager --soft-wrap
}

get_prompt_path() {
    PROMPT_NAME="$1"
    echo "$PROMPTS_DIR/$PROMPT_NAME.txt"
}

get_prompt() {
    PROMPT_NAME="$1"
    cat "$(get_prompt_path "$PROMPT_NAME")"
}

push_message "system" "$(get_prompt "$SYS_PROMPT_FLAG_VALUE")"


while true; do
    # get user message
    prompt_user_message

    if [ "$last_message_type" = "message" ]; then
        respond
        show_messages
    elif [ "$last_message_type" = "help" ]; then
        show_help
    elif [ "$last_message_type" = "save" ]; then
        save_chat
    elif [ "$last_message_type" = "copy" ]; then
        copy_last_message
    else
        exit
    fi

done