summaryrefslogtreecommitdiffstats
path: root/unfinished/chatpdf/__init__.py
diff options
context:
space:
mode:
authort.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-30 13:42:21 +0200
committert.me/xtekky <98614666+xtekky@users.noreply.github.com>2023-04-30 13:42:21 +0200
commite02094de5baac85613855c8a6c9ae1298324ad0e (patch)
treef3a8930ffc312c31916b7beeffac3ccc090f929c /unfinished/chatpdf/__init__.py
parentUpdate README.md (diff)
downloadgpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar.gz
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar.bz2
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar.lz
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar.xz
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.tar.zst
gpt4free-e02094de5baac85613855c8a6c9ae1298324ad0e.zip
Diffstat (limited to '')
-rw-r--r--unfinished/chatpdf/__init__.py87
1 files changed, 47 insertions, 40 deletions
diff --git a/unfinished/chatpdf/__init__.py b/unfinished/chatpdf/__init__.py
index 4c9d2d3e..30dc1d3e 100644
--- a/unfinished/chatpdf/__init__.py
+++ b/unfinished/chatpdf/__init__.py
@@ -1,59 +1,66 @@
import requests
import json
+from queue import Queue, Empty
+from threading import Thread
+from json import loads
+from re import findall
+
+
class Completion:
def request(prompt: str):
'''TODO: some sort of authentication + upload PDF from URL or local file
- Then you should get the atoken and chat ID
- '''
-
+ Then you should get the atoken and chat ID
+ '''
+
token = "your_token_here"
chat_id = "your_chat_id_here"
url = "https://chat-pr4yueoqha-ue.a.run.app/"
payload = json.dumps({
- "v": 2,
- "chatSession": {
- "type": "join",
- "chatId": chat_id
- },
- "history": [
- {
- "id": "VNsSyJIq_0",
- "author": "p_if2GPSfyN8hjDoA7unYe",
- "msg": "<start>",
- "time": 1682672009270
- },
- {
- "id": "Zk8DRUtx_6",
- "author": "uplaceholder",
- "msg": prompt,
- "time": 1682672181339
- }
- ]
- })
-
+ "v": 2,
+ "chatSession": {
+ "type": "join",
+ "chatId": chat_id
+ },
+ "history": [
+ {
+ "id": "VNsSyJIq_0",
+ "author": "p_if2GPSfyN8hjDoA7unYe",
+ "msg": "<start>",
+ "time": 1682672009270
+ },
+ {
+ "id": "Zk8DRUtx_6",
+ "author": "uplaceholder",
+ "msg": prompt,
+ "time": 1682672181339
+ }
+ ]
+ })
+
# TODO: fix headers, use random user-agent, streaming response, etc
headers = {
- 'authority': 'chat-pr4yueoqha-ue.a.run.app',
- 'accept': '*/*',
- 'accept-language': 'en-US,en;q=0.9',
- 'atoken': token,
- 'content-type': 'application/json',
- 'origin': 'https://www.chatpdf.com',
- 'referer': 'https://www.chatpdf.com/',
- 'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
- 'sec-ch-ua-mobile': '?0',
- 'sec-ch-ua-platform': '"Windows"',
- 'sec-fetch-dest': 'empty',
- 'sec-fetch-mode': 'cors',
- 'sec-fetch-site': 'cross-site',
- 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
- }
+ 'authority': 'chat-pr4yueoqha-ue.a.run.app',
+ 'accept': '*/*',
+ 'accept-language': 'en-US,en;q=0.9',
+ 'atoken': token,
+ 'content-type': 'application/json',
+ 'origin': 'https://www.chatpdf.com',
+ 'referer': 'https://www.chatpdf.com/',
+ 'sec-ch-ua': '"Chromium";v="112", "Google Chrome";v="112", "Not:A-Brand";v="99"',
+ 'sec-ch-ua-mobile': '?0',
+ 'sec-ch-ua-platform': '"Windows"',
+ 'sec-fetch-dest': 'empty',
+ 'sec-fetch-mode': 'cors',
+ 'sec-fetch-site': 'cross-site',
+ 'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36'
+ }
- response = requests.request("POST", url, headers=headers, data=payload).text
+ response = requests.request(
+ "POST", url, headers=headers, data=payload).text
Completion.stream_completed = True
return {'response': response}