Spaces:
Sleeping
Sleeping
File size: 1,740 Bytes
ec91473 b896b78 11c3841 b896b78 4189e1c 041dcde b896b78 310bca6 b04bdf0 b896b78 310bca6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import gradio as gr
import requests
import time
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(session=session, timeout=30)
answer = bard.get_answer(data, token=bardKey)['content']
# print(answer)
return answer
def responsenew(data):
return bardChat(data)
# remind_val = ["create a reminder", "create reminder", "remind me"]
# if remind_val in data:
# return "Reminder created!"
# else:
# return bardChat(data)
with gr.Blocks() as demo:
chatbot = gr.Chatbot()
msg = gr.Textbox()
clear = gr.ClearButton([msg, chatbot])
def respond(message, chat_history):
bot_message = responsenew(message)
chat_history.append((message, bot_message))
time.sleep(2)
return "", chat_history
msg.submit(respond, [msg, chatbot], [msg, chatbot])
if __name__ == "__main__":
demo.launch() |