ChatGPT-PyAPI/cli.py

#!/usr/bin/env python3

# Copyright (c) 2023 Julian Müller (ChaoticByte)

from os import environ
from sys import argv

if __name__ == "__main__":
    bing = False
    if len(argv) > 2:
        bing = argv[1] == "bing"
    if bing:
        from chatgpt_pyapi.bing import ChatGPT, Message, ConversationStyle
        # Read the path to the cookies file from a environment variable
        BING_COOKIES_FILE = environ["BING_COOKIES_FILE"]
        # Create ChatGPT API instance
        cgpt = ChatGPT(BING_COOKIES_FILE, ConversationStyle.balanced)
        user_input = " ".join(argv[2:])
    else:
        from chatgpt_pyapi.openai import ChatGPT, Message, Models
        # Read the API key from a environment variable
        API_KEY = environ["OPENAI_API_KEY"]
        # Create ChatGPT API instance
        cgpt = ChatGPT(API_KEY, model=Models.GPT_35_TURBO_0301)
        user_input = " ".join(argv[1:])
    print(cgpt.chat(Message(user_input)).text)