majorchik-api 0.0.0a0__py3-none-any.whl

browser_ai/__init__.py

@@ -0,0 +1,5 @@
+from .gemini_api import GeminiAPI
+from .deepseek_api import DeepSeekAPI
+from .exceptions import ProfileNotFoundError
+
+__all__ =["GeminiAPI", "DeepSeekAPI", "ProfileNotFoundError"]

browser_ai/base_api.py

@@ -0,0 +1,62 @@
+from abc import ABC, abstractmethod
+from playwright.sync_api import sync_playwright
+import os
+from .exceptions import ProfileNotFoundError
+
+class BaseWebAPI(ABC):
+    """
+    Базовый класс для всех Web API моделей.
+    Инкапсулирует логику запуска браузера, маскировки и управления ресурсами.
+    """
+    def __init__(self, profile_dir: str, url: str, headless: bool = True, disable_images: bool = True, setup_mode: bool = False):
+        self.profile_dir = os.path.abspath(profile_dir)
+        self.url = url
+        self.headless = headless
+        
+        if not setup_mode:
+            if not os.path.exists(self.profile_dir) or not os.listdir(self.profile_dir):
+                raise ProfileNotFoundError(f"Профиль не найден: {self.profile_dir}. Сначала создайте его через режим setup.")
+        else:
+            os.makedirs(self.profile_dir, exist_ok=True)
+            
+        self.playwright = sync_playwright().start()
+        
+        args =[
+            "--disable-blink-features=AutomationControlled",
+            "--no-sandbox",
+            "--disable-infobars"
+        ]
+        
+        if disable_images:
+            args.append("--blink-settings=imagesEnabled=false")
+            
+        self.browser = self.playwright.chromium.launch_persistent_context(
+            user_data_dir=self.profile_dir,
+            headless=headless,
+            channel="chrome",
+            ignore_default_args=["--enable-automation"],
+            args=args,
+            viewport={'width': 1280, 'height': 800},
+            permissions=["clipboard-read", "clipboard-write"]
+        )
+        self.page = self.browser.new_page()
+        self.page.add_init_script("Object.defineProperty(navigator, 'webdriver', {get: () => undefined})")
+        
+        self.page.goto(self.url, wait_until="domcontentloaded")
+        
+        if not setup_mode:
+            self.wait_for_ready()
+
+    @abstractmethod
+    def wait_for_ready(self):
+        pass
+
+    @abstractmethod
+    def ask(self, prompt: str, files: list = None) -> str:
+        pass
+
+    def close(self):
+        if hasattr(self, 'browser') and self.browser:
+            self.browser.close()
+        if hasattr(self, 'playwright') and self.playwright:
+            self.playwright.stop()

browser_ai/cli.py

@@ -0,0 +1,93 @@
+import sys
+import argparse
+import os
+from .gemini_api import GeminiAPI
+from .deepseek_api import DeepSeekAPI
+from .exceptions import ProfileNotFoundError
+
+def cmd_setup(args):
+    print(f"\n[Настройка профиля {args.model.upper()} | Директория: {args.profile_dir}]")
+    print("Открываем браузер...")
+    
+    try:
+        if args.model == "gemini":
+            api = GeminiAPI(profile_dir=args.profile_dir, headless=False, setup_mode=True)
+        else:
+            api = DeepSeekAPI(profile_dir=args.profile_dir, headless=False, setup_mode=True)
+            
+        print("\nПожалуйста, авторизуйтесь и пройдите все проверки (включая Cloudflare).")
+        print("Убедитесь, что интерфейс чата полностью загружен.")
+        input("\nНажми Enter в этой консоли для сохранения профиля и выхода...")
+        
+        api.close()
+        print(f"Профиль {args.model} успешно сохранен в: {args.profile_dir}")
+    except Exception as e:
+        print(f"Ошибка при настройке: {e}")
+
+
+def cmd_chat(args):
+    print(f"\nИнициализация {args.model.upper()} (профиль: {args.profile_dir}, headless: {args.headless})...")
+    ai = None
+    try:
+        if args.model == "gemini":
+            ai = GeminiAPI(profile_dir=args.profile_dir, headless=args.headless)
+        else:
+            ai = DeepSeekAPI(profile_dir=args.profile_dir, headless=args.headless)
+
+        print("Браузер готов. Вводите промпты (для выхода введите 'exit' или 'quit').")
+        print("-" * 50)
+
+        current_files = [os.path.abspath(f) for f in args.files] if args.files else None
+
+        while True:
+            prompt = input("\nПромпт: ").strip()
+            if prompt.lower() in ("exit", "quit"):
+                print("Завершение работы.")
+                break
+            if not prompt:
+                print("Промпт не может быть пустым.")
+                continue
+
+            print("\n[Ответ]:")
+            response = ai.ask(prompt, files=current_files)
+            print(response)
+            current_files = None
+
+    except ProfileNotFoundError as e:
+        print(f"\n[ОШИБКА] {e}")
+        print(f"Для создания профиля выполните команду:")
+        print(f"  browser-ai setup {args.model} --profile-dir \"{args.profile_dir}\"")
+    except KeyboardInterrupt:
+        print("\n\nПрерывание пользователя.")
+    except Exception as e:
+        print(f"\nПроизошла ошибка: {e}")
+    finally:
+        if ai:
+            ai.close()
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Browser AI - Пакет для работы с Gemini и DeepSeek.")
+    subparsers = parser.add_subparsers(dest="command", required=True)
+
+    # Команда setup
+    parser_setup = subparsers.add_parser("setup", help="Создать и настроить профиль браузера (авторизация)")
+    parser_setup.add_argument("model", choices=["gemini", "deepseek"], help="Модель для настройки")
+    parser_setup.add_argument("--profile-dir", required=True, help="Путь к директории для сохранения профиля")
+
+    # Команда chat
+    parser_chat = subparsers.add_parser("chat", help="Запустить интерактивный чат с нейросетью")
+    parser_chat.add_argument("model", choices=["gemini", "deepseek"], help="Модель для использования")
+    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="Пути к файлам для прикрепления к первому промпту")
+
+    args = parser.parse_args()
+
+    if args.command == "setup":
+        cmd_setup(args)
+    elif args.command == "chat":
+        cmd_chat(args)
+
+if __name__ == "__main__":
+    main()

