1 files changed, 43 insertions, 106 deletions

diff --git a/g4f/Provider/ChatgptLogin.py b/g4f/Provider/ChatgptLogin.py
index e4584d32..8b868f8e 100644
--- a/g4f/Provider/ChatgptLogin.py
+++ b/g4f/Provider/ChatgptLogin.py
@@ -1,70 +1,58 @@
 from __future__ import annotations
 
-import base64
-import os
-import re
+import os, re
+from aiohttp import ClientSession
 
-import requests
+from .base_provider import AsyncProvider, format_prompt
 
-from ..typing import Any, CreateResult
-from .base_provider import BaseProvider
 
-
-class ChatgptLogin(BaseProvider):
+class ChatgptLogin(AsyncProvider):
     url                   = "https://opchatgpts.net"
     supports_gpt_35_turbo = True
     working               = True
+    _nonce                = None
 
-    @staticmethod
-    def create_completion(
+    @classmethod
+    async def create_async(
+        cls,
         model: str,
         messages: list[dict[str, str]],
-        stream: bool, **kwargs: Any) -> CreateResult:
-        
+        **kwargs
+    ) -> str:
         headers = {
-            "authority"          : "chatgptlogin.ac",
-            "accept"             : "*/*",
-            "accept-language"    : "en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3",
-            "content-type"       : "application/json",
-            "origin"             : "https://opchatgpts.net",
-            "referer"            : "https://opchatgpts.net/chatgpt-free-use/",
-            "sec-ch-ua"          : '"Chromium";v="116", "Not)A;Brand";v="24", "Google Chrome";v="116"',
-            "sec-ch-ua-mobile"   : "?0",
-            "sec-ch-ua-platform" : '"Windows"',
-            "sec-fetch-dest"     : "empty",
-            "sec-fetch-mode"     : "cors",
-            "sec-fetch-site"     : "same-origin",
-            "user-agent"         : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36",
-            "x-wp-nonce"         : _get_nonce(),
-        }
-
-        conversation = _transform(messages)
-
-        json_data = {
-            "env"            : "chatbot",
-            "session"        : "N/A",
-            "prompt"         : "Converse as if you were an AI assistant. Be friendly, creative.",
-            "context"        : "Converse as if you were an AI assistant. Be friendly, creative.",
-            "messages"       : conversation,
-            "newMessage"     : messages[-1]["content"],
-            "userName"       : '<div class="mwai-name-text">User:</div>',
-            "aiName"         : '<div class="mwai-name-text">AI:</div>',
-            "model"          : "gpt-3.5-turbo",
-            "temperature"    : kwargs.get("temperature", 0.8),
-            "maxTokens"      : 1024,
-            "maxResults"     : 1,
-            "apiKey"         : "",
-            "service"        : "openai",
-            "embeddingsIndex": "",
-            "stop"           : "",
-            "clientId"       : os.urandom(6).hex()
+            "User-Agent"         : "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Safari/537.36",
+            "Accept"             : "*/*",
+            "Accept-language"    : "en,fr-FR;q=0.9,fr;q=0.8,es-ES;q=0.7,es;q=0.6,en-US;q=0.5,am;q=0.4,de;q=0.3",
+            "Origin"             : "https://opchatgpts.net",
+            "Alt-Used"           : "opchatgpts.net",
+            "Referer"            : "https://opchatgpts.net/chatgpt-free-use/",
+            "Sec-Fetch-Dest"     : "empty",
+            "Sec-Fetch-Mode"     : "cors",
+            "Sec-Fetch-Site"     : "same-origin",
         }
-
-        response = requests.post("https://opchatgpts.net/wp-json/ai-chatbot/v1/chat",
-            headers=headers, json=json_data)
-        
-        response.raise_for_status()
-        yield response.json()["reply"]
+        async with ClientSession(
+            headers=headers
+        ) as session:
+            if not cls._nonce:
+                async with session.get(
+                    "https://opchatgpts.net/chatgpt-free-use/",
+                    params={"id": os.urandom(6).hex()},
+                ) as response:
+                    result = re.search(r'data-nonce="(.*?)"', await response.text())
+                if not result:
+                    raise RuntimeError("No nonce value")
+                cls._nonce = result.group(1)
+            data = {
+                "_wpnonce": cls._nonce,
+                "post_id": 28,
+                "url": "https://opchatgpts.net/chatgpt-free-use",
+                "action": "wpaicg_chat_shortcode_message",
+                "message": format_prompt(messages),
+                "bot_id": 0
+            }
+            async with session.post("https://opchatgpts.net/wp-admin/admin-ajax.php", data=data) as response:
+                response.raise_for_status()
+                return (await response.json())["data"]
 
     @classmethod
     @property
@@ -76,55 +64,4 @@ class ChatgptLogin(BaseProvider):
             ("temperature", "float"),
         ]
         param = ", ".join([": ".join(p) for p in params])
-        return f"g4f.provider.{cls.__name__} supports: ({param})"
-
-
-def _get_nonce() -> str:
-    res = requests.get("https://opchatgpts.net/chatgpt-free-use/",
-        headers = {
-            "Referer"   : "https://opchatgpts.net/chatgpt-free-use/",
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"})
-
-    result = re.search(
-        r'class="mwai-chat mwai-chatgpt">.*<span>Send</span></button></div></div></div> <script defer src="(.*?)">',
-        res.text)
-    
-    if result is None:
-        return ""
-    
-    src            = result.group(1)
-    decoded_string = base64.b64decode(src.split(",")[-1]).decode("utf-8")
-    result         = re.search(r"let restNonce = '(.*?)';", decoded_string)
-
-    return "" if result is None else result.group(1)
-
-
-def _transform(messages: list[dict[str, str]]) -> list[dict[str, Any]]:
-    return [
-        {
-            "id"     : os.urandom(6).hex(),
-            "role"   : message["role"],
-            "content": message["content"],
-            "who"    : "AI: " if message["role"] == "assistant" else "User: ",
-            "html"   : _html_encode(message["content"]),
-        }
-        for message in messages
-    ]
-
-
-def _html_encode(string: str) -> str:
-    table = {
-        '"' : "&quot;",
-        "'" : "&#39;",
-        "&" : "&amp;",
-        ">" : "&gt;",
-        "<" : "&lt;",
-        "\n": "<br>",
-        "\t": "&nbsp;&nbsp;&nbsp;&nbsp;",
-        " " : "&nbsp;",
-    }
-
-    for key in table:
-        string = string.replace(key, table[key])
-
-    return string
+        return f"g4f.provider.{cls.__name__} supports: ({param})"
\ No newline at end of file