Spaces:
Running
Running
File size: 2,515 Bytes
d4adf88 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import re
import requests
import hashlib
from fake_useragent import UserAgent
class Account:
@staticmethod
def create():
r=requests.get("https://italygpt.it/",headers=Account._header)
f=r.text
tid=re.search('<input type=\"hidden\" name=\"next_id\" id=\"next_id\" value=\"(\w+)\">',f).group(1)
if len(tid)==0:
raise RuntimeError("NetWorkError:failed to get id.")
else:
Account._tid=tid
Account._raw="[]"
return Account
def next(next_id:str)->str:
Account._tid=next_id
return Account._tid
def get()->str:
return Account._tid
_header={
"Host": "italygpt.it",
"Referer":"https://italygpt.it/",
"User-Agent":"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36",#UserAgent().random,
"Accept":"text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8",
"Accept-Language":"zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2",
"Upgrade-Insecure-Requests":"1",
"Sec-Fetch-Dest":"document",
"Sec-Fetch-Mode":"navigate",
"Sec-Fetch-Site":"none",
"Sec-Fetch-User":"?1",
"Connection":"keep-alive",
"Alt-Used":"italygpt.it",
"Pragma":"no-cache",
"Cache-Control":"no-cache",
"TE": "trailers"
}
def settraw(raws:str):
Account._raw=raws
return Account._raw
def gettraw():
return Account._raw
class Completion:
@staticmethod
def create(
account_data,
prompt: str,
message=False
):
param={
"prompt":prompt.replace(" ","+"),
"creative":"off",
"internet":"false",
"detailed":"off",
"current_id":"0",
"code":"",
"gpt4":"false",
"raw_messages":account_data.gettraw(),
"hash":hashlib.sha256(account_data.get().encode()).hexdigest()
}
if(message):
param["raw_messages"]=str(message)
r = requests.get("https://italygpt.it/question",headers=account_data._header,params=param,stream=True)
account_data.next(r.headers["Next_id"])
account_data.settraw(r.headers["Raw_messages"])
for chunk in r.iter_content(chunk_size=None):
r.raise_for_status()
yield chunk.decode() |