Upload folder using huggingface_hub

.github/workflows/python_app.yaml

@@ -0,0 +1,39 @@
+# This workflow will install Python dependencies, run tests and lint with a single version of Python
+# For more information see: https://docs.github.com/en/actions/automating-builds-and-tests/building-and-testing-python
+
+name: Python application
+
+on:
+  push:
+    branches: [ "master" ]
+  pull_request:
+    branches: [ "master" ]
+
+permissions:
+  contents: read
+
+jobs:
+  build:
+
+    runs-on: ubuntu-latest
+
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python 3.12
+      uses: actions/setup-python@v3
+      with:
+        python-version: "3.12"
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install flake8 pytest
+        if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+    - name: Lint with flake8
+      run: |
+        # stop the build if there are Python syntax errors or undefined names
+        flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+        # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+        flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+    - name: Test with pytest
+      run: |
+        pytest

.github/workflows/update_space.yml

@@ -16,13 +16,17 @@ jobs:
     - name: Set up Python
       uses: actions/setup-python@v2
       with:
-        python-version: '3.9'
+        python-version: '3.12'
 
     - name: Install Gradio
       run: python -m pip install gradio
 
-    # - name: Log in to Hugging Face
-    #   run: python -c 'import huggingface_hub; huggingface_hub.login(token="${{ secrets.hf_token }}")'
+    - name: Install dependencies
+      run: |
+        python -m pip install -r requirements.txt
+
+    - name: Run tests
+      run: pytest --maxfail=3 --disable-warnings -q
 
     - name: Log in to Hugging Face
       run: |

.pytest_cache/.gitignore

@@ -0,0 +1,2 @@
+# Created by pytest automatically.
+*

.pytest_cache/CACHEDIR.TAG

@@ -0,0 +1,4 @@
+Signature: 8a477f597d28d172789f06886806bc55
+# This file is a cache directory tag created by pytest.
+# For information about cache directory tags, see:
+#	https://bford.info/cachedir/spec.html

.pytest_cache/README.md

@@ -0,0 +1,8 @@
+# pytest cache directory #
+
+This directory contains data from the pytest's cache plugin,
+which provides the `--lf` and `--ff` options, as well as the `cache` fixture.
+
+**Do not** commit this to version control.
+
+See [the docs](https://docs.pytest.org/en/stable/how-to/cache.html) for more information.

.pytest_cache/v/cache/nodeids

@@ -0,0 +1,8 @@
+[
+  "tests/test_graphics_tools.py::test_compute_difference",
+  "tests/test_graphics_tools.py::test_find_closest_color_idx",
+  "tests/test_graphics_tools.py::test_replace_colors_by_template",
+  "tests/test_tools.py::test_create_color_palette_square_grid",
+  "tests/test_tools.py::test_create_palette_image",
+  "tests/test_tools.py::test_extract_color_palette_basic"
+]

pyproject.toml

@@ -10,4 +10,5 @@ dependencies = [
     "openai-agents>=0.0.16",
     "pillow>=11.2.1",
     "scikit-learn>=1.6.1",
+    "pytest>=8.4.0",
 ]

requirements.txt

@@ -71,6 +71,8 @@ idna==3.10
     #   anyio
     #   httpx
     #   requests
+iniconfig==2.1.0
+    # via pytest
 jinja2==3.1.6
     # via gradio
 jiter==0.10.0
@@ -106,12 +108,15 @@ packaging==25.0
     #   gradio
     #   gradio-client
     #   huggingface-hub
+    #   pytest
 pandas==2.2.3
     # via gradio
 pillow==11.2.1
     # via
     #   brandsmith-gradio (pyproject.toml)
     #   gradio
+pluggy==1.6.0
+    # via pytest
 pydantic==2.11.5
     # via
     #   fastapi
@@ -127,7 +132,11 @@ pydantic-settings==2.9.1
 pydub==0.25.1
     # via gradio
 pygments==2.19.1
-    # via rich
+    # via
+    #   pytest
+    #   rich
+pytest==8.4.0
+    # via brandsmith-gradio (pyproject.toml)
 python-dateutil==2.9.0.post0
     # via pandas
 python-dotenv==1.1.0

tests/test_graphics_tools.py

@@ -0,0 +1,34 @@
+import sys
+import os
+import pytest
+from PIL import Image
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src')))
+import graphics_tools
+
+class DummyImageTemplate:
+    def __init__(self, image, colors):
+        self.image = image
+        self.colors = colors
+
+def test_find_closest_color_idx():
+    color = (100, 100, 100)
+    template_colors = [(0, 0, 0), (255, 255, 255), (110, 110, 110)]
+    idx = graphics_tools.find_closest_color_idx(color, template_colors)
+    assert idx == 2
+
+def test_compute_difference():
+    color1 = (10, 20, 30)
+    color2 = (20, 10, 40)
+    diff = graphics_tools.compute_difference(color1, color2)
+    assert diff == 30
+
+def test_replace_colors_by_template(tmp_path):
+    # Create a dummy image
+    img = Image.new('RGB', (2, 2), (100, 100, 100))
+    img_path = tmp_path / 'dummy.png'
+    img.save(img_path)
+    colors = [(100, 100, 100), (200, 200, 200)]
+    template = DummyImageTemplate(str(img_path), colors)
+    target_colors = [(1, 2, 3), (4, 5, 6)]
+    out_img = graphics_tools.replace_colors_by_template(template, target_colors)
+    assert isinstance(out_img, Image.Image)

tests/test_tools.py

@@ -0,0 +1,41 @@
+import sys
+import os
+import pytest
+import numpy as np
+from PIL import Image
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../src')))
+import tools
+from models import ColorPalette, ColorItem
+
+class DummyColorItem:
+    def __init__(self, R, G, B, count):
+        self.R = R
+        self.G = G
+        self.B = B
+        self.count = count
+
+class DummyColorPalette:
+    def __init__(self, colors):
+        self.colors = colors
+
+def test_extract_color_palette_basic():
+    img = Image.new('RGBA', (10, 10), (255, 0, 0, 255))
+    palette = tools.extract_color_palette(img, 1)
+    assert hasattr(palette, 'colors')
+    assert len(palette.colors) == 1
+    assert palette.colors[0].R == 255
+    assert palette.colors[0].G == 0
+    assert palette.colors[0].B == 0
+
+def test_create_color_palette_square_grid():
+    colors = [ColorItem(R=255, G=0, B=0, count=50), ColorItem(R=0, G=255, B=0, count=50)]
+    palette = ColorPalette(colors=colors)
+    img = tools.create_color_palette_square_grid(palette, total_pixels=100)
+    assert isinstance(img, Image.Image)
+    assert img.size[0] == img.size[1]  # Should be square
+
+def test_create_palette_image():
+    colors = [ColorItem(R=255, G=0, B=0, count=50), ColorItem(R=0, G=255, B=0, count=50)]
+    palette = ColorPalette(colors=colors)
+    img = tools.create_palette_image(palette)
+    assert isinstance(img, Image.Image)