dsora-gen 0.1.0__tar.gz

dsora_gen-0.1.0/.claude/settings.json

@@ -0,0 +1,8 @@
+{
+  "permissions": {
+    "allow": [
+      "Bash(chmod +x /Users/macbook/Documents/Personal/auto-gen-video-sora/sora_video_tool/run.sh)",
+      "Bash(chmod +x /Users/macbook/Documents/Personal/auto-gen-video-sora/run.sh)"
+    ]
+  }
+}

dsora_gen-0.1.0/.gitignore

@@ -0,0 +1,30 @@
+# Virtual environment
+.venv/
+venv/
+
+# Chrome profile data (session/cookies — KHONG commit len git)
+sora_video_tool/profiles/
+
+# App data (settings, run log, profiles cua user sau khi cai)
+~/.sora-tool/
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+*.pdb
+.pytest_cache/
+
+# Build artifacts
+dist/
+build/
+*.egg-info/
+
+# macOS
+.DS_Store
+.AppleDouble
+
+# Temp files
+*.tmp
+*.log

dsora_gen-0.1.0/PKG-INFO

@@ -0,0 +1,116 @@
+Metadata-Version: 2.4
+Name: dsora-gen
+Version: 0.1.0
+Summary: Tu dong tao video Sora theo batch voi nhieu Chrome profiles
+License: MIT
+Keywords: automation,playwright,sora,video
+Requires-Python: >=3.11
+Requires-Dist: click>=8.0.0
+Requires-Dist: playwright>=1.40.0
+Description-Content-Type: text/markdown
+
+# Sora Video Tool
+
+Tu dong tao video Sora theo batch voi nhieu Chrome profiles.
+
+## Cai dat
+
+```bash
+# Clone repo va vao thu muc
+cd auto-gen-video-sora
+
+# Cach 1: dung run.sh (tu dong cai dat moi thu)
+chmod +x run.sh
+./run.sh setup
+
+# Cach 2: thu cong
+python3 -m venv .venv && source .venv/bin/activate
+pip install -e .
+playwright install chromium   # chi can cho Playwright, khong phai cho Chrome thuc
+```
+
+## Su dung
+
+### Thiet lap lan dau
+
+```bash
+./run.sh setup
+# hoac
+sora setup
+```
+
+Flow:
+1. Nhap so luong profile
+2. Nhap ten alias cho tung profile
+3. Nhap username dsora
+4. Nhap backend key
+5. Nhap Telegram bot token / channel ID (tuy chon)
+6. Chrome mo ra -> dang nhap Google + Sora + cau hinh Settings
+
+### Dang nhap lai mot profile cu
+
+```bash
+sora setup --retry
+```
+
+### Chay tu dong
+
+```bash
+./run.sh run
+# hoac
+sora run
+```
+
+Tool se:
+- Lay danh sach san pham tu API
+- Chay lan luot tung profile (5 san pham / profile)
+- Gui thong bao Telegram sau moi san pham / profile
+- Luu lich su chay vao `~/.sora-tool/run_log.json`
+
+### Xem huong dan
+
+```bash
+sora help
+sora --help
+sora run --help
+```
+
+## Cau truc thu muc
+
+```
+auto-gen-video-sora/
+├── pyproject.toml              # Cau hinh PyPI
+├── run.sh                      # Script khoi dong nhanh
+├── README.md
+├── .gitignore
+└── sora_video_tool/            # Python package
+    ├── cli.py                  # Entry point (sora setup / run / help)
+    ├── constants.py            # Hang so toan cuc
+    ├── config.py               # Doc/ghi ~/.sora-tool/settings.json
+    ├── profile_manager.py      # Quan ly thu muc Chrome profile
+    ├── browser_launcher.py     # Mo Chrome (login + CDP automation)
+    ├── sora_bot.py             # Logic tu dong hoa Sora
+    ├── api_client.py           # Giao tiep voi backend API
+    ├── telegram_notifier.py    # Gui thong bao Telegram
+    ├── run_tracker.py          # Theo doi lich su chay
+    └── assets/
+        └── tshirt.png          # Anh mac dinh
+```
+
+## Du lieu luu tru
+
+Tat ca du lieu nguoi dung luu tai `~/.sora-tool/`:
+
+| File | Noi dung |
+|---|---|
+| `settings.json` | Username, backend key, telegram, profiles |
+| `run_log.json` | Lich su chay tung profile |
+| `profiles/profile_1/` | Chrome user data (session, cookies) |
+
+## Publish len PyPI
+
+```bash
+pip install build twine
+python -m build
+twine upload dist/*
+```

