summaryrefslogblamecommitdiffstats
path: root/unfinished/bard/__init__.py
blob: 2b26997fcdbb1d44ad674a99e1df783248a39238 (plain) (tree)


















































































































                                                                                                                                                              
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)