Merge pull request #601 from hpsj/main

Automatically obtain restNonce

gpt4free/hpgptai/__init__.py

@@ -5,20 +5,26 @@
 @File ：__init__.py.py
 @IDE ：PyCharm
 """
+import re
 import json
-import requests
+import base64
 import random
 import string
+import requests
+from fake_useragent import UserAgent
+
 
 class ChatCompletion:
     @staticmethod
     def create(
             messages: list,
-            context: str="Converse as if you were an AI assistant. Be friendly, creative.",
-            restNonce:str="9d6d743bd3",
-            proxy:str=None
+            context: str = "Converse as if you were an AI assistant. Be friendly, creative.",
+            restNonce: str = None,
+            proxy: str = None
     ):
         url = "https://chatgptlogin.ac/wp-json/ai-chatbot/v1/chat"
+        if not restNonce:
+            restNonce = ChatCompletion.get_restNonce(proxy)
         headers = {
             "Content-Type": "application/json",
             "X-Wp-Nonce": restNonce
@@ -27,7 +33,7 @@ class ChatCompletion:
         data = {
             "env": "chatbot",
             "session": "N/A",
-            "prompt": ChatCompletion.__build_prompt(context,messages),
+            "prompt": ChatCompletion.__build_prompt(context, messages),
             "context": context,
             "messages": messages,
             "newMessage": messages[-1]["content"],
@@ -48,7 +54,6 @@ class ChatCompletion:
             return res.json()
         return res.text
 
-
     @staticmethod
     def randomStr():
         return ''.join(random.choices(string.ascii_lowercase + string.digits, k=34))[:11]
@@ -66,12 +71,26 @@ class ChatCompletion:
         prompt += '\n' + "AI: "
         return prompt
 
-
+    @classmethod
+    def get_restNonce(cls, proxy: str = None):
+        url = "https://chatgptlogin.ac/"
+        headers = {
+            "Referer": "https://chatgptlogin.ac/",
+            "User-Agent": UserAgent().random
+        }
+        proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else None
+        res = requests.get(url, headers=headers, proxies=proxies)
+        src = re.search(
+            'class="mwai-chat mwai-chatgpt">.*<span>Send</span></button></div></div></div> <script defer src="(.*?)">',
+            res.text).group(1)
+        decoded_string = base64.b64decode(src.split(",")[-1]).decode('utf-8')
+        restNonce = re.search(r"let restNonce = '(.*?)';", decoded_string).group(1)
+        return restNonce
 
 
 class Completion:
     @staticmethod
-    def create(prompt: str,proxy:str):
+    def create(prompt: str, proxy: str):
         messages = [
             {
                 "content": prompt,
@@ -81,4 +100,4 @@ class Completion:
                 "who": "User: ",
             },
         ]
-        return ChatCompletion.create(messages=messages,proxy=proxy)
+        return ChatCompletion.create(messages=messages, proxy=proxy)