Fix Bard Provider

g4f/Provider/PerplexityAi.py

@@ -14,7 +14,7 @@ from .helper import format_prompt, get_browser
 class PerplexityAi(BaseProvider):
     url = "https://www.perplexity.ai"
     working = True
-    supports_gpt_4 = True
+    supports_gpt_35_turbo = True
     supports_stream = True
 
     @classmethod
@@ -30,10 +30,6 @@ class PerplexityAi(BaseProvider):
         display: bool = True,
         **kwargs
     ) -> CreateResult:
-        from selenium.webdriver.common.by import By
-        from selenium.webdriver.support.ui import WebDriverWait
-        from selenium.webdriver.support import expected_conditions as EC
-
         if browser:
             driver = browser
         else:
@@ -42,11 +38,18 @@ class PerplexityAi(BaseProvider):
             else:
                 driver = get_browser("", False, proxy)
 
+        from selenium.webdriver.common.by import By
+        from selenium.webdriver.support.ui import WebDriverWait
+        from selenium.webdriver.support import expected_conditions as EC
+
         prompt = format_prompt(messages)
 
         driver.get(f"{cls.url}/")
         wait = WebDriverWait(driver, timeout)
 
+        # Page loaded?
+        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "textarea[placeholder='Ask anything...']")))
+
         script = """
 window._message = window._last_message = "";
 window._message_finished = false;
@@ -75,9 +78,6 @@ WebSocket.prototype.send = function(...args) {
 """
         driver.execute_script(script)
 
-        # Page loaded?
-        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "textarea[placeholder='Ask anything...']")))
-
         if copilot:
             try:
                 driver.find_element(By.CSS_SELECTOR, "img[alt='User avatar']")

g4f/Provider/needs_auth/Bard.py

@@ -1,90 +1,96 @@
 from __future__ import annotations
 
-import json
+import time
-import random
+try:
-import re
+    from selenium.webdriver.remote.webdriver import WebDriver
+except ImportError:
+    class WebDriver():
+        pass
 
-from aiohttp import ClientSession
+from ...typing import CreateResult, Messages
+from ..base_provider import BaseProvider
+from ..helper import format_prompt, get_browser
 
-from ...typing import Messages
+class Bard(BaseProvider):
-from ..base_provider import AsyncProvider
-from ..helper import format_prompt, get_cookies
-
-
-class Bard(AsyncProvider):
     url = "https://bard.google.com"
-    needs_auth = True
     working = True
-    _snlm0e = None
+    needs_auth = True
 
     @classmethod
-    async def create_async(
+    def create_completion(
         cls,
         model: str,
         messages: Messages,
+        stream: bool,
         proxy: str = None,
-        cookies: dict = None,
+        browser: WebDriver = None,
+        display: bool = True,
         **kwargs
-    ) -> str:
+    ) -> CreateResult:
         prompt = format_prompt(messages)
-        if not cookies:
+        if browser:
-            cookies = get_cookies(".google.com")
+            driver = browser
-
-        headers = {
-            'authority': 'bard.google.com',
-            'origin': cls.url,
-            'referer': f'{cls.url}/',
-            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/111.0.0.0 Safari/537.36',
-            'x-same-domain': '1',
-        }
-        async with ClientSession(
-                cookies=cookies,
-                headers=headers
-            ) as session:
-            if not cls._snlm0e:
-                async with session.get(cls.url, proxy=proxy) as response:
-                    text = await response.text()
-
-                if match := re.search(r'SNlM0e\":\"(.*?)\"', text):
-                    cls._snlm0e = match.group(1)
-
         else:
-                    raise RuntimeError("No snlm0e value.")
+            if display:
-            params = {
+                driver, display = get_browser(None, True, proxy)
-                'bl': 'boq_assistant-bard-web-server_20230326.21_p0',
+            else:
-                '_reqid': random.randint(1111, 9999),
+                driver = get_browser(None, False, proxy)
-                'rt': 'c'
+
+        from selenium.webdriver.common.by import By
+        from selenium.webdriver.support.ui import WebDriverWait
+        from selenium.webdriver.support import expected_conditions as EC
+
+        try:
+            driver.get(f"{cls.url}/chat")
+            wait = WebDriverWait(driver, 10)
+            wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "div.ql-editor.textarea")))
+        except:
+            # Reopen browser for login
+            if not browser:
+                driver.quit()
+                # New browser should be visible
+                if display:
+                    display.stop()
+                driver = get_browser(None, False, proxy)
+                driver.get(f"{cls.url}/chat")
+                wait = WebDriverWait(driver, 240)
+                wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "div.ql-editor.textarea")))
+            else:
+                raise RuntimeError("Prompt textarea not found. You may not be logged in.")
+
+        try:
+            # Add hook in XMLHttpRequest
+            script = """
+const _http_request_open = XMLHttpRequest.prototype.open;
+window._message = "";
+XMLHttpRequest.prototype.open = function(method, url) {
+    if (url.includes("/assistant.lamda.BardFrontendService/StreamGenerate")) {
+        this.addEventListener("load", (event) => {
+            window._message = JSON.parse(JSON.parse(this.responseText.split("\\n")[3])[0][2])[4][0][1][0];
+        });
     }
-
+    return _http_request_open.call(this, method, url);
-            data = {
-                'at': cls._snlm0e,
-                'f.req': json.dumps([None, json.dumps([[prompt]])])
 }
+"""
+            driver.execute_script(script)
 
-            intents = '.'.join([
+            # Send prompt
-                'assistant',
+            driver.find_element(By.CSS_SELECTOR, "div.ql-editor.ql-blank.textarea").send_keys(prompt)
-                'lamda',
+            driver.find_element(By.CSS_SELECTOR, "button.send-button").click()
-                'BardFrontendService'
-            ])
-            async with session.post(
-                f'{cls.url}/_/BardChatUi/data/{intents}/StreamGenerate',
-                data=data,
-                params=params,
-                proxy=proxy
-            ) as response:
-                response = await response.text()
-                response = json.loads(response.splitlines()[3])[0][2]
-                response = json.loads(response)[4][0][1][0]
-                return response
 
-    @classmethod
+            # Read response
-    @property
+            script = "return window._message;"
-    def params(cls):
+            while True:
-        params = [
+                chunk = driver.execute_script(script)
-            ("model", "str"),
+                if chunk:
-            ("messages", "list[dict[str, str]]"),
+                    yield chunk
-            ("stream", "bool"),
+                    return
-            ("proxy", "str"),
+                else:
-        ]
+                    time.sleep(0.1)
-        param = ", ".join([": ".join(p) for p in params])
+        finally:
-        return f"g4f.provider.{cls.__name__} supports: ({param})"
+            driver.close()
+            if not browser:
+                time.sleep(0.1)
+                driver.quit()
+            if display:
+                display.stop()

g4f/models.py

@@ -5,7 +5,6 @@ from .Provider   import BaseProvider, RetryProvider
 from .Provider   import (
     Chatgpt4Online,
     ChatAnywhere,
-    PerplexityAi,
     GptForLove,
     ChatgptAi,
     DeepInfra,
@@ -77,7 +76,7 @@ gpt_4 = Model(
     name          = 'gpt-4',
     base_provider = 'openai',
     best_provider = RetryProvider([
-        Bing, Phind, PerplexityAi
+        Bing, Phind
     ])
 )