majorchik-api 0.0.2a2__tar.gz → 0.0.3__tar.gz

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/PKG-INFO

@@ -1,11 +1,12 @@
 Metadata-Version: 2.4
 Name: majorchik-api
-Version: 0.0.2a2
+Version: 0.0.3
 Summary: Пакет для взаимодействия с современными нейросетями (Gemini, DeepSeek) через автоматизацию браузера
 Author-email: GO Software <gosoftware2025@gmail.com>
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: playwright>=1.30.0
+Requires-Dist: playwright-stealth
 
 # Majorchik API (Browser AI)
 

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "majorchik-api"
-version = "0.0.2a2"
+version = "0.0.3"
 authors = [
     {name = "GO Software", email = "gosoftware2025@gmail.com"},
 ]
@@ -13,6 +13,7 @@ readme = "README.md"
 requires-python = ">=3.8"
 dependencies =[
     "playwright>=1.30.0",
+    "playwright-stealth"
 ]
 
 [project.scripts]

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/src/browser_ai/base_api.py

@@ -8,12 +8,13 @@ class BaseWebAPI(ABC):
     Базовый класс для всех Web API моделей.
     Инкапсулирует логику запуска браузера, маскировки и управления ресурсами.
     """
-    def __init__(self, profile_dir: str, url: str, headless: bool = True, disable_images: bool = True, setup_mode: bool = False, **kwargs):
+    def __init__(self, profile_dir: str, url: str, headless: bool = True, disable_images: bool = True, setup_mode: bool = False, apply_stealth: bool = True, **kwargs):
         self.profile_dir = os.path.abspath(profile_dir)
         self.url = url
         self.headless = headless
         self.disable_images = disable_images
         self.setup_mode = setup_mode
+        self.apply_stealth = apply_stealth
         self.extra_kwargs = kwargs
         
         if not self.setup_mode:
@@ -31,8 +32,9 @@ class BaseWebAPI(ABC):
         
         args =[
             "--disable-blink-features=AutomationControlled",
-            "--no-sandbox",
-            "--disable-infobars"
+            "--disable-infobars",
+            "--disable-quic",
+            "--test-type"
         ]
         
         if self.disable_images:
@@ -42,14 +44,33 @@ class BaseWebAPI(ABC):
             user_data_dir=self.profile_dir,
             headless=self.headless,
             channel="chrome",
-            ignore_default_args=["--enable-automation"],
+            ignore_default_args=["--enable-automation", "--no-sandbox"],
             args=args,
-            viewport={'width': 1280, 'height': 800},
+            no_viewport=True,
             permissions=["clipboard-read", "clipboard-write"]
         )
-        self.page = self.browser.new_page()
+        
+        # Используем дефолтную вкладку, если она есть, вместо открытия новой
+        if len(self.browser.pages) > 0:
+            self.page = self.browser.pages[0]
+        else:
+            self.page = self.browser.new_page()
+            
+        # Включаем маскировку (Stealth)
         self.page.add_init_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})")
         
+        if self.apply_stealth:
+            try:
+                try:
+                    from playwright_stealth import stealth_sync
+                    stealth_sync(self.page)
+                except ImportError:
+                    from playwright_stealth import Stealth
+                    Stealth().apply_stealth_sync(self.page)
+            except Exception as e:
+                print(f"Предупреждение: playwright-stealth не загрузился. Ошибка: {e}")
+                print("Подсказка: возможно, пакет установлен глобально, а IDE использует изолированное окружение (.venv).")
+            
         self.page.goto(self.url, wait_until="domcontentloaded")
         
         if not self.setup_mode:

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/src/browser_ai/cli.py

@@ -9,7 +9,7 @@ def cmd_setup(args):
     print("Открываем браузер...")
     
     try:
-        api = get_ai_client(args.model, profile_dir=args.profile_dir, headless=False, setup_mode=True)
+        api = get_ai_client(args.model, profile_dir=args.profile_dir, headless=False, setup_mode=True, model_id=args.model_id)
             
         print("\nПожалуйста, авторизуйтесь и пройдите все проверки (включая Cloudflare).")
         print("Убедитесь, что интерфейс чата полностью загружен.")
@@ -25,7 +25,7 @@ def cmd_chat(args):
     print(f"\nИнициализация {args.model.upper()} (профиль: {args.profile_dir}, headless: {args.headless})...")
     ai = None
     try:
-        ai = get_ai_client(args.model, profile_dir=args.profile_dir, headless=args.headless)
+        ai = get_ai_client(args.model, profile_dir=args.profile_dir, headless=args.headless, model_id=args.model_id)
 
         print("Браузер готов. Вводите промпты (для выхода введите 'exit' или 'quit').")
         print("-" * 50)
@@ -76,6 +76,7 @@ def main():
     parser_setup = subparsers.add_parser("setup", help="Создать и настроить профиль браузера (авторизация)")
     parser_setup.add_argument("model", choices=models_list, help="Модель для настройки")
     parser_setup.add_argument("--profile-dir", required=True, help="Путь к директории для сохранения профиля")
+    parser_setup.add_argument("--model-id", type=str, default=None, help="Специфичный ID (напр. gemini-2.5-flash-image)")
 
     # Команда chat
     parser_chat = subparsers.add_parser("chat", help="Запустить интерактивный чат с нейросетью")
@@ -83,6 +84,7 @@ def main():
     parser_chat.add_argument("--profile-dir", required=True, help="Путь к директории сохраненного профиля")
     parser_chat.add_argument("--headless", action="store_true", help="Скрыть визуально браузер (тихий режим)")
     parser_chat.add_argument("--files", nargs="*", help="Пути к файлам для прикрепления к первому промпту")
+    parser_chat.add_argument("--model-id", type=str, default=None, help="Специфичный ID (напр. gemini-2.5-flash-image)")
 
     args = parser.parse_args()
 

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/src/browser_ai/deepseek_api.py

@@ -2,6 +2,7 @@ from .base_api import BaseWebAPI
 from .exceptions import AuthRequiredError, FirstTokenTimeoutError, ResponseTimeoutError
 import time
 import re
+import sys
 
 def _clean_text(text: str) -> str:
     return re.sub(r'\n{3,}', '\n\n', text).strip()
@@ -130,9 +131,20 @@ class DeepSeekAPI(BaseWebAPI):
             initial_last_text = ""
 
         chat_input = self.page.locator("#chat-input, textarea, [contenteditable='true']").first
-        chat_input.fill(prompt)
+        chat_input.click(delay=30)
+        time.sleep(0.3)
+        
+        if len(prompt) > 200:
+            self.page.evaluate("text => navigator.clipboard.writeText(text)", prompt)
+            modifier = "Meta" if sys.platform == "darwin" else "Control"
+            chat_input.press(f"{modifier}+v", delay=50)
+            time.sleep(0.2)
+            chat_input.press_sequentially(" ", delay=30)
+        else:
+            chat_input.press_sequentially(prompt, delay=15)
+            
         time.sleep(0.5)
-        chat_input.press("Enter")
+        chat_input.press("Enter", delay=40)
 
         last_text = ""
         stable_count = 0
@@ -144,7 +156,7 @@ class DeepSeekAPI(BaseWebAPI):
             if (mds.length === 0) return false;
             const lastMd = mds[mds.length - 1];
             let curr = lastMd;
-            for (let i = 0; i < 5; i++) {
+            for (let i = 0; i < 7; i++) {
                 if (!curr) break;
                 let sibling = curr.nextElementSibling;
                 while (sibling) {
@@ -154,7 +166,7 @@ class DeepSeekAPI(BaseWebAPI):
                         sibling = sibling.nextElementSibling;
                         continue;
                     }
-                    const btns = sibling.querySelectorAll('div.ds-icon-button[role="button"]');
+                    const btns = sibling.querySelectorAll('div[role="button"][class*="ds-icon-button"], div[role="button"][class*="ds-button"]');
                     if (btns.length >= 2) {
                         btns[0].click();
                         return true;

majorchik_api-0.0.3/src/browser_ai/google_ai_studio_api.py

@@ -0,0 +1,281 @@
+from .base_api import BaseWebAPI
+from .exceptions import AuthRequiredError, FirstTokenTimeoutError, ResponseTimeoutError, QuotaExceededError
+import time
+import re
+import sys
+
+def _clean_text(text: str) -> str:
+    return re.sub(r'\n{3,}', '\n\n', text).strip()
+
+class GoogleAIStudioAPI(BaseWebAPI):
+    def __init__(self, profile_dir: str, headless: bool = True, setup_mode: bool = False, **kwargs):
+        model_id = kwargs.get("model_id")
+        base_url = "https://aistudio.google.com/prompts/new_chat"
+        url = f"{base_url}?model={model_id}" if model_id else base_url
+        
+        # Включаем картинки, так как они нужны для работы моделей генерирующих изображения (gemini-2.5-flash-image)
+        disable_images = kwargs.get("disable_images", False)
+
+        super().__init__(
+            profile_dir=profile_dir,
+            url=url,
+            headless=headless,
+            disable_images=disable_images,
+            setup_mode=setup_mode,
+            apply_stealth=False,  # Отключаем stealth, так как он ломает фронтенд Google AI Studio
+            **kwargs
+        )
+
+    def wait_for_ready(self):
+        for _ in range(30):
+            if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
+                raise AuthRequiredError("Требуется повторная авторизация в Google AI Studio.")
+            if self.page.locator("textarea,[contenteditable='true']").count() > 0:
+                return
+            time.sleep(1)
+        self.page.wait_for_selector("textarea, [contenteditable='true']", timeout=1000)
+
+    def ask(self, prompt: str, files: list = None, timeout: int = 600, first_token_timeout: int = 60, headless: bool = None, **kwargs) -> str:
+        if headless is not None and headless != self.headless:
+            self._start_browser(headless)
+            
+        overall_start_time = time.time()
+        if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
+            raise AuthRequiredError("Сессия истекла. Требуется повторная авторизация в Google AI Studio.")
+
+        self.page.evaluate("navigator.clipboard.writeText('').catch(() => {})")
+        
+        import random
+        
+        # 1. Разогреваем сессию случайными движениями мыши
+        try:
+            for _ in range(3):
+                self.page.mouse.move(random.randint(200, 800), random.randint(200, 600), steps=10)
+                self.page.wait_for_timeout(random.randint(100, 300))
+        except:
+            pass
+
+        if files:
+            try:
+                attach_btn = self.page.locator('[data-test-id="add-media-button"], button[aria-label*="Insert"], button[aria-label*="Attach"], [data-test-id="file-upload-button"]').first
+                # Эмулируем реальное наведение мыши перед кликом
+                if attach_btn.count() > 0 and attach_btn.is_visible():
+                    box = attach_btn.bounding_box()
+                    if box:
+                        self.page.mouse.move(box['x'] + box['width'] / 2, box['y'] + box['height'] / 2, steps=10)
+                        self.page.wait_for_timeout(200)
+                        self.page.mouse.click(box['x'] + box['width'] / 2, box['y'] + box['height'] / 2, delay=random.randint(40, 80))
+                    else:
+                        attach_btn.click(delay=50)
+                else:
+                    attach_btn.click(delay=50)
+                    
+                file_input = self.page.locator('input[type="file"]')
+                file_input.wait_for(state="attached", timeout=5000)
+                file_input.set_input_files(files)
+                
+                time.sleep(1.5)
+                loading_texts = ["Loading", "В ожидании", "Загрузка", "Uploading"]
+                for text in loading_texts:
+                    try:
+                        for el in self.page.get_by_text(text).all():
+                            if el.is_visible():
+                                el.wait_for(state="hidden", timeout=30000)
+                    except:
+                        pass
+                time.sleep(1)
+            except Exception as e:
+                print(f"Не удалось прикрепить файлы: {e}")
+
+        try:
+            initial_thumb_count = self.page.locator('button[aria-label="Good response"], .turn-footer button, button:has(.material-symbols-outlined:has-text("thumb_up"))').count()
+            initial_model_turns = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').count()
+        except:
+            initial_thumb_count = 0
+            initial_model_turns = 0
+
+        self.page.keyboard.press("Escape")
+        self.page.wait_for_timeout(300)
+
+        chat_input = self.page.locator("textarea, [contenteditable='true'], [role='textbox']").first
+        
+        # Эмулируем наведение мыши на поле ввода
+        if chat_input.count() > 0 and chat_input.is_visible():
+            box = chat_input.bounding_box()
+            if box:
+                self.page.mouse.move(box['x'] + 20, box['y'] + 10, steps=10)
+                self.page.wait_for_timeout(random.randint(100, 300))
+                self.page.mouse.click(box['x'] + 20, box['y'] + 10, delay=random.randint(30, 70))
+            else:
+                chat_input.click(delay=40)
+        else:
+            chat_input.click(delay=40)
+            
+        self.page.wait_for_timeout(300)
+
+        # Вставка из буфера, как ты делал вручную (это безопаснее, чем эмуляция сотен нажатий без ошибок)
+        self.page.evaluate("text => navigator.clipboard.writeText(text)", prompt)
+        modifier = "Meta" if sys.platform == "darwin" else "Control"
+        chat_input.press(f"{modifier}+v", delay=random.randint(40, 90))
+        
+        self.page.wait_for_timeout(random.randint(300, 600))
+        
+        # Делаем случайное движение перед нажатием кнопки отправки
+        self.page.mouse.move(random.randint(300, 700), random.randint(200, 500), steps=10)
+        self.page.wait_for_timeout(200)
+
+        try:
+            run_btn = self.page.locator('button:has-text("Run")').last
+            if run_btn.count() > 0 and run_btn.is_visible(timeout=500):
+                box = run_btn.bounding_box()
+                if box:
+                    self.page.mouse.move(box['x'] + box['width']/2, box['y'] + box['height']/2, steps=10)
+                    self.page.wait_for_timeout(200)
+                    self.page.mouse.click(box['x'] + box['width']/2, box['y'] + box['height']/2, delay=random.randint(30, 80))
+                else:
+                    run_btn.click(delay=50)
+            else:
+                chat_input.press("Control+Enter", delay=random.randint(40, 80))
+        except:
+            chat_input.press("Control+Enter", delay=random.randint(40, 80))
+
+        first_token_start_time = time.time()
+        completed = False
+        first_token_received = False
+
+        while True:
+            if time.time() - overall_start_time > timeout:
+                raise ResponseTimeoutError(f"Превышено максимальное время ожидания ответа ({timeout} сек).")
+            if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
+                raise AuthRequiredError("Сессия истекла. Требуется повторная авторизация в Google AI Studio.")
+            try:
+                error_indicators = [
+                    "user has exceeded quota",
+                    "You've reached your rate limit",
+                    "You've reached your quota for the day"
+                ]
+                for err_text in error_indicators:
+                    err_loc = self.page.get_by_text(err_text)
+                    if err_loc.count() > 0 and err_loc.first.is_visible():
+                        raise QuotaExceededError("Лимит запросов (токены) исчерпан в Google AI Studio.")
+            except QuotaExceededError:
+                raise
+            except Exception:
+                pass
+            try:
+                skip_btn = self.page.locator('button:has-text("Skip")')
+                if skip_btn.count() > 0:
+                    if skip_btn.last.is_visible():
+                        skip_btn.last.evaluate("el => el.click()")
+                        self.page.wait_for_timeout(500)
+            except:
+                pass
+            
+            if not first_token_received:
+                try:
+                    current_model_turns = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').count()
+                    if current_model_turns > initial_model_turns:
+                        first_token_received = True
+                    elif time.time() - first_token_start_time > first_token_timeout:
+                        raise FirstTokenTimeoutError(f"Модель не начала генерацию ответа за {first_token_timeout} секунд.")
+                except:
+                    pass
+
+            try:
+                current_thumb_count = self.page.locator('button[aria-label="Good response"], .turn-footer button, button:has(.material-symbols-outlined:has-text("thumb_up"))').count()
+                if current_thumb_count > initial_thumb_count:
+                    completed = True
+                    break
+            except:
+                pass
+            self.page.wait_for_timeout(1000)
+
+        if not completed:
+            print("Предупреждение: Не удалось дождаться кнопки thumb_up, пробуем извлечь текущий ответ (модель может быть image-генератором).")
+
+        self.page.wait_for_timeout(2000) # Даем UI немного времени на финальный рендеринг картинок
+
+        try:
+            last_turn = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').last
+            
+            # 0. Извлекаем сгенерированные изображения
+            js_extract_images = """
+            async (el) => {
+                const imgs = el.querySelectorAll('img:not(.avatar)');
+                let res = [];
+                for (let img of imgs) {
+                    let src = img.src;
+                    if (!src || src.startsWith('data:image/svg') || src.includes('gstatic.com')) continue;
+                    
+                    if (src.startsWith('blob:')) {
+                        try {
+                            let blob = await fetch(src).then(r => r.blob());
+                            let b64 = await new Promise(resolve => {
+                                let reader = new FileReader();
+                                reader.onloadend = () => resolve(reader.result);
+                                reader.readAsDataURL(blob);
+                            });
+                            res.push(b64);
+                        } catch(e) {
+                            res.push(src);
+                        }
+                    } else {
+                        res.push(src);
+                    }
+                }
+                const canvases = el.querySelectorAll('canvas');
+                canvases.forEach(c => {
+                    try { res.push(c.toDataURL('image/png')); } catch(e) {}
+                });
+                return res;
+            }
+            """
+            
+            extracted_images = []
+            try:
+                extracted_images = last_turn.evaluate(js_extract_images)
+            except Exception as e:
+                pass
+                
+            image_append_text = ""
+            if extracted_images:
+                image_append_text = "\n\n[Сгенерированные изображения (Base64/URL)]:\n" + "\n\n".join(extracted_images)
+
+            # Попытка 1: найти кнопку меню и скопировать как Markdown
+            copy_success = False
+            try:
+                options_btn = last_turn.locator('button[aria-label="Open options"], button:has(.material-symbols-outlined:has-text("more_vert"))').last
+                if options_btn.count() > 0 and options_btn.is_visible():
+                    options_btn.evaluate("el => el.click()")
+                    self.page.wait_for_timeout(500)
+                    
+                    copy_btn = self.page.locator('button[role="menuitem"]:has-text("Copy as markdown"), button[role="menuitem"]:has-text("Copy")').first
+                    if copy_btn.count() > 0 and copy_btn.is_visible():
+                        copy_btn.evaluate("el => el.click()")
+                        self.page.wait_for_timeout(1000)
+                        copy_success = True
+            except:
+                pass
+
+            if copy_success:
+                clip_val = self.page.evaluate("navigator.clipboard.readText().catch(() => '')")
+                if clip_val or extracted_images:
+                    return clip_val + image_append_text
+            
+            # Попытка 2: Читаем текст прямо из DOM
+            try:
+                last_turn.evaluate("""(el) => {
+                    const footers = el.querySelectorAll('.turn-footer, .actions-container');
+                    footers.forEach(f => f.style.display = 'none');
+                }""")
+            except:
+                pass
+                
+            text = last_turn.inner_text()
+            if text.strip() or extracted_images:
+                return _clean_text(text) + image_append_text
+                
+        except Exception as e:
+            return f"Ошибка при извлечении текста/изображений: {e}"
+
+        return "Не удалось получить ответ (вероятно, изменения в верстке Google AI Studio)."

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/src/majorchik_api.egg-info/PKG-INFO

@@ -1,11 +1,12 @@
 Metadata-Version: 2.4
 Name: majorchik-api
-Version: 0.0.2a2
+Version: 0.0.3
 Summary: Пакет для взаимодействия с современными нейросетями (Gemini, DeepSeek) через автоматизацию браузера
 Author-email: GO Software <gosoftware2025@gmail.com>
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: playwright>=1.30.0
+Requires-Dist: playwright-stealth
 
 # Majorchik API (Browser AI)
 

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/src/majorchik_api.egg-info/requires.txt

@@ -1 +1,2 @@
 playwright>=1.30.0
+playwright-stealth

{majorchik_api-0.0.2a2 → majorchik_api-0.0.3}/tests/test_upload.py

@@ -22,11 +22,16 @@ def main():
     
     profile_dir = input("Введите абсолютный или относительный путь до папки профиля (например ./my_profiles/google_ai_studio_default): ").strip()
 
+    model_id = None
+    if choice == "1":
+        model_id = input("Введите ID модели или нажмите Enter для стандартной (например gemini-2.5-flash-image): ").strip()
+        model_id = model_id if model_id else None
+
     ai = None
     try:
         if choice == "1":
             print("Запускаем Google AI Studio (без headless, чтобы видеть процесс)...")
-            ai = get_ai_client("google_ai_studio", profile_dir=profile_dir, headless=False)
+            ai = get_ai_client("google_ai_studio", profile_dir=profile_dir, headless=False, model_id=model_id)
         elif choice == "2":
             print("Запускаем DeepSeek (без headless, чтобы видеть процесс)...")
             ai = get_ai_client("deepseek", profile_dir=profile_dir, headless=False)

majorchik_api-0.0.2a2/src/browser_ai/google_ai_studio_api.py

@@ -1,169 +0,0 @@
-from .base_api import BaseWebAPI
-from .exceptions import AuthRequiredError, FirstTokenTimeoutError, ResponseTimeoutError, QuotaExceededError
-import time
-import re
-
-def _clean_text(text: str) -> str:
-    return re.sub(r'\n{3,}', '\n\n', text).strip()
-
-class GoogleAIStudioAPI(BaseWebAPI):
-    def __init__(self, profile_dir: str, headless: bool = True, setup_mode: bool = False, **kwargs):
-        super().__init__(
-            profile_dir=profile_dir,
-            url="https://aistudio.google.com/app/prompts/new_chat",
-            headless=headless,
-            disable_images=True,
-            setup_mode=setup_mode,
-            **kwargs
-        )
-
-    def wait_for_ready(self):
-        for _ in range(30):
-            if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
-                raise AuthRequiredError("Требуется повторная авторизация в Google AI Studio.")
-            if self.page.locator("textarea,[contenteditable='true']").count() > 0:
-                return
-            time.sleep(1)
-        self.page.wait_for_selector("textarea, [contenteditable='true']", timeout=1000)
-
-    def ask(self, prompt: str, files: list = None, timeout: int = 600, first_token_timeout: int = 60, headless: bool = None, **kwargs) -> str:
-        if headless is not None and headless != self.headless:
-            self._start_browser(headless)
-            
-        overall_start_time = time.time()
-        if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
-            raise AuthRequiredError("Сессия истекла. Требуется повторная авторизация в Google AI Studio.")
-
-        self.page.evaluate("navigator.clipboard.writeText('').catch(() => {})")
-
-        if files:
-            try:
-                self.page.locator('[data-test-id="add-media-button"], button[aria-label*="Insert"]').first.click()
-                file_input = self.page.locator('input[type="file"]')
-                file_input.wait_for(state="attached", timeout=5000)
-                file_input.set_input_files(files)
-                
-                time.sleep(1.5)
-                loading_texts =["Loading", "В ожидании", "Загрузка", "Uploading"]
-                for text in loading_texts:
-                    try:
-                        for el in self.page.get_by_text(text).all():
-                            if el.is_visible():
-                                el.wait_for(state="hidden", timeout=30000)
-                    except:
-                        pass
-                time.sleep(1)
-            except Exception as e:
-                print(f"Не удалось прикрепить файлы: {e}")
-
-        try:
-            initial_thumb_count = self.page.locator('button[aria-label="Good response"], .turn-footer button, button:has(.material-symbols-outlined:has-text("thumb_up"))').count()
-            initial_model_turns = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').count()
-        except:
-            initial_thumb_count = 0
-            initial_model_turns = 0
-
-        self.page.keyboard.press("Escape")
-        self.page.wait_for_timeout(300)
-
-        chat_input = self.page.locator("textarea,[contenteditable='true'],[role='textbox']").first
-        chat_input.click(force=True)
-        chat_input.fill(prompt)
-        self.page.wait_for_timeout(500)
-
-        chat_input.press("Control+Enter")
-
-        first_token_start_time = time.time()
-        completed = False
-        first_token_received = False
-
-        while True:
-            if time.time() - overall_start_time > timeout:
-                raise ResponseTimeoutError(f"Превышено максимальное время ожидания ответа ({timeout} сек).")
-            if "accounts.google.com" in self.page.url or "signin" in self.page.url.lower():
-                raise AuthRequiredError("Сессия истекла. Требуется повторная авторизация в Google AI Studio.")
-            try:
-                error_indicators =[
-                    "user has exceeded quota",
-                    "You've reached your rate limit",
-                    "You've reached your quota for the day"
-                ]
-                for err_text in error_indicators:
-                    err_loc = self.page.get_by_text(err_text)
-                    if err_loc.count() > 0 and err_loc.first.is_visible():
-                        raise QuotaExceededError("Лимит запросов (токены) исчерпан в Google AI Studio.")
-            except QuotaExceededError:
-                raise
-            except Exception:
-                pass
-            try:
-                skip_btn = self.page.locator('button:has-text("Skip")')
-                if skip_btn.count() > 0:
-                    if skip_btn.last.is_visible():
-                        skip_btn.last.click(timeout=1000, force=True)
-                        self.page.wait_for_timeout(500)
-            except:
-                pass
-            if not first_token_received:
-                try:
-                    current_model_turns = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').count()
-                    if current_model_turns > initial_model_turns:
-                        first_token_received = True
-                    elif time.time() - first_token_start_time > first_token_timeout:
-                        raise FirstTokenTimeoutError(f"Модель не начала генерацию ответа за {first_token_timeout} секунд.")
-                except:
-                    pass
-
-            try:
-                current_thumb_count = self.page.locator('button[aria-label="Good response"], .turn-footer button, button:has(.material-symbols-outlined:has-text("thumb_up"))').count()
-                if current_thumb_count > initial_thumb_count:
-                    completed = True
-                    break
-            except:
-                pass
-            self.page.wait_for_timeout(1000)
-
-        if not completed:
-            return "Не удалось дождаться завершения генерации (кнопка thumb_up не появилась)."
-
-        self.page.wait_for_timeout(1000)
-
-        try:
-            last_turn = self.page.locator('ms-chat-turn, div[data-turn-role="Model"], div.chat-turn-container.model').last
-            
-            copy_success = False
-            try:
-                options_btn = last_turn.locator('button[aria-label="Open options"], button:has(.material-symbols-outlined:has-text("more_vert"))').last
-                if options_btn.count() > 0 and options_btn.is_visible():
-                    options_btn.click(timeout=3000, force=True)
-                    self.page.wait_for_timeout(500)
-                    
-                    copy_btn = self.page.locator('button[role="menuitem"]:has-text("Copy as markdown"), button[role="menuitem"]:has-text("Copy")').first
-                    if copy_btn.count() > 0 and copy_btn.is_visible():
-                        copy_btn.click(timeout=3000, force=True)
-                        self.page.wait_for_timeout(1000)
-                        copy_success = True
-            except:
-                pass
-
-            if copy_success:
-                clip_val = self.page.evaluate("navigator.clipboard.readText().catch(() => '')")
-                if clip_val:
-                    return clip_val
-            
-            try:
-                last_turn.evaluate("""(el) => {
-                    const footers = el.querySelectorAll('.turn-footer, .actions-container');
-                    footers.forEach(f => f.style.display = 'none');
-                }""")
-            except:
-                pass
-                
-            text = last_turn.inner_text()
-            if text.strip():
-                return _clean_text(text)
-                
-        except Exception as e:
-            return f"Ошибка при извлечении текста: {e}"
-
-        return "Не удалось получить текст ответа."