antigravity-remote 3.1.0__py3-none-any.whl

antigravity_remote/utils/ocr.py

@@ -0,0 +1,98 @@
+"""OCR utilities for Antigravity Remote."""
+
+import logging
+from typing import Optional, Tuple
+
+import mss
+from PIL import Image
+
+logger = logging.getLogger(__name__)
+
+# Lazy import pytesseract
+_pytesseract = None
+
+
+def _get_pytesseract():
+    """Lazy load pytesseract with path configuration."""
+    global _pytesseract
+    if _pytesseract is None:
+        try:
+            import pytesseract
+            from ..config import config
+            
+            if config.tesseract_path:
+                pytesseract.pytesseract.tesseract_cmd = config.tesseract_path
+            
+            _pytesseract = pytesseract
+        except ImportError:
+            logger.error("pytesseract not installed. Run: pip install pytesseract")
+            raise
+    return _pytesseract
+
+
+def scan_screen() -> Tuple[str, int]:
+    """
+    Capture screenshot and extract text using OCR.
+    
+    Returns:
+        Tuple of (extracted text, image hash for change detection).
+    """
+    pytesseract = _get_pytesseract()
+    
+    with mss.mss() as sct:
+        monitor = sct.monitors[1]
+        sct_img = sct.grab(monitor)
+        img = Image.frombytes('RGB', sct_img.size, sct_img.bgra, 'raw', 'BGRX')
+        
+        # Extract text
+        text = pytesseract.image_to_string(img).lower()
+        
+        # Simple hash for change detection (first 10KB of image data)
+        img_hash = hash(img.tobytes()[:10000])
+        
+        return text, img_hash
+
+
+# Keyword lists for detection
+APPROVAL_KEYWORDS = [
+    "run command", "accept changes", "proceed", "approve",
+    "allow", "confirm", "yes or no", "y/n", "always allow",
+    "do you want", "permission", "authorize"
+]
+
+DONE_KEYWORDS = [
+    "anything else", "let me know", "task complete", "done!",
+    "successfully", "finished", "completed", "all set",
+    "ready for", "is there anything"
+]
+
+ERROR_KEYWORDS = [
+    "error:", "failed", "exception", "traceback", "cannot",
+    "permission denied", "not found", "invalid", "quota exceeded"
+]
+
+
+def detect_keywords(text: str) -> Optional[Tuple[str, str]]:
+    """
+    Detect important keywords in screen text.
+    
+    Args:
+        text: The OCR-extracted text to search.
+        
+    Returns:
+        Tuple of (category, keyword) if detected, None otherwise.
+        Categories: 'approval', 'done', 'error'
+    """
+    for keyword in APPROVAL_KEYWORDS:
+        if keyword in text:
+            return ('approval', keyword)
+    
+    for keyword in DONE_KEYWORDS:
+        if keyword in text:
+            return ('done', keyword)
+    
+    for keyword in ERROR_KEYWORDS:
+        if keyword in text:
+            return ('error', keyword)
+    
+    return None

antigravity_remote/utils/screenshot.py

@@ -0,0 +1,49 @@
+"""Screenshot utilities for Antigravity Remote."""
+
+import logging
+import os
+import tempfile
+from typing import Optional
+
+import mss
+import mss.tools
+
+logger = logging.getLogger(__name__)
+
+
+def take_screenshot() -> Optional[str]:
+    """
+    Capture a screenshot of the primary monitor.
+    
+    Returns:
+        Path to the temporary screenshot file, or None on failure.
+    """
+    try:
+        with mss.mss() as sct:
+            monitor = sct.monitors[1]  # Primary monitor
+            sct_img = sct.grab(monitor)
+            
+            # Save to temp file
+            with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
+                mss.tools.to_png(sct_img.rgb, sct_img.size, output=tmp.name)
+                logger.debug(f"Screenshot saved to {tmp.name}")
+                return tmp.name
+                
+    except Exception as e:
+        logger.error(f"Error taking screenshot: {e}")
+        return None
+
+
+def cleanup_screenshot(path: str) -> None:
+    """
+    Remove a temporary screenshot file.
+    
+    Args:
+        path: Path to the screenshot file.
+    """
+    try:
+        if path and os.path.exists(path):
+            os.remove(path)
+            logger.debug(f"Cleaned up screenshot: {path}")
+    except Exception as e:
+        logger.warning(f"Error cleaning up screenshot: {e}")

