Hugging Face's logo

Spaces:
vivym
/
dice-image
Runtime error

App Files Community
dice-image
File size: 3,650 Bytes 
61e513e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
 
import gradio as gr
from PIL import Image
from openpyxl import Workbook


def gray_to_dice_1(x: int) -> int:
    if 0 <= x and x <= 41:
        return 1
    elif 41 < x and x <= 83:
        return 2
    elif 83 < x and x <= 124:
        return 3
    elif 124 < x and x <= 165:
        return 4
    elif 165 < x and x <= 206:
        return 5
    elif 206 < x and x <= 247:
        return 6
    else:
        return 6


def gray_to_dice_2(x: int) -> int:
    if 0 <= x and x <= 41:
        return 0
    elif 41 < x and x <= 83:
        return 1
    elif 83 < x and x <= 124:
        return 2
    elif 124 < x and x <= 165:
        return 3
    elif 165 < x and x <= 206:
        return 4
    elif 206 < x and x <= 247:
        return 5
    else:
        return 6


def dice_to_image(
    image: Image.Image,
    resolution_x: int,
    resolution_y: int,
    dice_size: int,
):
    resolution_x, resolution_y = int(resolution_x), int(resolution_y)
    dice_size = int(dice_size)

    image = image.convert("L")
    image = image.resize((resolution_x, resolution_y))

    dice_img_1 = Image.new("L", (image.size[0] * dice_size, image.size[1] * dice_size))
    dice_img_2 = Image.new("L", (image.size[0] * dice_size, image.size[1] * dice_size))

    dices = [Image.new("L", (dice_size, dice_size))] + [
        Image.open(f"images/dices/{i}.jpeg").resize((dice_size, dice_size)).convert("L")
        for i in range(1, 7)
    ]

    wb = Workbook()
    for sheet_name in wb.sheetnames:
        wb.remove(wb[sheet_name])
    ws1 = wb.create_sheet("Style #1")
    ws2 = wb.create_sheet("Style #2")

    for i in range(image.size[0]):
        for j in range(image.size[1]):
            pixel = int(image.getpixel((i, j)))
            dice_index_1 = gray_to_dice_1(pixel)
            dice_index_2 = gray_to_dice_2(pixel)

            ws1.cell(row=j + 1, column=i + 1).value = dice_index_1
            ws2.cell(row=j + 1, column=i + 1).value = dice_index_2

            box = (
                i * dice_size,
                j * dice_size,
                (i + 1) * dice_size,
                (j + 1) * dice_size,
            )
            dice_img_1.paste(dices[dice_index_1], box=box)
            dice_img_2.paste(dices[dice_index_2], box=box)

    wb.save("results.xlsx")

    return dice_img_1, dice_img_2, "results.xlsx"


def main():
    with gr.Blocks() as app:
        gr.Markdown("Image to Dice")

        with gr.Row(variant="panel"):
            image_input = gr.Image(type="pil")

        with gr.Row(variant="panel"):
            resolution_x = gr.Number(value=200, precision=0, label="分辨率（宽）", show_label=True)
            resolution_y = gr.Number(value=200, precision=0, label="分辨率（高）", show_label=True)
            dice_size = gr.Number(value=16, precision=0, label="骰子大小", show_label=True)

        with gr.Row(variant="panel"):
            run_button = gr.Button("生成")

        with gr.Row(variant="panel"):
            image_output1 = gr.Image(type="pil", label="不用全黑", show_label=True)
            image_output2 = gr.Image(type="pil", label="用全黑", show_label=True)

        with gr.Row(variant="panel"):
            excel_file = gr.File(label="Excel结果", show_label=True)

        run_button.click(
            dice_to_image,
            inputs=[
                image_input,
                resolution_x,
                resolution_y,
                dice_size,
            ],
            outputs=[
                image_output1,
                image_output2,
                excel_file,
            ],
        )

        app.launch(share=True)


if __name__ == "__main__":
    main()