browser_ai/deepseek_api.py

@@ -0,0 +1,141 @@
+from .base_api import BaseWebAPI
+import time
+import re
+
+def _clean_text(text: str) -> str:
+    return re.sub(r'\n{3,}', '\n\n', text).strip()
+
+class DeepSeekAPI(BaseWebAPI):
+    def __init__(self, profile_dir: str, headless: bool = True, setup_mode: bool = False):
+        super().__init__(
+            profile_dir=profile_dir,
+            url="https://chat.deepseek.com/",
+            headless=headless,
+            disable_images=False,
+            setup_mode=setup_mode
+        )
+
+    def wait_for_ready(self):
+        self._handle_cloudflare()
+        self.page.wait_for_selector("textarea, #chat-input", timeout=30000)
+
+    def _handle_cloudflare(self):
+        time.sleep(3)
+        try:
+            for frame in self.page.frames:
+                if "cloudflare" in frame.url or "turnstile" in frame.url:
+                    print("Обнаружен Cloudflare, пытаюсь пройти проверку...")
+                    box = frame.locator("input[type='checkbox'], .cb-c, body")
+                    if box.count() > 0:
+                        box.first.click()
+                        time.sleep(5)
+        except Exception as e:
+            print(f"Ошибка при прохождении Cloudflare: {e}")
+
+    def ask(self, prompt: str, files: list = None) -> str:
+        self.page.evaluate("navigator.clipboard.writeText('').catch(() => {})")
+
+        if files:
+            try:
+                file_input = self.page.locator('input[type="file"]')
+                file_input.set_input_files(files)
+                time.sleep(1.5)
+                loading_texts =["обработка", "Uploading", "Parsing", "processing", "В ожидании", "Загрузка"]
+                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(2)
+            except Exception as e:
+                print(f"Не удалось прикрепить файлы: {e}")
+
+        md_selector = ".ds-markdown, .markdown-body, div[class*='markdown'], div[class*='prose']"
+        
+        try:
+            initial_md_count = self.page.locator(md_selector).count()
+            initial_texts = self.page.locator(md_selector).all_inner_texts()
+            initial_last_text = initial_texts[-1].strip() if initial_texts else ""
+        except:
+            initial_md_count = 0
+            initial_last_text = ""
+
+        chat_input = self.page.locator("textarea, #chat-input").first
+        chat_input.fill(prompt)
+        time.sleep(0.5)
+        chat_input.press("Enter")
+
+        last_text = ""
+        stable_count = 0
+        max_wait_sec = 120
+
+        js_find_and_click = """
+        () => {
+            const mds = document.querySelectorAll('.ds-markdown, .markdown-body, div[class*="markdown"], div[class*="prose"]');
+            if (mds.length === 0) return false;
+            const lastMd = mds[mds.length - 1];
+            let curr = lastMd;
+            for (let i = 0; i < 5; i++) {
+                if (!curr) break;
+                let sibling = curr.nextElementSibling;
+                while (sibling) {
+                    const isInputBlock = (sibling.matches && sibling.matches('textarea, #chat-input, form, [contenteditable="true"]')) || 
+                                         (sibling.querySelector && sibling.querySelector('textarea, #chat-input, form, [contenteditable="true"]'));
+                    if (isInputBlock) {
+                        sibling = sibling.nextElementSibling;
+                        continue;
+                    }
+                    const btns = sibling.querySelectorAll('div.ds-icon-button[role="button"]');
+                    if (btns.length >= 2) {
+                        btns[0].click();
+                        return true;
+                    }
+                    sibling = sibling.nextElementSibling;
+                }
+                curr = curr.parentElement;
+            }
+            return false;
+        }
+        """
+
+        for _ in range(max_wait_sec):
+            time.sleep(1)
+            try:
+                current_md_count = self.page.locator(md_selector).count()
+                texts = self.page.locator(md_selector).all_inner_texts()
+                texts =[t.strip() for t in texts if t.strip()]
+                current_text = texts[-1] if texts else ""
+            except:
+                current_md_count = 0
+                current_text = ""
+
+            has_new_message = False
+            if current_md_count > initial_md_count:
+                has_new_message = True
+            elif current_text and current_text != initial_last_text:
+                has_new_message = True
+
+            if has_new_message:
+                clicked = self.page.evaluate(js_find_and_click)
+                if clicked:
+                    time.sleep(0.5)
+                    clip_val = self.page.evaluate("navigator.clipboard.readText().catch(() => '')")
+                    if clip_val:
+                        return clip_val
+                
+                if current_text and current_text == last_text:
+                    stable_count += 1
+                else:
+                    stable_count = 0
+
+                last_text = current_text
+
+                if stable_count >= 15:
+                    break
+
+        if last_text:
+            return _clean_text(last_text)
+        else:
+            return "Не удалось найти ответ в DOM. Возможно, изменилась верстка."

