feat: added aistudio support

Author: STAR-173

.gitignore

@@ -0,0 +1,49 @@
+# --- General ---
+.DS_Store
+.vscode/
+.idea/
+*.log
+*.png
+
+# --- Python ---
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+.env
+venv/
+env/
+ENV/
+.venv/
+
+# --- Verification ---
+verify_aistudio.py
+verify_flow.py
+
+# --- Playwright ---
+test-results/
+playwright-report/
+blob-report/
+playwright/.cache/
+
+# --- OS Generated ---
+Thumbs.db
+ehthumbs.db
+Desktop.ini

CONTRIBUTING.md

@@ -0,0 +1,92 @@
+# Contributing to LLMSession
+
+Thank you for your interest in contributing to **LLMSession**! This project is a monorepo containing both the Python and Node.js implementations. We welcome contributions to either (or both) parts of the library.
+
+## 1. Project Structure
+
+*   **`python/`**: Contains the Python source code and packaging config.
+*   **`node/`**: Contains the TypeScript source code and NPM config.
+
+Please ensure you are working in the correct subdirectory for your changes.
+
+## 2. Python Development
+
+### Setup
+1.  Navigate to the python directory: `cd python`
+2.  Create a virtual environment (optional but recommended):
+    ```bash
+    python -m venv venv
+    source venv/bin/activate  # Linux/Mac
+    # or
+    .\venv\Scripts\activate   # Windows
+    ```
+3.  Install the package in **Editable Mode**:
+    ```bash
+    pip install -e .
+    ```
+    *This allows you to modify code in `src/` and test it immediately without reinstalling.*
+
+### Testing
+We use a manual verification script to test the full flow (Browser launch -> Auth -> Prompt).
+```bash
+python verify_flow.py
+```
+*Tip: You can modify `verify_flow.py` to use `headless=False` to watch the bot work.*
+
+## 3. Node.js Development
+
+### Setup
+1.  Navigate to the node directory: `cd node`
+2.  Install dependencies:
+    ```bash
+    npm install
+    ```
+
+### Building
+Since this is a TypeScript project, you must compile the code.
+*   **One-off build**:
+    ```bash
+    npm run build
+    ```
+*   **Watch mode (Recommended for Dev)**:
+    ```bash
+    npx tsc -w
+    ```
+    *Keep this running in a separate terminal to auto-compile changes.*
+
+### Testing
+To run the verification flow using `ts-node` (skips build step) or the compiled dist:
+```bash
+# Run directly from TypeScript source
+npx ts-node verify_flow.ts
+```
+
+## 4. Safety Warning (Credentials)
+
+To test changes, you will likely need to put real credentials into `verify_flow.py` or `verify_flow.ts`.
+
+> [!CAUTION]
+> **NEVER commit files containing your real passwords.**
+> If you modify the verification scripts with real credentials, please revert those changes before pushing, or ensure they are ignored.
+
+## 5. Pull Request Process
+
+1.  **Fork & Clone**: Fork the repo and clone it locally.
+2.  **Branch**: Create a feature branch (`git checkout -b feature/amazing-feature`).
+3.  **Code**: Implement your changes.
+    *   **Python**: Adhere to PEP 8.
+    *   **Node**: Adhere to standard ESLint/Prettier conventions.
+4.  **Verify**: Run the verification scripts in both languages if your change affects shared logic (like selectors).
+5.  **Commit**: Use descriptive commit messages.
+6.  **Push & PR**: Push to your fork and open a Pull Request against `main`.
+
+## 6. Adding New Selectors
+
+If you are fixing a broken selector because ChatGPT updated their UI:
+1.  Update `python/src/llm_session/providers/chatgpt.py`
+2.  Update `node/src/providers/ChatGPT.ts`
+3.  **Please update both** if possible to keep the libraries in sync.
+
+## License
+
+By contributing, you agree that your contributions will be licensed under the MIT License.

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025
+
+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.

README.md

