Feat: Reset conversation

README.md

@@ -1 +1,84 @@
-# Hello world
+# Bing GPT
+ChatGPT with internet access
+
+## Requirements
+- A Microsoft Account with early access to http://bing.com/chat
+- Microsoft Edge
+
+## Setup
+### Checking access
+- Install the latest version of Microsoft Edge
+- Open http://bing.com/chat
+- If you see a chat feature, you are good to go
+
+### Getting authentication
+- Open the developer tools (F12)
+- Go to the Application tab → Storage → Cookies
+- Find the cookie named "_U"
+- Copy the value of the cookie
+- Method 1
+  - `export BING_U="<COOKIE_VALUE>"`
+- Method 2
+  - Use it as command line argument later
+
+## Installation
+- `python3 -m pip install BingGPT`
+
+## Demo usage
+- If `BING_U` in environment variables: `python3 -m BingGPT`
+- Else: `python3 -m BingGPT "<COOKIE_VALUE>"`
+
+## Developer
+Use Async for the best experience
+
+```python
+import asyncio
+from BingGPT import Chatbot
+
+async def main():
+    """
+    Main function
+    """
+    print("Initializing...")
+    bot = Chatbot()
+    await bot.start()
+    while True:
+        prompt = input("\nYou:\n")
+        if prompt == "!exit":
+            break
+        elif prompt == "!help":
+            print("""
+            !help - Show this help message
+            !exit - Exit the program
+            !reset - Reset the conversation
+            """)
+            continue
+        elif prompt == "!reset":
+            await bot.reset()
+            continue
+        print("Bot:")
+        print((await bot.ask(prompt=prompt))["item"]["messages"][1]["text"])
+    await bot.close()
+
+
+if __name__ == "__main__":
+    print(
+        """
+        BingGPT - A demo of reverse engineering the Bing GPT chatbot
+        Repo: github.com/acheong08/BingGPT
+        By: Antonio Cheong
+
+        !help for help
+
+        Type !exit to exit
+        Enter twice to send message
+    """
+    )
+    asyncio.run(main())
+
+
+```
+
+## Work in progress
+- Response streaming (Easily achievable)
+- Error handling

example.env

@@ -0,0 +1 @@
+export BING_U="<COOKIE_VALUE>"

src/BingGPT.py

@@ -5,6 +5,7 @@ import asyncio
 import json
 import os
 import uuid
+import sys
 
 import requests
 import websockets.client as websockets
@@ -113,7 +114,7 @@ class Conversation:
         }
         # Create cookies
         cookies = {
-            "_U": os.environ.get("BING_U"),
+            "_U": os.environ.get("BING_U") or sys.argv[1],
         }
         # Send GET request
         response = requests.get(
@@ -201,7 +202,7 @@ class Chatbot:
         self.conversation: Conversation
         self.chat_hub: ChatHub
 
-    async def a_start(self) -> None:
+    async def start(self) -> None:
         """
         Separate initialization to allow async
         """
@@ -209,17 +210,25 @@ class Chatbot:
         self.chat_hub = ChatHub()
         await self.chat_hub.init(conversation=self.conversation)
 
-    async def a_ask(self, prompt: str) -> str:
+    async def ask(self, prompt: str) -> str:
         """
         Ask a question to the bot
         """
         return await self.chat_hub.ask(prompt=prompt)
 
-    async def a_close(self):
+    async def close(self):
         """
         Close the connection
         """
         await self.chat_hub.close()
+    
+    async def reset(self):
+        """
+        Reset the conversation
+        """
+        await self.close()
+        await self.start()
+    
 
 
 def get_input(prompt):
@@ -252,14 +261,24 @@ async def main():
     """
     print("Initializing...")
     bot = Chatbot()
-    await bot.a_start()
+    await bot.start()
     while True:
         prompt = get_input("\nYou:\n")
         if prompt == "!exit":
             break
+        elif prompt == "!help":
+            print("""
+            !help - Show this help message
+            !exit - Exit the program
+            !reset - Reset the conversation
+            """)
+            continue
+        elif prompt == "!reset":
+            await bot.reset()
+            continue
         print("Bot:")
-        print((await bot.a_ask(prompt=prompt))["item"]["messages"][1]["text"])
-    await bot.a_close()
+        print((await bot.ask(prompt=prompt))["item"]["messages"][1]["text"])
+    await bot.close()
 
 
 if __name__ == "__main__":
@@ -269,6 +288,8 @@ if __name__ == "__main__":
         Repo: github.com/acheong08/BingGPT
         By: Antonio Cheong
 
+        !help for help
+
         Type !exit to exit
         Enter twice to send message
     """