From 38dbe4b8e5ca7f9bc0508e1ba1bf878fd6d8c19c Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Mon, 22 Jan 2024 03:38:11 +0100
Subject: Improve CreateImagesBing Sort providers by category

---
 g4f/Provider/selenium/AItianhuSpace.py | 117 +++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 g4f/Provider/selenium/AItianhuSpace.py

(limited to 'g4f/Provider/selenium/AItianhuSpace.py')

diff --git a/g4f/Provider/selenium/AItianhuSpace.py b/g4f/Provider/selenium/AItianhuSpace.py
new file mode 100644
index 00000000..9878bd5a
--- /dev/null
+++ b/g4f/Provider/selenium/AItianhuSpace.py
@@ -0,0 +1,117 @@
+from __future__ import annotations
+
+import time
+import random
+
+from ...typing import CreateResult, Messages
+from ..base_provider import AbstractProvider
+from ..helper import format_prompt, get_random_string
+from ...webdriver import WebDriver, WebDriverSession
+from ... import debug
+
+class AItianhuSpace(AbstractProvider):
+    url = "https://chat3.aiyunos.top/"
+    working = True
+    supports_stream = True
+    supports_gpt_35_turbo = True
+    _domains = ["aitianhu.com", "aitianhu1.top"]
+
+    @classmethod
+    def create_completion(
+        cls,
+        model: str,
+        messages: Messages,
+        stream: bool,
+        domain: str = None,
+        proxy: str = None,
+        timeout: int = 120,
+        webdriver: WebDriver = None,
+        headless: bool = True,
+        **kwargs
+    ) -> CreateResult:
+        if not model:
+            model = "gpt-3.5-turbo"
+        if not domain:
+            rand = get_random_string(6)
+            domain = random.choice(cls._domains)
+            domain = f"{rand}.{domain}"
+        if debug.logging:
+            print(f"AItianhuSpace | using domain: {domain}")
+        url = f"https://{domain}"
+        prompt = format_prompt(messages)
+
+        with WebDriverSession(webdriver, "", headless=headless, proxy=proxy) as driver:
+            from selenium.webdriver.common.by import By
+            from selenium.webdriver.support.ui import WebDriverWait
+            from selenium.webdriver.support import expected_conditions as EC
+
+            wait = WebDriverWait(driver, timeout)
+
+            # Bypass devtools detection
+            driver.get("https://blank.page/")
+            wait.until(EC.visibility_of_element_located((By.ID, "sheet")))
+            driver.execute_script(f"""
+    document.getElementById('sheet').addEventListener('click', () => {{
+        window.open('{url}', '_blank');
+    }});
+    """)
+            driver.find_element(By.ID, "sheet").click()
+            time.sleep(10)
+
+            original_window = driver.current_window_handle
+            for window_handle in driver.window_handles:
+                if window_handle != original_window:
+                    driver.close()
+                    driver.switch_to.window(window_handle)
+                    break
+
+            # Wait for page load
+            wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "textarea.n-input__textarea-el")))
+
+            # Register hook in XMLHttpRequest
+            script = """
+const _http_request_open = XMLHttpRequest.prototype.open;
+window._last_message = window._message = "";
+window._loadend = false;
+XMLHttpRequest.prototype.open = function(method, url) {
+    if (url == "/api/chat-process") {
+        this.addEventListener("progress", (event) => {
+            const lines = this.responseText.split("\\n");
+            try {
+                window._message = JSON.parse(lines[lines.length-1])["text"];
+            } catch(e) { }
+        });
+        this.addEventListener("loadend", (event) => {
+            window._loadend = true;
+        });
+    }
+    return _http_request_open.call(this, method, url);
+}
+"""
+            driver.execute_script(script)
+
+            # Submit prompt
+            driver.find_element(By.CSS_SELECTOR, "textarea.n-input__textarea-el").send_keys(prompt)
+            driver.find_element(By.CSS_SELECTOR, "button.n-button.n-button--primary-type.n-button--medium-type").click()
+
+            # Read response
+            while True:
+                chunk = driver.execute_script("""
+if (window._message && window._message != window._last_message) {
+    try {
+        return window._message.substring(window._last_message.length);
+    } finally {
+        window._last_message = window._message;
+    }
+}
+if (window._loadend) {
+    return null;
+}
+return "";
+""")
+                if chunk:
+                    yield chunk
+                elif chunk != "":
+                    break
+                else:
+                    time.sleep(0.1)
\ No newline at end of file
-- 
cgit v1.2.3