browser_ai/exceptions.py

@@ -0,0 +1,3 @@
+class ProfileNotFoundError(Exception):
+    """Возникает, когда указанная директория профиля не существует или пуста."""
+    pass

browser_ai/gemini_api.py

@@ -0,0 +1,116 @@
+from .base_api import BaseWebAPI
+import time
+import re
+
+def _clean_text(text: str) -> str:
+    return re.sub(r'\n{3,}', '\n\n', text).strip()
+
+class GeminiAPI(BaseWebAPI):
+    def __init__(self, profile_dir: str, headless: bool = True, setup_mode: bool = False):
+        super().__init__(
+            profile_dir=profile_dir,
+            url="https://aistudio.google.com/app/prompts/new_chat",
+            headless=headless,
+            disable_images=True,
+            setup_mode=setup_mode
+        )
+
+    def wait_for_ready(self):
+        self.page.wait_for_selector("textarea, [contenteditable='true']", timeout=15000)
+
+    def ask(self, prompt: str, files: list = None) -> str:
+        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()
+        except:
+            initial_thumb_count = 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")
+
+        max_wait_sec = 120
+        start_time = time.time()
+        completed = False
+
+        while time.time() - start_time < max_wait_sec:
+            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 "Не удалось получить текст ответа."

majorchik_api-0.0.0a0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: majorchik-api
+Version: 0.0.0a0
+Summary: Пакет для взаимодействия с современными нейросетями (Gemini, DeepSeek) через автоматизацию браузера
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: playwright>=1.30.0
+
+# Browser AI API
+
+Инструмент для работы с Gemini и DeepSeek через браузерную автоматизацию (Playwright).
+Поддерживает как консольное использование, так и работу в качестве Python-библиотеки.

majorchik_api-0.0.0a0.dist-info/RECORD

@@ -0,0 +1,11 @@
+browser_ai/__init__.py,sha256=Ijc8PmoI0UVuFz2rl4Z-ktylWWEgyxgqP3FNEnbetRM,185
+browser_ai/base_api.py,sha256=9VGVdCTAARePLUnHLkb8956TE8qrM8o9A3_RLlNebJs,2435
+browser_ai/cli.py,sha256=qQp-d40JazvOEfE0uRwgYniBm-m5QlRlYSBYiqT6iGQ,4462
+browser_ai/deepseek_api.py,sha256=DjPq8xu2qeNCJ40J1nAJo69z_YRP6EyRKlHWX8eV5TI,5806
+browser_ai/exceptions.py,sha256=7mgtuXSxLKwOsVtMLT_s4JinuDTN4VpDxCKoUE5VPug,192
+browser_ai/gemini_api.py,sha256=_8y1ygKr8dLwP8lU8ENvhB3hsCgJ1V8ia2M8g--cv3A,5008
+majorchik_api-0.0.0a0.dist-info/METADATA,sha256=E3EFr-GjvVRdrPo7tAC8DzpQ6PZCremABBLcUd0emhs,657
+majorchik_api-0.0.0a0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+majorchik_api-0.0.0a0.dist-info/entry_points.txt,sha256=72XHK1czTPYqgdlXssWXnAuIfzMy02EorsTx7mgNJJY,51
+majorchik_api-0.0.0a0.dist-info/top_level.txt,sha256=7OVRfN-g4pELdYb-THSCJyaDRJDatlImuU2Ud0HoceE,11
+majorchik_api-0.0.0a0.dist-info/RECORD,,

majorchik_api-0.0.0a0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

majorchik_api-0.0.0a0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+browser-ai = browser_ai.cli:main

majorchik_api-0.0.0a0.dist-info/top_level.txt

@@ -0,0 +1 @@
+browser_ai