First commit

Author: brunneis

.gitignore

@@ -1,5 +1,6 @@
 # Custom
 .dccache
+config.json
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

package_name/__init__.py renamed to chatgpt/__init__.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
-from .package_name import *
+from .chatgpt import *
 
 __version__ = '0.0.1b0'

chatgpt/chatgpt.py

@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from uuid import uuid4 as uuid
+import json
+import requests
+from .errors import CallError
+
+
+class Conversation:
+    DEFAULT_MODEL_NAME = 'text-davinci-002-render'
+
+    def __init__(
+        self,
+        model_name: str = None,
+        conversation_id: str = None,
+        parent_message_id: str = None,
+        access_token: str = None,
+    ):
+        self.load_config()
+
+        self._model_name = model_name or self.DEFAULT_MODEL_NAME
+
+        if self._access_token is None:
+            self._access_token = access_token
+
+        if self._conversation_id is None:
+            self._conversation_id = conversation_id
+
+        if self._first_parent_message_id is None:
+            self._first_parent_message_id = parent_message_id
+        self._message_id = parent_message_id
+
+    def load_config(self, config_path: str = None):
+        if config_path is None:
+            config_path = 'config.json'
+
+        with open(config_path, 'r') as f:
+            config = json.load(f)
+
+        self._access_token = config['access_token']
+        self._conversation_id = config['conversation_id']
+        self._first_parent_message_id = config['parent_message_id']
+
+    def chat(self, message: str):
+        self._parent_message_id = self._message_id
+        self._message_id = str(uuid())
+
+        url = "https://chat.openai.com/backend-api/conversation"
+
+        payload = json.dumps(
+            {
+                "action": "next",
+                "messages": [
+                    {
+                        "id": self._message_id,
+                        "role": "user",
+                        "content": {
+                            "content_type": "text",
+                            "parts": [message]
+                        }
+                    }
+                ],
+                "conversation_id": self._conversation_id,
+                "parent_message_id": self._parent_message_id,
+                "model": self._model_name
+            }
+        )
+        headers = {
+            'Authorization': f'Bearer {self._access_token}',
+        }
+
+        response = requests.request("POST", url, headers=headers, data=payload)
+        payload = response.text
+
+        try:
+            last_item = payload.split(('data:'))[-2]
+            text_items = json.loads(last_item)['message']['content']['parts']
+            text = '\n'.join(text_items)
+            postprocessed_text = text.replace(r'\n+', '\n')
+            return postprocessed_text
+        except IndexError:
+            error_message = json.loads(payload)['detail']
+            raise CallError(error_message)
+
+    def reset(self):
+        self._message_id = self._first_parent_message_id

chatgpt/errors.py

@@ -0,0 +1,4 @@
+class CallError(Exception):
+
+    def __init__(self, message):
+        super().__init__(message)

chatgpt/interceptor.js

@@ -0,0 +1,40 @@
+const textareas = document.getElementsByTagName('textarea');
+const lastTextarea = textareas[textareas.length - 1];
+
+let buttons = document.getElementsByTagName('button');
+buttons = Array.prototype.slice.call(buttons);
+const lastButton = buttons[buttons.length - 1];
+
+lastTextarea.value = 'Hello'
+lastButton.click();
+
+var interval = setInterval(function() {
+  if (lastButton.disabled === false) {
+    clearInterval(interval);
+    lastTextarea.value = "Let's start"
+    lastButton.click();
+  }
+}, 100);
+
+function fetchInterceptor(fetch) {
+  return function(...args) {
+    const body = JSON.parse(args[1]?.body);
+    const headers = args[1]?.headers;
+    
+    const accessToken = headers?.Authorization.split('Bearer')[1].trim()
+    const conversationId = body?.conversation_id
+    const parentMessageId = body?.parent_message_id
+
+    if (accessToken && conversationId && parentMessageId) {
+      const config = {
+        access_token: accessToken,
+        conversation_id: body.conversation_id,
+        parent_message_id: body.parent_message_id,
+      };
+      console.log(JSON.stringify(config));
+    }
+
+    return fetch(...args);
+  };
+}
+window.fetch = fetchInterceptor(window.fetch);

chatgpt/package_name/__init__.py

@@ -0,0 +1,6 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+from .chatgpt import *
+
+__version__ = '0.0.1b0'

package_name/package_name.py renamed to chatgpt/package_name/chatgpt.py

@@ -0,0 +1,4 @@
+from chatgpt import Conversation
+
+conversation = Conversation()
+print(conversation.chat("Hola."))

playground/main.py

@@ -3,15 +3,15 @@
 
 from setuptools import setup
 from setuptools import find_packages
-import package_name
+import chatgpt
 
 setup(
-    name='package_name',
-    version=package_name.__version__,
-    description='description_value',
-    url='url_value',
-    author='author_value',
-    author_email='email_value',
+    name='chatgpt',
+    version=chatgpt.__version__,
+    description='',
+    url='https://github.com/brunneis/chatgpt-python',
+    author='Rodrigo Martínez Castaño',
+    author_email='rodrigo@martinez.gal',
     license='GNU General Public License v3 (GPLv3)',
     packages=find_packages(),
     zip_safe=False,