dsora_gen-0.1.0/README.md

@@ -0,0 +1,105 @@
+# Sora Video Tool
+
+Tu dong tao video Sora theo batch voi nhieu Chrome profiles.
+
+## Cai dat
+
+```bash
+# Clone repo va vao thu muc
+cd auto-gen-video-sora
+
+# Cach 1: dung run.sh (tu dong cai dat moi thu)
+chmod +x run.sh
+./run.sh setup
+
+# Cach 2: thu cong
+python3 -m venv .venv && source .venv/bin/activate
+pip install -e .
+playwright install chromium   # chi can cho Playwright, khong phai cho Chrome thuc
+```
+
+## Su dung
+
+### Thiet lap lan dau
+
+```bash
+./run.sh setup
+# hoac
+sora setup
+```
+
+Flow:
+1. Nhap so luong profile
+2. Nhap ten alias cho tung profile
+3. Nhap username dsora
+4. Nhap backend key
+5. Nhap Telegram bot token / channel ID (tuy chon)
+6. Chrome mo ra -> dang nhap Google + Sora + cau hinh Settings
+
+### Dang nhap lai mot profile cu
+
+```bash
+sora setup --retry
+```
+
+### Chay tu dong
+
+```bash
+./run.sh run
+# hoac
+sora run
+```
+
+Tool se:
+- Lay danh sach san pham tu API
+- Chay lan luot tung profile (5 san pham / profile)
+- Gui thong bao Telegram sau moi san pham / profile
+- Luu lich su chay vao `~/.sora-tool/run_log.json`
+
+### Xem huong dan
+
+```bash
+sora help
+sora --help
+sora run --help
+```
+
+## Cau truc thu muc
+
+```
+auto-gen-video-sora/
+├── pyproject.toml              # Cau hinh PyPI
+├── run.sh                      # Script khoi dong nhanh
+├── README.md
+├── .gitignore
+└── sora_video_tool/            # Python package
+    ├── cli.py                  # Entry point (sora setup / run / help)
+    ├── constants.py            # Hang so toan cuc
+    ├── config.py               # Doc/ghi ~/.sora-tool/settings.json
+    ├── profile_manager.py      # Quan ly thu muc Chrome profile
+    ├── browser_launcher.py     # Mo Chrome (login + CDP automation)
+    ├── sora_bot.py             # Logic tu dong hoa Sora
+    ├── api_client.py           # Giao tiep voi backend API
+    ├── telegram_notifier.py    # Gui thong bao Telegram
+    ├── run_tracker.py          # Theo doi lich su chay
+    └── assets/
+        └── tshirt.png          # Anh mac dinh
+```
+
+## Du lieu luu tru
+
+Tat ca du lieu nguoi dung luu tai `~/.sora-tool/`:
+
+| File | Noi dung |
+|---|---|
+| `settings.json` | Username, backend key, telegram, profiles |
+| `run_log.json` | Lich su chay tung profile |
+| `profiles/profile_1/` | Chrome user data (session, cookies) |
+
+## Publish len PyPI
+
+```bash
+pip install build twine
+python -m build
+twine upload dist/*
+```

dsora_gen-0.1.0/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "dsora-gen"
+version = "0.1.0"
+description = "Tu dong tao video Sora theo batch voi nhieu Chrome profiles"
+readme = "README.md"
+license = { text = "MIT" }
+requires-python = ">=3.11"
+keywords = ["sora", "automation", "video", "playwright"]
+dependencies = [
+    "playwright>=1.40.0",
+    "click>=8.0.0",
+]
+
+[project.scripts]
+dsg = "sora_video_tool.cli:main"
+
+[tool.hatch.build.targets.wheel]
+packages = ["sora_video_tool"]

dsora_gen-0.1.0/run.sh

