bard api

app.py

@@ -14,6 +14,7 @@ os.environ['_BARD_API_KEY']="YgjksaVw8UQahn8uKwL3Zve_98YtnUIePVY16j7d8Z3y-tXYrUI
 
 
 
+
 def huggChat(data):
     # start a new huggingchat connection
     chatbot = hugchat.ChatBot(cookie_path="cookies.json")
@@ -30,8 +31,26 @@ def huggChat(data):
     return 'Hello Sir'
 
 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", os.getenv("_BARD_API_KEY")) 
+    
+    # Create a Bard object with the session and a timeout of 30 seconds
+    bard = Bard(session=session, timeout=30)
     
-    print(Bard().get_answer(data)['content'])
+    print(bard.get_answer(data)['content'])
 
 def query(payload):
     response = requests.post(API_URL, headers=headers, json=payload)