|
from requests import Session |
|
from re import search |
|
from random import randint |
|
from json import dumps, loads |
|
from random import randint |
|
from urllib.parse import urlencode |
|
from dotenv import load_dotenv; load_dotenv() |
|
from os import getenv |
|
|
|
from bard.typings import BardResponse |
|
|
|
token = getenv('1psid') |
|
proxy = getenv('proxy') |
|
|
|
temperatures = { |
|
0 : "Generate text strictly following known patterns, with no creativity.", |
|
0.1: "Produce text adhering closely to established patterns, allowing minimal creativity.", |
|
0.2: "Create text with modest deviations from familiar patterns, injecting a slight creative touch.", |
|
0.3: "Craft text with a mild level of creativity, deviating somewhat from common patterns.", |
|
0.4: "Formulate text balancing creativity and recognizable patterns for coherent results.", |
|
0.5: "Generate text with a moderate level of creativity, allowing for a mix of familiarity and novelty.", |
|
0.6: "Compose text with an increased emphasis on creativity, while partially maintaining familiar patterns.", |
|
0.7: "Produce text favoring creativity over typical patterns for more original results.", |
|
0.8: "Create text heavily focused on creativity, with limited concern for familiar patterns.", |
|
0.9: "Craft text with a strong emphasis on unique and inventive ideas, largely ignoring established patterns.", |
|
1 : "Generate text with maximum creativity, disregarding any constraints of known patterns or structures." |
|
} |
|
|
|
class Completion: |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create( |
|
prompt : str = 'hello world', |
|
temperature : int = None, |
|
conversation_id : str = '', |
|
response_id : str = '', |
|
choice_id : str = '') -> BardResponse: |
|
|
|
if temperature: |
|
prompt = f'''settings: follow these settings for your response: [temperature: {temperature} - {temperatures[temperature]}] | prompt : {prompt}''' |
|
|
|
client = Session() |
|
client.proxies = { |
|
'http': f'http://{proxy}', |
|
'https': f'http://{proxy}' } if proxy else None |
|
|
|
client.headers = { |
|
'authority' : 'bard.google.com', |
|
'content-type' : 'application/x-www-form-urlencoded;charset=UTF-8', |
|
'origin' : 'https://bard.google.com', |
|
'referer' : 'https://bard.google.com/', |
|
'user-agent' : 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36', |
|
'x-same-domain' : '1', |
|
'cookie' : f'__Secure-1PSID={token}' |
|
} |
|
|
|
snlm0e = search(r'SNlM0e\":\"(.*?)\"', client.get('https://bard.google.com/').text).group(1) |
|
|
|
params = urlencode({ |
|
'bl' : 'boq_assistant-bard-web-server_20230326.21_p0', |
|
'_reqid' : randint(1111, 9999), |
|
'rt' : 'c', |
|
}) |
|
|
|
response = client.post(f'https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate?{params}', |
|
data = { |
|
'at': snlm0e, |
|
'f.req': dumps([None, dumps([ |
|
[prompt], |
|
None, |
|
[conversation_id, response_id, choice_id], |
|
]) |
|
]) |
|
} |
|
) |
|
|
|
chat_data = loads(response.content.splitlines()[3])[0][2] |
|
if not chat_data: print('error, retrying'); Completion.create(prompt, temperature, conversation_id, response_id, choice_id) |
|
|
|
json_chat_data = loads(chat_data) |
|
results = { |
|
'content' : json_chat_data[0][0], |
|
'conversation_id' : json_chat_data[1][0], |
|
'response_id' : json_chat_data[1][1], |
|
'factualityQueries' : json_chat_data[3], |
|
'textQuery' : json_chat_data[2][0] if json_chat_data[2] is not None else '', |
|
'choices' : [{'id': i[0], 'content': i[1]} for i in json_chat_data[4]], |
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
return BardResponse(results) |
|
|