@@ -0,0 +1,32 @@
+#!/usr/bin/env zsh
+# ----------------------------------------------------------------------------
+# run.sh — Khoi dong Sora Video Tool
+# Su dung: ./run.sh setup | ./run.sh run | ./run.sh help
+# ----------------------------------------------------------------------------
+set -e
+
+ROOT="$(cd "$(dirname "$0")" && pwd)"
+VENV="$ROOT/.venv"
+PYTHON="$VENV/bin/python"
+PIP="$VENV/bin/pip"
+
+# 1. Tao virtualenv neu chua co
+if [ ! -f "$PYTHON" ]; then
+  echo "[run.sh] Tao virtual environment ..."
+  python3 -m venv "$VENV"
+fi
+
+# 2. Cai dependencies
+echo "[run.sh] Kiem tra dependencies ..."
+"$PIP" install -q --upgrade pip
+"$PIP" install -q -e "$ROOT"
+
+# 3. Cai Playwright browsers neu chua co
+if [ ! -d "$HOME/Library/Caches/ms-playwright" ] && [ ! -d "$HOME/.cache/ms-playwright" ]; then
+  echo "[run.sh] Cai Playwright browsers ..."
+  "$VENV/bin/playwright" install chromium
+fi
+
+# 4. Chay lenh — truyen tat ca tham so
+echo "[run.sh] Chay: dsg $*"
+"$VENV/bin/dsg" "$@"

dsora_gen-0.1.0/sora_video_tool/.gitignore

@@ -0,0 +1,18 @@
+# Virtual environment
+.venv/
+
+# Chrome profile data (contains personal sessions/cookies – never commit)
+profiles/
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+
+# macOS
+.DS_Store
+
+# Temp files
+*.tmp
+*.log

dsora_gen-0.1.0/sora_video_tool/README.md

@@ -0,0 +1,106 @@
+# sora_video_tool
+
+Manage multiple isolated Chrome profiles and automate Sora video generation.
+
+---
+
+## Project structure
+
+```
+sora_video_tool/
+├── main.py            # CLI entry point
+├── config.py          # Global settings (URLs, paths, browser options)
+├── profile_manager.py # Create / list persistent profile directories
+├── browser_launcher.py# Launch Chromium with persistent contexts
+├── sora_bot.py        # Automation logic (stubs – fill in when ready)
+├── data_provider.py   # Mock prompt/image data source
+├── requirements.txt
+└── profiles/          # Auto-created; one sub-folder per profile
+    ├── profile_1/
+    ├── profile_2/
+    └── ...
+```
+
+---
+
+## 1 — Install dependencies
+
+```bash
+# (Recommended) create a virtual environment first
+python -m venv .venv
+source .venv/bin/activate        # macOS / Linux
+# .venv\Scripts\activate         # Windows
+
+# Install Python packages
+pip install -r requirements.txt
+```
+
+## 2 — Install Playwright browsers
+
+```bash
+playwright install chromium
+```
+
+> Only Chromium is needed.  Run `playwright install` if you also want
+> Firefox and WebKit.
+
+---
+
+## 3 — Run the tool
+
+### Create + launch N profiles and navigate each to Sora
+
+```bash
+python main.py --profiles 3
+```
+
+This will:
+1. Create `profiles/profile_1`, `profiles/profile_2`, `profiles/profile_3`
+2. Open three Chromium windows simultaneously, each with its own profile
+3. Navigate each window to `https://sora.com`
+
+Log in to **Google** and **Sora** manually in each window.
+Your session (cookies, tokens) is saved automatically inside the profile folder.
+Close all windows when done — next time you run the tool, you will still be logged in.
+
+---
+
+### Other usage examples
+
+```bash
+# Create profiles without opening a browser
+python main.py --profiles 3 --no-launch
+
+# Open browsers but skip auto-navigation to Sora
+python main.py --profiles 3 --no-sora
+
+# Use the default (1 profile)
+python main.py
+```
+
+---
+
+## 4 — Adding automation later
+
+1. Open `sora_bot.py` and implement the stub functions (`open_sora`,
+   `upload_image`, `submit_prompt`, `create_video`).
+2. Use `data_provider.py` → `get_next_job()` to feed prompts and images.
+3. Call your bot from `main.py` after the browser context is created.
+
+Each profile's `BrowserContext` is already authenticated — no extra login
+logic is required once you have logged in manually.
+
+---
+
+## Configuration
+
+Edit `config.py` to change:
+
+| Setting               | Default         | Description                              |
+|-----------------------|-----------------|------------------------------------------|
+| `PROFILES_DIR`        | `./profiles`    | Where profile folders are stored         |
+| `DEFAULT_PROFILE_COUNT` | `1`           | Profiles used when `--profiles` is omitted |
+| `SORA_URL`            | `https://sora.com` | URL opened on launch                  |
+| `BROWSER_HEADLESS`    | `False`         | Set `True` for headless / CI runs        |
+| `BROWSER_CHANNEL`     | `"chromium"`    | `"chrome"` to use system Chrome          |
+| `BROWSER_SLOW_MO`     | `0`             | ms delay between actions (for debugging) |

