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 __init__(self, _token, proxy: str or None = None) -> None:
# self.client = Session()
# self.client.proxies = {
# 'http': f'http://{proxy}',
# 'https': f'http://{proxy}' } if proxy else None
# self.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}'
# }
# self.snlm0e = self.__init_client()
# self.conversation_id = ''
# self.response_id = ''
# self.choice_id = ''
# self.reqid = randint(1111, 9999)
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]],
}
# self.conversation_id = results['conversation_id']
# self.response_id = results['response_id']
# self.choice_id = results['choices'][0]['id']
# self.reqid += 100000
return BardResponse(results)
