Extended CLI to support Bing's Chatbot and updated README

README.md

@@ -14,3 +14,15 @@ You need an API key for the official ChatGPT API.
 ### Unofficial API for Bing Chatbot
 
 This requires your Cookies to be exported to a json file. See [EdgeGPT's README](https://github.com/acheong08/EdgeGPT) for more infos.
+
+## CLI
+
+Using the official API via ChatGPT:
+```
+OPENAI_API_KEY="..." ./cli.py Hello
+```
+
+Using Bing's Chatbot:
+```
+BING_COOKIES_FILE="~/cookies.json" ./cli.py bing Hello
+```

cli.py

@@ -5,13 +5,22 @@
 from os import environ
 from sys import argv
 
-from chatgpt_pyapi.openai import ChatGPT, Message, Models
-
-# Read the API key from a environment variable
-API_KEY = environ["OPENAI_API_KEY"]
-
 if __name__ == "__main__":
-    # Create ChatGPT API instance
+    bing = False
-    cgpt = ChatGPT(API_KEY, model=Models.GPT_35_TURBO_0301)
+    if len(argv) > 2:
-    user_input = " ".join(argv[1:])
+        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)