dsora_gen-0.1.0/sora_video_tool/__init__.py

@@ -0,0 +1,3 @@
+"""sora_video_tool — Tự động tạo video Sora theo batch với nhiều Chrome profiles."""
+
+__version__ = "0.1.0"

dsora_gen-0.1.0/sora_video_tool/api_client.py

@@ -0,0 +1,49 @@
+"""
+Giao tiep voi backend API.
+
+GET  /products/pending?username=...   — lay danh sach san pham can xu ly
+PATCH /products/:id/done              — danh dau san pham hoan thanh
+"""
+from __future__ import annotations
+
+import json
+import urllib.error
+import urllib.request
+
+from .constants import API_BASE_URL
+
+
+def get_pending_products(username: str) -> list[dict]:
+    """Lay danh sach san pham co status = not_start cua user."""
+    url = f"{API_BASE_URL}/products/pending?username={username}"
+    req = urllib.request.Request(url, headers={"Accept": "application/json"})
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            return json.loads(resp.read().decode())
+    except urllib.error.HTTPError as e:
+        raise RuntimeError(f"Loi API ({e.code}): {e.reason}") from e
+    except Exception as e:
+        raise RuntimeError(f"Khong the ket noi API: {e}") from e
+
+
+def mark_product_done(product_id: int, backend_key: str) -> dict:
+    """Danh dau san pham la done sau khi xu ly xong."""
+    url = f"{API_BASE_URL}/products/{product_id}/done"
+    req = urllib.request.Request(
+        url,
+        data=b"",
+        method="PATCH",
+        headers={
+            "x-backend-key": backend_key,
+            "Content-Length": "0",
+        },
+    )
+    try:
+        with urllib.request.urlopen(req, timeout=15) as resp:
+            return json.loads(resp.read().decode())
+    except urllib.error.HTTPError as e:
+        if e.code == 401:
+            raise RuntimeError("Backend key khong hop le (401 Unauthorized).") from e
+        raise RuntimeError(f"Loi API mark done ({e.code}): {e.reason}") from e
+    except Exception as e:
+        raise RuntimeError(f"Khong the ket noi API: {e}") from e

dsora_gen-0.1.0/sora_video_tool/browser_launcher.py

