import numpy as np import gradio as gr import requests from bardapi.constants import SESSION_HEADERS from bardapi import Bard import threading import logging import json import os def load_cookie(element_name): with open(os.getcwd() + "/" + "bard.json", "r", encoding="utf-8") as file: data = json.load(file) if isinstance(data, list): for item in data: if item.get("name") == element_name: return item.get("value") return None def _refresh_cookie(session, force=False, inputData=None): s = session cookies = [] if force == True: if inputData != None: rcookies = inputData else: rcookies = { '__Secure-1PSID': 'cAjTLu0tbHFzi4i7SN02J-Sk09hAUxRW2cRweWXdOv43S3XcfSqm6GRkIfXU_MLI040bXg.', '__Secure-1PSIDCC':'ACA-OxNf3jPFp4Q2EGce1dvSnC_Rmc0LwNHrxHMXtx6J9tK2KiNhQesn1qV2SYxSsXYNjYIFAUk', '__Secure-1PSIDTS':'sidts-CjEB3e41haWxCIIO6I3Uglgpesu4OIQtj8xcUsXy2vo4wH_iI9g5W2XqS4mYXA0R5V3kEAA' } cookies.append({'name': '__Secure-1PSID', 'value': rcookies['__Secure-1PSID']}) cookies.append({'name': '__Secure-1PSIDTS', 'value': rcookies['__Secure-1PSIDTS']}) cookies.append({'name': '__Secure-1PSIDCC', 'value': rcookies['__Secure-1PSIDCC']}) for k, v in s.cookies.get_dict().items(): cookies.append({'name': k, 'value': v}) with open("bard.json", "w") as outfile: outfile.write(json.dumps(cookies)) """ curl 'https://accounts.google.com/RotateCookies' \ -H 'authority: accounts.google.com' \ -H 'accept: */*' \ -H 'accept-language: zh-CN,zh;q=0.9' \ -H 'cache-control: no-cache' \ -H 'content-type: application/json' \ -H 'cookie: CONSENT=PENDING+126; SOCS=CAISHwgCEhJnd3NfMjAyMzA5MTktMF9SQzEaBXpoLUNOIAEaBgiAj66oBg; OTZ=7216350_24_24__24_; SID=bgjTLuZd5FNRwaqPlNR-g4YJZjm_Gso043r6aomscB7qpTCthr0ofKjCkOKcrvZc6M8bFQ.; __Secure-1PSID=bgjTLuZd5FNRwaqPlNR-g4YJZjm_Gso043r6aomscB7qpTCtCtmdXnqmmvICf_dqN2E0Ew.; __Secure-3PSID=bgjTLuZd5FNRwaqPlNR-g4YJZjm_Gso043r6aomscB7qpTCtgQaKSr89I03JVG_tgACzhA.; HSID=AlvhATI6KzcNKnabh; SSID=Ae8TCtjtt27en1e44; APISID=g5cwoAMD8d2_sO11/A-Rwczmgi0udFgCoQ; SAPISID=OzZnHNamhN3jXLNC/A2vmqDIgjbrTejueT; __Secure-1PAPISID=OzZnHNamhN3jXLNC/A2vmqDIgjbrTejueT; __Secure-3PAPISID=OzZnHNamhN3jXLNC/A2vmqDIgjbrTejueT; ACCOUNT_CHOOSER=AFx_qI7_HEzgTYKnWcZUdfRgVgKIae2pD7PXhhfk44t3IwPv_ecwcqhG0Of35gtj_CJan2ftXCcI-pbHHiuHrI-uBgwRN3prTpTlK-lfdTcLGIeK7EX-bQmUk-RP5eyvTibD2B1TeFIk; SEARCH_SAMESITE=CgQItpkB; LSID=o.console.cloud.google.com|o.myaccount.google.com|s.HK|s.youtube:bgjTLs6wckDwgc9bMe0STv7Ddq2FdtHhlBl1QZNfwXDMEEIif8dB7SQdook7LsRyaLoo0g.; __Host-1PLSID=o.console.cloud.google.com|o.myaccount.google.com|s.HK|s.youtube:bgjTLs6wckDwgc9bMe0STv7Ddq2FdtHhlBl1QZNfwXDMEEIiq0UCdDLPvW9AYTEoD10U2A.; __Host-3PLSID=o.console.cloud.google.com|o.myaccount.google.com|s.HK|s.youtube:bgjTLs6wckDwgc9bMe0STv7Ddq2FdtHhlBl1QZNfwXDMEEIi7K0n0ZBE55ocXwF_87WeRQ.; __Host-GAPS=1:kiYlEzR-7QR6lpenXGRKZdnXzGV6tjibaIhaGTaU0EZ49y1_dA9_GnN-GfJ2_2FiiyJlq75j3ycG4NTdvXCI4E0_pmfstA:5ib-ikZoBeIN7BET; AEC=Ackid1Tlb_Cwsk2VvvwFAn486fnHL27O-kzZtpJLRzf6WBHs9TZUyOIviQ; NID=511=JTxtvHyIWGcJdIXl9QiMeIudlBg-h08jlOKyQGQbEb-tUBlFOAX3B-RElc47doqva2WB-EWnkeT6DVLlBwP3v7sYSJqnMN5oXP7RojJT_FQ-XtxxYS_Io_q-_IDh3uwUPXXtJsxgJr95aHghqkxgdkBHIiUC9yvLcGwIjVQOySuOPIXIJNIXVKGQOmvIeRWnlUAEzxIVVSvgyirUJj2vRU-nmQ-It_k9tcM1t_vmepTdoT4RdJHRdPqV6AQrEd1llZvSGPxmZusAhfzBeJy2NOUn96zS4zpsh1897x_BPd_62uiELKCAO0_haso8FuL6E40HseWGhvhPgKR2vaCMyPoF4d0j; 1P_JAR=2023-10-10-10; __Secure-1PSIDTS=sidts-CjIB3e41hVwPZ-GA-le4Qb8npu9IgvD-BIaueTRorj7f0euJUNARPE7QMdM907q2f-pUVRAA; __Secure-3PSIDTS=sidts-CjIB3e41hVwPZ-GA-le4Qb8npu9IgvD-BIaueTRorj7f0euJUNARPE7QMdM907q2f-pUVRAA; SIDCC=ACA-OxOmjCOQXoVSBIpW9MaHGfCgY8vgHeTdKdn5m91k_ExTUlvNxz4yz2bHIm2YO4GzE2viNEw; __Secure-1PSIDCC=ACA-OxNJir5rKUGGcgK7Qs4DVftEa14sbyZbpuczcTox1OhSlX15KmaNz2thUSCOit_4gkMOLok; __Secure-3PSIDCC=ACA-OxOsXDQ0RoYiLiu1EysjUZU_tD3rPxSMXjZWC-iWGTg1ETB0gmzuJ55H8gAgv7S_8dXo_Zc' \ -H 'origin: https://accounts.google.com' \ -H 'pragma: no-cache' \ -H 'referer: https://accounts.google.com/RotateCookiesPage?og_pid=658&rot=3&origin=https%3A%2F%2Fbard.google.com&exp_id=3701166' \ -H 'sec-ch-ua: "Google Chrome";v="117", "Not;A=Brand";v="8", "Chromium";v="117"' \ -H 'sec-ch-ua-arch: "arm"' \ -H 'sec-ch-ua-bitness: "64"' \ -H 'sec-ch-ua-full-version: "117.0.5938.149"' \ -H 'sec-ch-ua-full-version-list: "Google Chrome";v="117.0.5938.149", "Not;A=Brand";v="8.0.0.0", "Chromium";v="117.0.5938.149"' \ -H 'sec-ch-ua-mobile: ?0' \ -H 'sec-ch-ua-model: ""' \ -H 'sec-ch-ua-platform: "macOS"' \ -H 'sec-ch-ua-platform-version: "13.4.1"' \ -H 'sec-ch-ua-wow64: ?0' \ -H 'sec-fetch-dest: empty' \ -H 'sec-fetch-mode: same-origin' \ -H 'sec-fetch-site: same-origin' \ -H 'user-agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36' \ -H 'x-chrome-id-consistency-request: version=1,client_id=77185425430.apps.googleusercontent.com,device_id=8d089f9c-340b-4b9a-91aa-176c7fc95f62,signin_mode=all_accounts,signout_mode=show_confirmation' \ -H 'x-client-data: CJW2yQEIprbJAQipncoBCIaMywEIlKHLAQiGoM0BCNy9zQEIucrNAQjD0c0BCInTzQEI0tbNAQj11s0BCKfYzQEIl9nNAQj5wNQVGPTJzQE=' \ --data-raw '[658,"-3903137320083272820"]' \ --compressed """ def rotateCookie(): url = 'https://accounts.google.com/RotateCookies' session, token = getSession() headers = { "accept": "*/*", "accept-language": "zh-CN,zh;q=0.9", "cache-control": "no-cache", "content-type": "application/json", "pragma": "no-cache", "sec-ch-ua": "\"Google Chrome\";v=\"117\", \"Not;A=Brand\";v=\"8\", \"Chromium\";v=\"117\"", "sec-ch-ua-arch": "\"arm\"", "sec-ch-ua-bitness": "\"64\"", "sec-ch-ua-full-version": "\"117.0.5938.149\"", "sec-ch-ua-full-version-list": "\"Google Chrome\";v=\"117.0.5938.149\", \"Not;A=Brand\";v=\"8.0.0.0\", \"Chromium\";v=\"117.0.5938.149\"", "sec-ch-ua-mobile": "?0", "sec-ch-ua-model": "\"\"", "sec-ch-ua-platform": "\"macOS\"", "sec-ch-ua-platform-version": "\"13.4.1\"", "sec-ch-ua-wow64": "?0", "sec-fetch-dest": "empty", "sec-fetch-mode": "same-origin", "sec-fetch-site": "same-origin", "x-chrome-id-consistency-request": "version=1,client_id=77185425430.apps.googleusercontent.com,device_id=8d089f9c-340b-4b9a-91aa-176c7fc95f62,signin_mode=all_accounts,signout_mode=show_confirmation", "x-client-data": "CJW2yQEIprbJAQipncoBCIaMywEIlKHLAQiGoM0BCNy9zQEIucrNAQjD0c0BCInTzQEI0tbNAQj11s0BCKfYzQEIl9nNAQj5wNQVGPTJzQE=", } data = "[658,\"-3903137320083272820\"]" cookies = session.cookies r = requests.post(url, headers=headers, cookies=cookies, data=data) session.cookies.update(r.cookies) _refresh_cookie(session) def autoRotate(): threading.Timer(600, autoRotate).start() print("rotate cookie") rotateCookie() def getSession(): _1PSID = load_cookie("__Secure-1PSID") _1PSIDTS = load_cookie("__Secure-1PSIDTS") _1PSIDCC = load_cookie("__Secure-1PSIDCC") session = requests.Session() session.headers = SESSION_HEADERS session.cookies.set("__Secure-1PSID", _1PSID) session.cookies.set("__Secure-1PSIDTS", _1PSIDTS) session.cookies.set("__Secure-1PSIDCC", _1PSIDCC) return session, _1PSID def _bard(question, image=None): session, token = getSession() bard = Bard(token = token, session = session) if image != None: result = bard.ask_about_image(question, image) else: result = bard.get_answer(question) print(result) _refresh_cookie(bard.session) return result def initLogger(): logging.basicConfig(filename="app.log", filemode='a', format='%(asctime)s,%(msecs)d %(name)s %(levelname)s %(message)s', datefmt='%H:%M:%S', level=logging.INFO) logging.info("init logger") def refreshFromUI(x): pre = "refresh:" if x.strip() != "" and x.startswith(pre): inputData = x[len(pre):] if len(inputData) > 0: cookies = json.loads(inputData) _refresh_cookie(requests.Session(), True, cookies) return "ok" def bard_text(x): ok = refreshFromUI(x) if ok != None: return ok res = _bard(x) logger = logging.getLogger('app') logger.info(json.dumps({"q":x, "a":res, "i":None},separators=(',', ':'))) return res['content'] def bard_image(img, x): img_content = open(img, 'rb').read() res = _bard(x, img_content) logger = logging.getLogger('app') logger.info(json.dumps({"q":x, "a":res, "i":img}, separators=(',', ':'))) return img, res['content'] def init(): initLogger() #_refresh_cookie(requests.Session(), True) autoRotate() pass with gr.Blocks() as demo: init() gr.Markdown("A viture tool of label image with google bard.") with gr.Tab("Text"): text_input = gr.Textbox() text_output = gr.Markdown() text_button = gr.Button("submit") with gr.Tab("Image"): with gr.Row(): image_input = gr.Image(type='filepath') image_output = gr.Image(type='filepath') with gr.Row(): image_text_input = gr.Textbox() image_text_output = gr.Markdown() image_button = gr.Button("submit") text_button.click(bard_text, inputs=text_input, outputs=text_output) image_button.click(bard_image, inputs=[image_input, image_text_input], outputs=[image_output, image_text_output]) demo.launch()