Spaces:
Runtime error
Runtime error
File size: 2,063 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
from typing import Callable, Iterable, List, Literal, TypeVar
T = TypeVar("T")
Conj = Literal["and", "or"]
def join_english(
items: Iterable[T],
to_str: Callable[[T], str] = str,
conj: Conj = "and",
) -> str:
s = list(map(to_str, items))
l = len(s)
assert l > 0
if l == 1:
return s[0]
if l == 2:
return f"{s[0]} {conj} {s[1]}"
return ", ".join(s[:-1]) + f", {conj} " + s[-1]
def format_image_with_channels(
channels: List[int],
conj: Conj = "and",
plural: bool = False,
) -> str:
assert len(channels) > 0
named = {1: "grayscale", 3: "RGB", 4: "RGBA"}
if all(x in named for x in channels):
if plural:
return join_english(channels, lambda c: named[c], conj=conj) + " images"
else:
return (
"a " + join_english(channels, lambda c: named[c], conj=conj) + " image"
)
if plural:
return f"images with {join_english(channels, conj=conj)} channel(s)"
else:
return f"an image with {join_english(channels, conj=conj)} channel(s)"
def format_color_with_channels(
channels: List[int],
conj: Conj = "and",
plural: bool = False,
) -> str:
assert len(channels) > 0
named = {1: "grayscale", 3: "RGB", 4: "RGBA"}
if all(x in named for x in channels):
if plural:
return join_english(channels, lambda c: named[c], conj=conj) + " colors"
else:
return (
"a " + join_english(channels, lambda c: named[c], conj=conj) + " color"
)
if plural:
return f"color with {join_english(channels, conj=conj)} channel(s)"
else:
return f"a color with {join_english(channels, conj=conj)} channel(s)"
_CHANNEL_NUMBER_NAME = {1: "GRAY", 3: "RGB", 4: "RGBA"}
def format_channel_numbers(input_channels: int, output_channels: int) -> str:
i = _CHANNEL_NUMBER_NAME.get(input_channels, str(input_channels))
o = _CHANNEL_NUMBER_NAME.get(output_channels, str(output_channels))
return f"{i}🠚{o}"
|