Spaces:
Build error
Build error
import gradio | |
import requests | |
import os | |
import json | |
from bardapi import Bard | |
bardKey = os.environ.get('_BARD_API_KEY') | |
def bardChat(data): | |
# Create a session object using the requests library | |
session = requests.Session() | |
# Set the headers for the session | |
session.headers = { | |
"Host": "bard.google.com", | |
"X-Same-Domain": "1", | |
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.114 Safari/537.36", | |
"Content-Type": "application/x-www-form-urlencoded;charset=UTF-8", | |
"Origin": "https://bard.google.com", | |
"Referer": "https://bard.google.com/", | |
} | |
# Set the "__Secure-1PSID" cookie with the Bard API key | |
session.cookies.set("__Secure-1PSID", bardKey) | |
# Create a Bard object with the session and a timeout of 30 seconds | |
bard = Bard(token=bardKey, session=session, timeout=30) | |
answer = bard.get_answer(data)['content'] | |
print(answer) | |
return json.dumps({'message':answer,'action':'null'}) | |
def responsenew(data): | |
return bardChat(data) | |
gradio_interface = gradio.Interface( | |
fn = responsenew, | |
inputs = "text", | |
outputs = "text" | |
) | |
gradio_interface.launch() | |