toolchemy 0.2.185__py3-none-any.whl

toolchemy/vision/caption_overlay.py

@@ -0,0 +1,77 @@
+import os
+from dataclasses import dataclass
+
+from PIL import Image, ImageDraw, ImageFont, ImageFilter
+
+from toolchemy.utils.locations import Locations
+
+
+@dataclass
+class Caption:
+    text: str
+    y: int
+    font: ImageFont.FreeTypeFont | None = None
+    font_name: str | None = None
+    size: int | None = None
+    color: tuple | None = None
+
+    def __post_init__(self):
+        if self.font_name is None:
+            self.font_name = "Pacifico.ttf"
+        if self.size is None:
+            self.size = 60
+        self.font = ImageFont.truetype(self.font_name, self.size)
+        if self.color is None:
+            self.color = (255, 255, 255, 255)
+
+
+
+def add_caption(input_img_path: str, captions: list[Caption], output_img_path: str | None = None):
+    # fonts = ["Pacifico.ttf", "Anton.ttf"]
+
+    if output_img_path is None:
+        path_root, path_ext = os.path.splitext(input_img_path)
+        output_img_path = f"{path_root}_out{path_ext}"
+
+    img = Image.open(input_img_path).convert("RGBA")
+
+    txt = Image.new("RGBA", img.size, (255,255,255,0))
+    draw = ImageDraw.Draw(txt)
+
+    x_start = 50
+    for i, caption in enumerate(captions):
+        draw.text((x_start, caption.y), caption.text, font=caption.font, fill=caption.color)
+
+    glow = txt.filter(ImageFilter.GaussianBlur(5))
+
+    out = Image.alpha_composite(img.convert("RGBA"), glow)
+    out = Image.alpha_composite(out, txt)
+
+    out.convert("RGB").save(output_img_path)
+
+
+def main():
+    std_text = "*bilet uprawnia do jednej soboty lub niedzieli przed kompem, czy coś :)"
+    std_y_pos = 1900
+    std_font_size = 45
+    locations = Locations()
+    font_name_pacifico = "/path/to/Pacifico-Regular.ttf"
+    font_name_open_sans_italic = "/path/to/OpenSans-Italic-VariableFont_wdth,wght.ttf"
+    bottom_font_name = font_name_open_sans_italic
+    month_font_name = font_name_open_sans_italic
+    add_caption(locations.in_data("kima1.jpg"), captions=[
+        Caption(text="Lorem ipsum", y=30, size=100, font_name=font_name_pacifico),
+        Caption(text="dolor sit amet, consectetur adipiscing elit, sed do eiusmod", y=210, size=90, font_name=font_name_pacifico),
+        Caption(text="tempor incididunt ut labore et dolore magna aliqua.", y=390, size=60, font_name=month_font_name),
+        Caption(text=std_text, y=std_y_pos-300, size=std_font_size, font_name=bottom_font_name),
+    ])
+    add_caption(locations.in_data("kima2.jpg"), captions=[
+        Caption(text="Ut enim ad minim veniam,", y=30, size=100, font_name=font_name_pacifico),
+        Caption(text="quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.", y=210, size=90,
+                font_name=font_name_pacifico),
+        Caption(text=std_text, y=std_y_pos, size=std_font_size, font_name=bottom_font_name),
+    ])
+
+
+if __name__ == "__main__":
+    main()

toolchemy/vision/image.py

