MVP Version

.gitattributes

@@ -0,0 +1,6 @@
+*.ts filter=lfs diff=lfs merge=lfs -text
+*.mp4 filter=lfs diff=lfs merge=lfs -text
+*.m4v filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text

.github/workflows/ci.yml

@@ -0,0 +1,39 @@
+name: CI Workflow
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout repository
+      uses: actions/checkout@v4
+
+    - name: Set up Python
+      uses: actions/setup-python@v5
+      with:
+        python-version: '3.12'
+
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip setuptools
+        sudo apt-get update && sudo apt-get install -y python3-opencv tesseract-ocr libtesseract-dev libleptonica-dev
+        pip install -r requirements.txt
+
+    - name: Verify installation
+      run: |
+        python -c "import cv2; import pytesseract; print(cv2.__version__); print(pytesseract.get_tesseract_version())"
+
+    - name: Run Black
+      run: |
+        pip install black
+        black --check .
+
+#    - name: Run tests
+#      run: |
+#        python -m unittest discover -s tests -t .

.gitignore

@@ -0,0 +1,7 @@
+__pycache__/
+*.py[cod]
+*$py.class
+
+*.venv
+
+assets/

README.md

@@ -0,0 +1,68 @@
+# Winner Finder
+
+Predict the winner of a wheel spin during live streams (from [pointauc.com](https://pointauc.com/)) before it's determined.
+
+## Description
+
+Want to predict the result of a wheel spin before the winner is announced? Use this tool! Here's a preview of how it works:
+
+![preview.gif](./preview.gif)
+
+## Installation
+
+1. Install the project:
+
+    ```shell
+    # clone the repository
+    git clone 
+
+    # create a virtual environment
+    python -m venv myenv
+
+    # activate the virtual environment
+    source ./.venv/bin/activate
+
+    # install dependencies
+    pip install -r requirements.txt
+    ```
+
+2. Install [tesseract](https://tesseract-ocr.github.io/tessdoc/Installation.html), which is required for text recognition in images. Make sure to install the necessary language packs from the [tessdata](https://github.com/tesseract-ocr/tessdata) repository.
+
+3. Install [streamlink](https://github.com/streamlink/streamlink), which allows you to grab a video stream from any platforms (Twitch, YouTube, Kick, etc...).
+
+   > ℹ️ [yt_dlp](https://github.com/yt-dlp/yt-dlp) can be used as alternative.
+
+## Run
+
+Run the program before the wheel spin starts:
+
+
+```shell
+streamlink --twitch-low-latency --stdout <channel link> best | python main.py winner
+
+# or you can use a shorthand
+./utils run <channel link>
+```
+
+## Lint
+
+Run the following commands to lint your code:
+
+```shell
+black .
+```
+
+## Test
+
+Run the test suite with:
+
+```shell
+python -m unittest discover -s tests -t .
+```
+
+## TODO list:
+
+- Properly handle ellipsis cases (the wheel isn't always a perfect circle)
+- Choose a more flexible solution for the config
+- Fix GitHub Actions
+- Improve linter setup

config/__init__.py

@@ -0,0 +1,41 @@
+from .config import (
+    ANGLE_WINDOW_LEN,
+    ASK_LENGTH,
+    CALCULATION_STEP,
+    EXCLUDE_SECONDS_RANGE,
+    FRAMES_STEP_FOR_LENGTH_DETECTION,
+    LOOK_BEHIND_FRAMES_FOR_LENGTH_DETECTION,
+    MAX_MEAN_ANGLE_DELTA,
+    MIN_SKIP_OF_WHEEL_SPIN,
+    MIN_SKIP_SEC,
+    NASTY_OPTIMIZATION,
+    READ_STEP,
+    SPIN_BUFFER_SIZE,
+    SPIN_BUFFER_SIZE_FOR_LENGTH_DETECTION,
+    SPIN_DETECT_LENGTH,
+    TESSERACT_LANG,
+    VISUALIZATION_ENABLED,
+)
+from .logger_config import add_global_event_time, setup_logger, update_global_event_time
+
+__all__ = [
+    "setup_logger",
+    "update_global_event_time",
+    "add_global_event_time",
+    "READ_STEP",
+    "ANGLE_WINDOW_LEN",
+    "CALCULATION_STEP",
+    "SPIN_BUFFER_SIZE",
+    "MIN_SKIP_OF_WHEEL_SPIN",
+    "MIN_SKIP_SEC",
+    "MAX_MEAN_ANGLE_DELTA",
+    "NASTY_OPTIMIZATION",
+    "SPIN_DETECT_LENGTH",
+    "SPIN_BUFFER_SIZE_FOR_LENGTH_DETECTION",
+    "FRAMES_STEP_FOR_LENGTH_DETECTION",
+    "LOOK_BEHIND_FRAMES_FOR_LENGTH_DETECTION",
+    "EXCLUDE_SECONDS_RANGE",
+    "ASK_LENGTH",
+    "TESSERACT_LANG",
+    "VISUALIZATION_ENABLED",
+]

config/config.py

@@ -0,0 +1,21 @@
+READ_STEP = 2
+ANGLE_WINDOW_LEN = 30
+CALCULATION_STEP = ANGLE_WINDOW_LEN * 3
+SPIN_BUFFER_SIZE = 3
+# The close to accurate is 1/4. The most accurate 1/3.
+MIN_SKIP_OF_WHEEL_SPIN = 1 / 3
+MIN_SKIP_SEC = 10
+MAX_MEAN_ANGLE_DELTA = 10.0
+TESSERACT_LANG = "rus+eng"  # "eng"
+
+NASTY_OPTIMIZATION = True
+
+SPIN_DETECT_LENGTH = True
+SPIN_BUFFER_SIZE_FOR_LENGTH_DETECTION = 10
+FRAMES_STEP_FOR_LENGTH_DETECTION = 10
+LOOK_BEHIND_FRAMES_FOR_LENGTH_DETECTION = 5
+EXCLUDE_SECONDS_RANGE = (30, 180)
+ASK_LENGTH = True
+
+
+VISUALIZATION_ENABLED = False

config/logger_config.py

@@ -0,0 +1,66 @@
+import logging
+import time
+
+import colorlog
+
+global_event_time = time.time()
+
+
+class DynamicAdapter(logging.LoggerAdapter):
+    def process(self, msg, kwargs):
+        extra = kwargs.get("extra", {})
+        extra_str = " ".join([f"{key}={value}" for key, value in extra.items()])
+        msg = f"{msg} {extra_str}" if extra_str else msg
+
+        return msg, kwargs
+
+
+class TimeDeltaFilter(logging.Filter):
+    def filter(self, record):
+        # Calculate the time difference in seconds from the global event
+        current_time = time.time()
+        delta_time = current_time - global_event_time
+        record.time_since_event = delta_time
+        return True
+
+
+def setup_logger(level=logging.DEBUG) -> DynamicAdapter:
+    logger = logging.getLogger("logger")
+    if logger.hasHandlers():
+        return DynamicAdapter(logger, {})
+
+    logger.setLevel(level)
+
+    formatter = colorlog.ColoredFormatter(
+        "%(asctime)s.%(msecs)03d (%(time_since_event).2fs ago) - %(log_color)s%(levelname)-8s%(reset)s %(white)s%(message)s",
+        datefmt="%Y-%m-%d %H:%M:%S",
+        reset=True,
+        log_colors={
+            "DEBUG": "cyan",
+            "INFO": "green",
+            "WARNING": "yellow",
+            "ERROR": "red",
+            "CRITICAL": "red,bg_white",
+        },
+        secondary_log_colors={},
+        style="%",
+    )
+
+    console_handler = logging.StreamHandler()
+    console_handler.setLevel(level)
+    console_handler.setFormatter(formatter)
+
+    logger.addHandler(console_handler)
+    logger.addFilter(TimeDeltaFilter())
+
+    return DynamicAdapter(logger, {})
+
+
+def update_global_event_time() -> None:
+    global global_event_time
+    global_event_time = time.time()
+
+
+def add_global_event_time(sec: int) -> None:
+    global global_event_time
+    global_event_time += sec