antigravity_remote-3.1.0.dist-info/METADATA

@@ -0,0 +1,114 @@
+Metadata-Version: 2.4
+Name: antigravity-remote
+Version: 3.1.0
+Summary: Remote control your Antigravity AI assistant via Telegram
+Project-URL: Homepage, https://github.com/kubrat/antigravity-remote
+Project-URL: Repository, https://github.com/kubrat/antigravity-remote
+Author: Kubrat
+License-Expression: MIT
+License-File: LICENSE
+Keywords: ai,antigravity,automation,bot,remote,telegram
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Communications :: Chat
+Requires-Python: >=3.10
+Requires-Dist: fastapi>=0.95.0
+Requires-Dist: mss>=9.0.0
+Requires-Dist: pillow>=10.0.0
+Requires-Dist: pyautogui>=0.9.53
+Requires-Dist: pydub>=0.25.1
+Requires-Dist: pygetwindow>=0.0.9
+Requires-Dist: pyperclip>=1.8.2
+Requires-Dist: python-telegram-bot>=20.0
+Requires-Dist: speechrecognition>=3.10.0
+Requires-Dist: uvicorn>=0.21.0
+Requires-Dist: websockets>=10.0
+Provides-Extra: dev
+Requires-Dist: black>=23.0.0; extra == 'dev'
+Requires-Dist: mypy>=1.0.0; extra == 'dev'
+Requires-Dist: pytest>=7.0.0; extra == 'dev'
+Provides-Extra: voice
+Requires-Dist: speechrecognition>=3.10.0; extra == 'voice'
+Description-Content-Type: text/markdown
+
+# Antigravity Remote 🚀
+
+Remote control your [Antigravity](https://antigravity.dev) AI assistant via Telegram.
+
+[![Telegram Bot](https://img.shields.io/badge/Telegram-@antigravityrcbot-blue?logo=telegram)](https://t.me/antigravityrcbot)
+[![PyPI](https://img.shields.io/pypi/v/antigravity-remote)](https://pypi.org/project/antigravity-remote/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+## Features
+
+- 📱 **Message Relay** - Send instructions from your phone
+- 📸 **Screenshots** - View screen anytime
+- ⚡ **Quick Replies** - One-tap Yes/No/Proceed buttons
+- 🔐 **Secure** - Your data never leaves your PC
+
+## Quick Start
+
+### 1. Install
+
+```bash
+pip install antigravity-remote
+```
+
+### 2. Register
+
+```bash
+antigravity-remote --register
+```
+
+Enter your Telegram User ID (get it from [@userinfobot](https://t.me/userinfobot)).
+
+### 3. Run
+
+```bash
+antigravity-remote
+```
+
+### 4. Control from Telegram
+
+👉 **[@antigravityrcbot](https://t.me/antigravityrcbot)**
+
+## Requirements
+
+- Python 3.10+
+- Windows
+
+## Commands (in Telegram)
+
+| Command | Description |
+|---------|-------------|
+| `/status` | Take screenshot |
+| `/scroll up/down` | Scroll chat |
+| `/accept` / `/reject` | Accept/reject |
+| `/key ctrl+s` | Send key combo |
+| `/quick` | Quick reply buttons |
+| Any text | Relay to Antigravity |
+
+## How It Works
+
+```
+📱 Your Phone        ☁️ Server          💻 Your PC
+      │ Message bot       │                   │
+      ├──────────────────►│                   │
+      │                   │ WebSocket         │
+      │                   ├──────────────────►│
+      │                   │                   │ Execute
+      │                   │◄──────────────────┤ Screenshot
+      │◄──────────────────┤                   │
+```
+
+Your bot token stays on our secure server. You only run a lightweight agent on your PC.
+
+## License
+
+MIT © Kubrat

antigravity_remote-3.1.0.dist-info/RECORD

@@ -0,0 +1,23 @@
+antigravity_remote/__init__.py,sha256=ZtnsWjQxZfmE2U8z3mbjWa95JtUoliaM00auAv2SGMA,293
+antigravity_remote/__main__.py,sha256=RwTPfkqVbqW6Sb_3T_BIe09F3heKx6W0KWKSXxBYTNQ,3596
+antigravity_remote/agent.py,sha256=WRggaT8gg9a_cD1htS0d9l-iZ3bJBpP_XgIktoGCMG0,15157
+antigravity_remote/bot.py,sha256=v5RP757-5KMTzxyxJpfKz9tfubyXZ7dt0QcmBk_hjIw,5059
+antigravity_remote/config.py,sha256=MVyJ49AEXR6-WynWaYXuPW3FwdU6RLUKstE7u8HhH28,2236
+antigravity_remote/state.py,sha256=6LoHeTfJQVHU9PpiC5Hhi6V4CKWJx-qTOnEreQLfG6Y,2217
+antigravity_remote/handlers/__init__.py,sha256=_YKjoXLlv0hYHkZNwDMsa4Ze3UjgXChDCaMR_aw3-WE,1513
+antigravity_remote/handlers/ai.py,sha256=_YUjUNsqJ0GxAbxLE1Yy1Hw38K2C5RSQ6VEeOIbm4ck,4501
+antigravity_remote/handlers/base.py,sha256=Yjq7QN15zBExMrd8jGSmbYKJS3_dpP6Ovq0zMLva2rc,1062
+antigravity_remote/handlers/control.py,sha256=N7d84jUVnJLI4fhfKKsOllVIkqotDF3pbhMDBdmwiJU,3660
+antigravity_remote/handlers/files.py,sha256=TS9JB5CFeiIGkpnNY7Puk-hkQugVgn0g9Exa2gisEQM,3724
+antigravity_remote/handlers/monitoring.py,sha256=zo5CbpBya-YHBE4CYaGBf9xvbXW2bKujHiiYK6gIpFc,7386
+antigravity_remote/handlers/quick.py,sha256=Vd2TjFXLzwMqYIqq7Fb6BJDnjllYZVQ1_7QGWXZOsnE,4881
+antigravity_remote/handlers/screen.py,sha256=0funSy1eYPIfeeGymgSAIgFpaUu3HhLv2h7G82ec3UY,3337
+antigravity_remote/utils/__init__.py,sha256=lwRicer6FCk4Op7zPee3Ezdra104PxDigdMpDvPYt_o,465
+antigravity_remote/utils/automation.py,sha256=lR4uSXkbQSB7EZ2w2iZW9UeSRSQz8xvR3eJWmf04RA8,3992
+antigravity_remote/utils/ocr.py,sha256=STdJoh3_zsKvHx0dCLeqEcUM4-F4b7nQ2A57fzUK2Hs,2738
+antigravity_remote/utils/screenshot.py,sha256=eup60Iles_MMNBbcmLWFcvi8DAddUzrYmbPwRuxDg-8,1369
+antigravity_remote-3.1.0.dist-info/METADATA,sha256=84-27D5NhS8f6c78LF8isEp_Y9dPQNWkXUPCsS6Swzw,3592
+antigravity_remote-3.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+antigravity_remote-3.1.0.dist-info/entry_points.txt,sha256=DlNcH8xFN_aK4PFd-grvOLSLiYdh4YaaPj0EsNnoI9I,72
+antigravity_remote-3.1.0.dist-info/licenses/LICENSE,sha256=zwQvwBcTdGhYEnFEzOLpV3YSwsEu27lLR3GYhmQ-Tvg,1084
+antigravity_remote-3.1.0.dist-info/RECORD,,

antigravity_remote-3.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

antigravity_remote-3.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+antigravity-remote = antigravity_remote.__main__:main

antigravity_remote-3.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Kubrat
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.