@@ -0,0 +1,147 @@
+"""
+Quan ly vong doi Chrome.
+
+LOGIN MODE   : Mo Chrome binh thuong bang subprocess (khong co Playwright).
+AUTOMATE MODE: Mo Chrome bang subprocess voi --remote-debugging-port,
+               sau do Playwright ket noi qua CDP.
+=> Khong co banner "controlled by automated software".
+"""
+from __future__ import annotations
+
+import socket
+import subprocess
+import time
+from pathlib import Path
+
+from .constants import (
+    CHROME_BINARY,
+    CDP_PORT_BASE,
+    CHROME_LAUNCH_TIMEOUT_SEC,
+    BROWSER_SLOW_MO,
+    SORA_URL,
+)
+
+
+# ---------------------------------------------------------------------------
+# Internal helpers
+# ---------------------------------------------------------------------------
+
+def _wait_for_cdp(port: int, proc: subprocess.Popen, timeout: float = 30.0) -> None:
+    """Block cho den khi Chrome mo cong CDP, kiem tra process con song."""
+    deadline = time.time() + timeout
+    while time.time() < deadline:
+        # Neu Chrome thoat som thi bao loi ngay
+        ret = proc.poll()
+        if ret is not None:
+            stderr_out = ""
+            if proc.stderr:
+                stderr_out = proc.stderr.read().decode(errors="replace")[:500]
+            raise RuntimeError(
+                f"Chrome thoat ngay lap tuc (exit code {ret}).\n"
+                f"Stderr: {stderr_out or '(trong)'}"
+            )
+        try:
+            with socket.create_connection(("127.0.0.1", port), timeout=1):
+                return
+        except OSError:
+            time.sleep(0.4)
+    # Het timeout — lay stderr de debug
+    stderr_out = ""
+    if proc.stderr:
+        stderr_out = proc.stderr.read().decode(errors="replace")[:500]
+    raise TimeoutError(
+        f"Chrome khong mo cong CDP {port} sau {timeout}s.\n"
+        f"Stderr: {stderr_out or '(trong)'}\n"
+        "Thu tat tat ca Chrome dang mo roi chay lai."
+    )
+
+
+def _spawn_chrome(
+    profile_path: Path,
+    *,
+    debug_port: int | None = None,
+    url: str | None = None,
+) -> subprocess.Popen:
+    cmd = [
+        CHROME_BINARY,
+        f"--user-data-dir={profile_path}",
+        "--no-first-run",
+        "--no-default-browser-check",
+        "--mute-audio",
+        "--disable-features=OptimizationGuideModelDownloading",
+    ]
+    if debug_port:
+        cmd.append(f"--remote-debugging-port={debug_port}")
+    if url:
+        cmd.append(url)
+    print(f"  [chrome] CMD: {' '.join(cmd[:4])} ...")
+    return subprocess.Popen(
+        cmd,
+        stdout=subprocess.DEVNULL,
+        stderr=subprocess.PIPE,  # capture de debug
+    )
+
+
+# ---------------------------------------------------------------------------
+# Login mode
+# ---------------------------------------------------------------------------
+
+def launch_chrome_for_login(profile_path: Path, url: str | None = None) -> subprocess.Popen:
+    """Mo mot cua so Chrome thong thuong de user dang nhap thu cong."""
+    print(f"  [chrome] Mo profile: {profile_path.name} ...")
+    return _spawn_chrome(profile_path, url=url)
+
+
+# ---------------------------------------------------------------------------
+# Automate mode
+# ---------------------------------------------------------------------------
+
+def _remove_singleton_lock(profile_path: Path) -> None:
+    """Xoa file khoa con lai neu Chrome bi tat dot ngot lan truoc."""
+    lock = profile_path / "SingletonLock"
+    if lock.exists():
+        lock.unlink()
+        print(f"  [chrome] Da xoa SingletonLock cu: {lock}")
+
+
+def launch_chrome_for_automation(
+    profile_path: Path,
+    debug_port: int,
+    timeout: float = 30.0,
+) -> tuple:
+    """
+    Mo Chrome voi cong CDP va ket noi Playwright qua CDP.
+    Tra ve (proc, playwright, browser, context, page).
+    Caller phai goi pw.stop() va close_chrome(proc) khi xong.
+    """
+    from playwright.sync_api import sync_playwright
+
+    profile_name = profile_path.name
+    _remove_singleton_lock(profile_path)
+
+    print(f"  [chrome] Khoi dong: {profile_name} (port {debug_port}) ...")
+    proc = _spawn_chrome(profile_path, debug_port=debug_port, url=SORA_URL)
+
+    print(f"  [chrome] Cho CDP san sang (port {debug_port}) ...")
+    _wait_for_cdp(debug_port, proc=proc, timeout=timeout)
+
+    print(f"  [chrome] Ket noi Playwright ...")
+    pw = sync_playwright().start()
+    browser = pw.chromium.connect_over_cdp(
+        f"http://127.0.0.1:{debug_port}",
+        slow_mo=BROWSER_SLOW_MO,
+    )
+    context = browser.contexts[0] if browser.contexts else browser.new_context()
+    page = context.pages[0] if context.pages else context.new_page()
+
+    print(f"  [chrome] San sang: {profile_name}")
+    return proc, pw, browser, context, page
+
+
+def close_chrome(proc: subprocess.Popen) -> None:
+    """Tat Chrome mot cach an toan."""
+    proc.terminate()
+    try:
+        proc.wait(timeout=5)
+    except subprocess.TimeoutExpired:
+        proc.kill()