@@ -0,0 +1,89 @@
+import base64
+from PIL import Image, UnidentifiedImageError
+from io import BytesIO
+
+from toolchemy.utils.logger import get_logger
+
+
+class UnknownImageFormatError(Exception):
+    pass
+
+
+class ImageProcessor:
+    def __init__(self, image_path: str):
+        self._logger = get_logger()
+        self._img = None
+        self._image_path = image_path
+
+    @property
+    def img(self) -> Image.Image:
+        self._open()
+        return self._img
+
+    @property
+    def base64(self) -> str:
+        self._open()
+        buffered = BytesIO()
+        self._img.save(buffered, format=self._img.format)  # Save as JPEG in memory
+        img_bytes = buffered.getvalue()
+        return base64.b64encode(img_bytes).decode('utf-8')
+
+    def metadata(self) -> dict:
+        self._open()
+
+        img_file = BytesIO()
+        self._img.save(img_file, self._img.format)
+
+        metadata = {
+            "width": self._img.width,
+            "height": self._img.height,
+            "size_bytes": img_file.tell(),
+            "format": self._img.format,
+            "format_description": self._img.format_description,
+            "filename": self._img.filename,
+        }
+
+        return metadata
+
+    def scale(self, max_edge_len: int, upscale: bool = False):
+        cur_w = self._img.width
+        cur_h = self._img.height
+
+        self._logger.info(f"> size (w, h): ({cur_w}, {cur_h})")
+
+        if cur_h > cur_w:
+            resize_ratio = cur_h / max_edge_len
+        else:
+            resize_ratio = cur_w / max_edge_len
+
+        self._logger.info(f"> resize ratio: {resize_ratio}")
+
+        if resize_ratio <= 1.0 and not upscale:
+            self._logger.info(f"upscaling disabled, skipping")
+            return
+
+        new_h = int(cur_h // resize_ratio)
+        new_w = int(cur_w // resize_ratio)
+
+        self._logger.info(f"> new size (w, h): ({new_w}, {new_h})")
+
+        self._img = self._img.resize((new_w, new_h))
+
+    def __enter__(self):
+        self._open()
+        return self
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        self._close()
+
+    def _open(self):
+        if self._img is None:
+            try:
+                self._img = Image.open(self._image_path)
+            except UnidentifiedImageError as e:
+                raise UnknownImageFormatError(str(e))
+
+    def _close(self):
+        if self._img:
+            self._img.close()
+            self._img = None

toolchemy-0.2.185.dist-info/METADATA

@@ -0,0 +1,25 @@
+Metadata-Version: 2.4
+Name: toolchemy
+Version: 0.2.185
+Summary: a set of auxiliary programming tools
+License-File: LICENSE
+Author: Cyprian Nosek
+Author-email: cyprian.nosek@protonmail.com
+Requires-Python: >=3.12,<3.13
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: colorlog (>=6.10.1,<7.0.0)
+Requires-Dist: diskcache (>=5.6.3,<6.0.0)
+Requires-Dist: google-genai (>=1.55.0,<2.0.0)
+Requires-Dist: jsonschema (>=4.26.0,<5.0.0)
+Requires-Dist: mlflow (>=3.8.1,<4.0.0)
+Requires-Dist: neptune-scale (>=0.13.0,<0.14.0)
+Requires-Dist: numpy (==2.0.2)
+Requires-Dist: ollama (>=0.6.0,<0.7.0)
+Requires-Dist: openai (>=1.82.1,<2.0.0)
+Requires-Dist: pillow (>=11.2.1,<12.0.0)
+Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
+Requires-Dist: tenacity (>=9.1.2,<10.0.0)
+Requires-Dist: tinydb (>=4.8.2,<5.0.0)
+Requires-Dist: torch (>=2.7.0,<3.0.0)
+Requires-Dist: wyoming (>=1.7.1,<2.0.0)

toolchemy-0.2.185.dist-info/RECORD

@@ -0,0 +1,36 @@
+toolchemy/__main__.py,sha256=WSJ1UlOGt4ZcWWGB7HRAdeAFcna0Dx7hhiWCi7QjCqM,163
+toolchemy/ai/clients/__init__.py,sha256=MaTGkzEWZL6BVhgAdd2zfFc9bzYD5p8DRYIc2m_hbus,640
+toolchemy/ai/clients/common.py,sha256=Ccb3Y1jNZUxtburSk5kddO8rrnaBjgxKtAADlb1WLV8,18943
+toolchemy/ai/clients/dummy_model_client.py,sha256=hBCVzXFfV61OYdgU4hIJNIxlm9-w63ME-jfQyTo1GFg,2417
+toolchemy/ai/clients/factory.py,sha256=xMGfaagJ3D-yEksKRouwQ4bHMpVMrFergbRDrs7d9QU,1953
+toolchemy/ai/clients/gemini_client.py,sha256=oUnMXtdRqTY-xpIsuFKtyiXmdsIcX92IEIgeVEGUDK4,2486
+toolchemy/ai/clients/ollama_client.py,sha256=_PYI_udyTBmtKZlvRECzeWKrrR-NwTlaGJS_1COMmok,3152
+toolchemy/ai/clients/openai_client.py,sha256=20Yt8azldRzbd8ylKqoD5frIEj8LhScjSYhGCt0WcWk,3611
+toolchemy/ai/clients/pricing.py,sha256=FuAuarQIaLl-e6KLzLbiWawWpqjY_kV4poO_qiTuHfA,1953
+toolchemy/ai/clients/whisper_client.py,sha256=AQcY2FiGLu0JTYw0zzjjH-7WyZ6NbQWrtxFa6iYhR4M,5252
+toolchemy/ai/prompter.py,sha256=vp6IrCu3c8IJmXGzVjFjOWdf3EvnUW0Z1Bn5STHAIbs,4933
+toolchemy/ai/trackers/__init__.py,sha256=FUv2NuGSmCpDrbE7rmeG0VmQr9rJRC-Y_D0MuQa0e-M,214
+toolchemy/ai/trackers/common.py,sha256=Y8_QHD1BAsJcBEHAekO-0Txe2MeBwGqh9TM53Ta49qk,6202
+toolchemy/ai/trackers/mlflow_tracker.py,sha256=fVX3YdNEDs9aS5UhlY4boaAsz2ArKnawrhLKli0AS2s,8627
+toolchemy/ai/trackers/neptune_tracker.py,sha256=vNvPzfrqqi7tRCfBadB-4VWobE5JR1i6ae23SQQC7I0,4639
+toolchemy/db/lightdb.py,sha256=Bp7AimIuOfp2jIORFvBndk8kuKnEV12Ua1hoQXR5brY,9377
+toolchemy/utils/__init__.py,sha256=6Mp65G6Ce_zY86AxvjyzkMjU_e8cbeij7TTHYqLwO0U,574
+toolchemy/utils/at_exit_collector.py,sha256=YXupFjnujFK86n7dxH1Qol_i7FjNXJyo2q5f4SA3OEU,3682
+toolchemy/utils/cacher/__init__.py,sha256=onsfc6z_rPQeo2rA3Ckwf8prtzg7wzEp_JE2cTEgyBI,430
+toolchemy/utils/cacher/cacher_diskcache.py,sha256=mgZwy3lmgK3-22qGOfl35VX3ZFfRy-1JC8dmHK7Hjbc,4714
+toolchemy/utils/cacher/cacher_pickle.py,sha256=totHrA39WrLmuW8AGg584YHx_FYoYSopCvmWyfTxrxg,6531
+toolchemy/utils/cacher/cacher_shelve.py,sha256=9ywARTqVHVlRwKETF0oOQXXnHd6QSekkKBJIl8L0r1M,7352
+toolchemy/utils/cacher/common.py,sha256=xOD6NlYxNJlhSKQAYhwTZmgU8YhzXmG8DjfDrlIPTL4,5028
+toolchemy/utils/datestimes.py,sha256=jt2L26WwuST8tfieDuhajgr24mkA6INC5ae-kYGRfIw,2586
+toolchemy/utils/locations.py,sha256=ql63bI_oPeuqGeUPYnqLCi-TMqO0o2LM3Ibe_S62PIc,3966
+toolchemy/utils/logger.py,sha256=VxR_Chr-3aWnTJ7NtwrN7bvznmlO--NCgy2eWwB-lkE,2463
+toolchemy/utils/timer.py,sha256=sQzGnxlPgAC8bxS0ROHKVCNctUFAP6b5UJcsM9Q-__w,596
+toolchemy/utils/utils.py,sha256=JCRNRBJ4vRWzl-GtbetJs8GM72ImVAbDkPIXrvp3IE0,5167
+toolchemy/vision/__init__.py,sha256=p-sQaVd1FUQ1LMGdF51y28Z9IJvMVWaCdvrL0-jKDgc,141
+toolchemy/vision/caption_overlay.py,sha256=fLxHMD7IPSh5XJC5pvdPf1wZvHidlOvnXt_wgW7UJjk,2825
+toolchemy/vision/image.py,sha256=5q7MRK2SEyqWPB62duuzJqGmqBfQP1Tgphu01nP7O8s,2392
+toolchemy-0.2.185.dist-info/METADATA,sha256=wJElgbNStdDSNOu1GnJR4WyA26XSyBAVsXiCnUxEzzg,942
+toolchemy-0.2.185.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+toolchemy-0.2.185.dist-info/entry_points.txt,sha256=6l1k81pMAx7Ap1FaX8qW8SDwsONnNLOb28QQD9t6gto,66
+toolchemy-0.2.185.dist-info/licenses/LICENSE,sha256=JY5EPk3yvwZqQHk8JrQ-TFARdyfVTK9uUKV6CuKo_Bs,1062
+toolchemy-0.2.185.dist-info/RECORD,,

toolchemy-0.2.185.dist-info/WHEEL

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

toolchemy-0.2.185.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+prompt-studio=toolchemy.ai.prompter:run_studio
+

toolchemy-0.2.185.dist-info/licenses/LICENSE

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