Upload main.py

main.py

@@ -582,8 +582,25 @@ def gemini_pro(
         token = bard_api_key
     else:
         token = COOKIE_BARD_TOKEN
+    owner_base = f"""
+    Your name is Randy Dev. A kind and friendly AI assistant that answers in
+    a short and concise answer. Give short step-by-step reasoning if required.
+
+    Today is {dt.now():%A %d %B %Y %H:%M}
+    """
     try:
-        bard = Bard(token=token)
+        session = requests.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/"
+        }
+        session.cookies.set("__Secure-1PSID", token)
+        bard = Bard(token=token, session=session, timeout=30)
+        bard.get_answer(owner_base)["content"]
         message = bard.get_answer(query)["content"]
         return {
             "status": "true",