@@ -0,0 +1,166 @@
+# LLMSession
+
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![PyPI version](https://badge.fury.io/py/llm-session.svg)](https://pypi.org/project/llm-session/)
+[![npm version](https://badge.fury.io/js/llm-session.svg)](https://www.npmjs.com/package/llm-session)
+
+A zero-configuration tool to automate interactions with web-based LLM providers (currently ChatGPT). It handles authentication, session persistence, and chained prompt execution programmatically.
+
+## Features
+- **Zero-Config Setup**: Automatically handles browser binaries via Playwright.
+- **Session Persistence**: Reuses cookies/storage for subsequent runs (no need to login every time).
+- **Cross-Language Sharing**: Sessions are shared between Python and Node.js environments automatically.
+- **Smart OTP Handling**: Supports non-blocking callbacks for 2FA/OTP challenges.
+- **Resilient**: Allows custom CSS selectors to adapt to UI changes without updating the library.
+
+## Prerequisites
+You must set the following environment variables, or pass them directly to the constructor:
+- `CHATGPT_EMAIL`
+- `CHATGPT_PASSWORD`
+- `CHATGPT_GOOGLE_LOGIN` (Optional: set to "true" if using Google Auth. *Note: Google Auth is experimental and may require manual intervention.*)
+
+## Disclaimer
+> [!WARNING]
+> **Cloudflare/Bot Detection**: Automated interactions with ChatGPT are subject to high-security bot detection (Cloudflare). This library uses standard browser automation and may be blocked. For production reliability, please use the Official OpenAI API.
+
+This tool automates a third-party web interface. It is subject to breakage if the target website changes its DOM structure. Use responsibly and in accordance with the provider's Terms of Service.
+
+---
+
+## Python Usage
+
+### Installation
+```bash
+pip install llm-session
+```
+
+### Quick Start
+```python
+import logging
+from llm_session import Automator
+
+# 1. Configure Standard Logging
+logging.basicConfig(level=logging.INFO)
+
+# 2. Define OTP Callback (Optional, but recommended for headless envs)
+def my_otp_handler():
+    # In production, you might fetch this from an email API
+    return input("Enter OTP Code sent to email: ")
+
+# 3. Initialize
+bot = Automator(
+    provider="chatgpt",
+    headless=False,  # Set to True for production (CURRENTLY ONLY WORKS WITH headless=False)
+    credentials={
+        "email": "your_email@example.com", 
+        "password": "your_password",
+        "method": "email" # or "google"
+    },
+    on_otp_required=my_otp_handler
+)
+
+# 4. Single Prompt
+print(bot.process_prompt("Hello, world!"))
+
+# 5. Chained Prompt (Inject previous response)
+chain = [
+    "Write a haiku about Python.",
+    "Translate this haiku to Spanish: {{previous}}"
+]
+responses = bot.process_chain(chain)
+print(responses)
+
+bot.close()
+```
+
+---
+
+## Node.js Usage
+
+### Installation
+```bash
+npm install llm-session
+```
+
+### Quick Start
+```typescript
+import { Automator } from 'llm-session';
+
+async function main() {
+    // 1. Define OTP Callback
+    const onOtpRequired = async () => {
+        console.log("Please check your email for code.");
+        // Return code from your logic here
+        return "123456"; 
+    };
+
+    // 2. Initialize
+    const bot = new Automator(
+        "chatgpt", 
+        false, // headless
+        {
+            email: "your_email@example.com",
+            password: "your_password"
+        },
+        undefined, // sessionPath (optional)
+        {
+            onOtpRequired: onOtpRequired,
+            // Optional: Inject your own logger (e.g., Winston, Pino)
+            // logger: myLoggerInstance 
+        }
+    );
+    
+    try {
+        await bot.init();
+
+        const response = await bot.processPrompt("Hello from Node.js!");
+        console.log(response);
+
+    } finally {
+        await bot.close();
+    }
+}
+main();
+```
+
+---
+
+## Advanced Configuration (Resilience)
+
+Websites change their layout often. If ChatGPT updates their CSS class names, you don't need to wait for a package update. You can inject your own selectors during initialization.
+
+**Python:**
+```python
+bot = Automator(
+    provider="chatgpt", 
+    config={
+        "selectors": {
+            "textarea": "#new-prompt-id",
+            "send_btn": ".new-send-button-class",
+            "assistant_msg": ".new-message-wrapper"
+        }
+    }
+)
+```
+
+**Node.js:**
+```typescript
+const bot = new Automator("chatgpt", true, undefined, undefined, {
+    selectors: {
+        textarea: "#new-prompt-id",
+        send_btn: ".new-send-button-class"
+    }
+});
+```
+
+## Session Management
+This library stores browser cookies and local storage in your OS's standard user data directory (e.g., `%LOCALAPPDATA%/LLMSession` on Windows, `~/.local/share/LLMSession` on Linux).
+
+*   **Cross-Language:** If you login using the Python script, the Node.js script will automatically detect the existing session and skip login (and vice-versa).
+*   **Persistence:** Sessions persist across reboots.
+
+## Contributing
+We welcome contributions! Please see [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines on how to build the project locally.
+
+## License
+This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.

pyproject.toml

@@ -0,0 +1,28 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "llm_session"
+version = "0.1.3"
+description = "A zero-configuration tool to automate web-based LLM interactions."
+readme = "README.md"
+authors = [{ name = "LLM_Session", email = "shoaib.star.a@gmail.com" }]
+license = { text = "MIT" }
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+dependencies = [
+    "playwright>=1.30.0",
+    "appdirs>=1.4.4",
+]
+requires-python = ">=3.8"
+
+[project.urls]
+"Homepage" = "https://github.com/star-173/llm_session"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+exclude = ["tests*", "node_modules*", "dist*", "build*"]

src/llm_session/__init__.py

@@ -0,0 +1,3 @@
+from .automator import Automator
+
+__all__ = ["Automator"]