temp.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import json\n",
    "import random\n",
    "import re\n",
    "import string\n",
    "import os\n",
    "import requests\n",
    "\n",
    "\n",
    "class Bard:\n",
    "    def __init__(self, timeout=6, proxies=None, session=None):\n",
    "        '''\n",
    "        Initialize Bard\n",
    "\n",
    "        :param timeout: (`int`, *optional*)\n",
    "            Timeout in seconds when connecting bard server. The timeout is used on each request.\n",
    "        :param proxies: (`Dict[str, str]`, *optional*)\n",
    "            A dictionary of proxy servers to use by protocol or endpoint, e.g., `{'http': 'foo.bar:3128',\n",
    "            'http://hostname': 'foo.bar:4012'}`. The proxies are used on each request.\n",
    "        :param session: (`requests.Session`, *optional*)\n",
    "            An existing requests.Session object to be used for making HTTP requests.\n",
    "        '''\n",
    "        self.proxies = proxies\n",
    "        self.timeout = timeout\n",
    "        headers = {\n",
    "            \"Host\": \"bard.google.com\",\n",
    "            \"X-Same-Domain\": \"1\",\n",
    "            \"User-Agent\": \"Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36\",\n",
    "            \"Content-Type\": \"application/x-www-form-urlencoded;charset=UTF-8\",\n",
    "            \"Origin\": \"https://bard.google.com\",\n",
    "            \"Referer\": \"https://bard.google.com/\",\n",
    "        }\n",
    "        self._reqid = int(\"\".join(random.choices(string.digits, k=4)))\n",
    "        self.conversation_id = \"\"\n",
    "        self.response_id = \"\"\n",
    "        self.choice_id = \"\"\n",
    "        \n",
    "        if session is None:\n",
    "            self.session = requests.Session()\n",
    "            self.session.headers = headers\n",
    "            self.session.cookies.set(\"__Secure-1PSID\", os.environ[\"_BARD_API_KEY\"])\n",
    "        else:\n",
    "            print('ssssss')\n",
    "            if os.getenv(\"_BARD_API_KEY\") is None:\n",
    "                raise 'Set _BARD_API_KEY as an environment variable before passing session as arguments.'\n",
    "            self.session = session\n",
    "        \n",
    "        self.SNlM0e = self._get_snim0e()\n",
    "\n",
    "    def _get_snim0e(self):\n",
    "        resp = self.session.get(url=\"https://bard.google.com/\", timeout=self.timeout, proxies=self.proxies)\n",
    "        if resp.status_code != 200:\n",
    "            raise Exception(f\"Response Status: {resp.status_code}\")\n",
    "        return re.search(r\"SNlM0e\\\":\\\"(.*?)\\\"\", resp.text).group(1)\n",
    "\n",
    "    def get_answer(self, input_text: str) -> dict:\n",
    "        params = {\n",
    "            \"bl\": \"boq_assistant-bard-web-server_20230419.00_p1\",\n",
    "            \"_reqid\": str(self._reqid),\n",
    "            \"rt\": \"c\",\n",
    "        }\n",
    "        input_text_struct = [\n",
    "            [input_text],\n",
    "            None,\n",
    "            [self.conversation_id, self.response_id, self.choice_id],\n",
    "        ]\n",
    "        data = {\n",
    "            \"f.req\": json.dumps([None, json.dumps(input_text_struct)]),\n",
    "            \"at\": self.SNlM0e,\n",
    "        }\n",
    "        resp = self.session.post(\n",
    "            \"https://bard.google.com/_/BardChatUi/data/assistant.lamda.BardFrontendService/StreamGenerate\",\n",
    "            params=params,\n",
    "            json=data,\n",
    "            timeout=self.timeout,\n",
    "            proxies=self.proxies\n",
    "        )\n",
    "\n",
    "        resp_dict = json.loads(resp.content.splitlines()[3])[0][2]\n",
    "        if resp_dict is None:\n",
    "            return {\"content\": f\"Response Error: {resp.content}.\"}\n",
    "        parsed_answer = json.loads(resp_dict)\n",
    "        bard_answer = {\n",
    "            \"content\": parsed_answer[0][0],\n",
    "            \"conversation_id\": parsed_answer[1][0],\n",
    "            \"response_id\": parsed_answer[1][1],\n",
    "            \"factualityQueries\": parsed_answer[3],\n",
    "            \"textQuery\": parsed_answer[2][0] if parsed_answer[2] is not None else \"\",\n",
    "            \"choices\": [{\"id\": i[0], \"content\": i[1]} for i in parsed_answer[4]],\n",
    "        }\n",
    "        self.conversation_id = bard_answer[\"conversation_id\"]\n",
    "        self.response_id = bard_answer[\"response_id\"]\n",
    "        self.choice_id = bard_answer[\"choices\"][0][\"id\"]\n",
    "        self._reqid += 100000\n",
    "\n",
    "        return bard_answer"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "os.environ[\"_BARD_API_KEY\"] = \"VwhVqhvFpnOm_sSsLszdru_DWm8otZBt3hPR9fud2BePGTk2jQmGmfcbKb_vyr3IjJFKEA.\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "session = requests.Session()\n",
    "session.headers = {\n",
    "            \"Host\": \"bard.google.com\",\n",
    "            \"X-Same-Domain\": \"1\",\n",
    "            \"User-Agent\": \"Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36\",\n",
    "            \"Content-Type\": \"application/x-www-form-urlencoded;charset=UTF-8\",\n",
    "            \"Origin\": \"https://bard.google.com\",\n",
    "            \"Referer\": \"https://bard.google.com/\",\n",
    "        }\n",
    "session.cookies.set(\"__Secure-1PSID\", os.environ[\"_BARD_API_KEY\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "Bard(session=session).get_answer(\"hi\")['content']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "from bardapi import Bard\n",
    "import os\n",
    "\n",
    "os.environ['_BARD_API_KEY']=\"VwhVqhvFpnOm_sSsLszdru_DWm8otZBt3hPR9fud2BePGTk2jQmGmfcbKb_vyr3IjJFKEA.\"\n",
    "Bard().get_answer(\"나와 내 동년배들이 좋아하는 뉴진스에 대해서 알려줘\")['content']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "venv",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.9"
  },
  "orig_nbformat": 4
 },
 "nbformat": 4,
 "nbformat_minor": 2
}