main.py

import sys
import webbrowser
from argparse import ArgumentParser
from lib.client import Client
from lib.utils import check_for_updates, get_command
from colorama import Fore, Style
from pyperclip import copy as copy_to_clipboard
# https://stackoverflow.com/a/76833666
from rich.console import Console
from rich.markdown import Markdown

check_for_updates()

COMMAND_PREFIX = '!'

parser = ArgumentParser()
console = Console()
client = Client()

parser.add_argument('-o', '--open', '--web', action = 'store_true', help = 'Open the conversation in your default web browser', required = False)
parser.add_argument('-da', '--delete-all', action = 'store_true', help = 'Delete all conversations on your account', required = False)
parser.add_argument('-d', '--delete', action = 'store_true', help = 'Delete the conversation after getting the response', required = False)

args, remainder = parser.parse_known_args()
args = vars(args)

if args['delete_all']:
    if client.delete_all_conversations():
        print(f'{Fore.GREEN}Successfully deleted all conversations on your account!{Style.RESET_ALL}')
        sys.exit(0)
    else:
        print(f'{Fore.RED}Could not delete some conversations!{Style.RESET_ALL}')
        sys.exit(1)

cli_prompt = remainder[0] if remainder else None
uuid: str = client.create_new_chat()['uuid']
response: str = ''

if cli_prompt:
    if args['open']:
        webbrowser.open(f'https://claude.ai/chat/{uuid}', new = 0, autoraise = True)

    response = client.send_message(cli_prompt, uuid)

    if args['open']:
        sys.exit(0)

    if not response:
        print(f'{Fore.RED}Something went wrong!{Style.RESET_ALL}')
        client.delete_conversation(uuid)
        sys.exit(1)

    print()
    console.print(Markdown(f'{response}'))
    print()

    if args['delete']:
        client.delete_conversation(uuid)

    sys.exit(0)

print()
while True:
    prompt = str(input(f'{Fore.GREEN}>{Style.RESET_ALL} ')).strip()

    if not prompt:
        continue

    # execute commands
    if prompt.startswith(COMMAND_PREFIX):
        command = get_command(prompt, COMMAND_PREFIX)

        if command in ['quit', 'q', 'exit']:
            break

        elif command in ['open', 'o', 'web']:
            # https://stackoverflow.com/a/4217032
            webbrowser.open(f'https://claude.ai/chat/{uuid}', new = 0, autoraise = True)
            break

        elif command in ['cp', 'copy']:
            copy_to_clipboard(response)
            continue

        elif command in ['del', 'delete', 'remove', 'rm']:
            if client.delete_conversation(uuid):
                print(f'{Fore.GREEN}Successfully deleted the current conversation! Exiting...{Style.RESET_ALL}')
            else:
                print(f'{Fore.RED}Could not delete the current conversation! Aborting...{Style.RESET_ALL}')
            break

        else:
            pass

    response = client.send_message(prompt, uuid)

    if not response:
        print(f'{Fore.RED}Something went wrong!{Style.RESET_ALL}')
        client.delete_conversation(uuid)
        sys.exit(1)

    print()
    console.print(Markdown(f'{response}'))
    print()