Stable-Texturify/Tex_Remap.py at main · Voxatro/Stable-Texturify · GitHub

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
import base64
import io
import json
import requests
from PIL import Image
import yaml
def image_to_base64(image_path):
    with Image.open(image_path) as img:
        buffer = io.BytesIO()
        img.save(buffer, format="PNG")
        return base64.b64encode(buffer.getvalue()).decode()

def base64_to_image(base64_data, output_path):
    img_data = base64.b64decode(base64_data)
    img = Image.open(io.BytesIO(img_data))
    img.save(output_path, format="PNG")

with open("Render_config.yaml", "r") as f:
    config = yaml.safe_load(f)
sd_model = config["sd_model_checkpoint"]
img1 = image_to_base64("tex_fusion.png")
bin_uv = image_to_base64("binary_uv.png")
bin_uv_layout = image_to_base64("binary_uv_layout.png")

img_list = [img1]
url = "http://127.0.0.1:7860/sdapi/v1/img2img"
# model_list = requests.get("http://127.0.0.1:7860/sdapi/v1/sd-models")
# print(model_list.text)
for i,img in enumerate(img_list):
    data = {
        "init_images": [img],

        'prompt': 'vrass,textured',
        'negative_prompt': 'blur,motion blur',
        'enable_hr': False,
        'denoising_strength': 0.2,
        # 'hr_upscaler': 'Latent (bicubic)',
        # 'hr_scale': 1.0,
        'seed': -1,
        'sampler_name': 'Euler a',
        'batch_size': 1,
        'steps': 20,
        #'quick_steps': 20,
        'cfg_scale': 7,
        'width': 512,
        'height': 512,
        # 'override_settings': {'CLIP_stop_at_last_layers': 1},
        'override_settings': {'sd_model_checkpoint': 'imageToVroidStudio_12.ckpt'},

        # 'override_settings_restore_afterwards': 'false',
        # 'hr_second_pass_steps': 20,
    "alwayson_scripts": {
        "controlnet": {
        "args": [
            {
            "enabled":True,
            # "module":"tile_resample",
            "input_image": bin_uv_layout,
            "model": "control_v11p_sd15_seg",
            "resize_mode": 0,
            "processor_res":512,
            "weight":1,
            # "threshold_a":1

            },
            {
            "enabled":False,
            "input_image": bin_uv,
            # "module":'invert',
            "model": "control_v11p_sd15s2_lineart_anime",
                "resize_mode": 0,
                "processor_res":512,
            "weight":0.5

            },
            {
            "enabled":True,
            "module":"tile_resample",
            "input_image": img,
            "model": "control_v11f1e_sd15_tile",
            "resize_mode": 0,
            "processor_res":2048,
            "weight":1,
            "threshold_a":1
            }
        ]
        }
    }
    }

    headers = {"Content-Type": "application/json"}

    response = requests.post(url, data=json.dumps(data), headers=headers)

    if response.status_code == 200:
        print("Request was successful")
        response_data = response.json()
        output_image_path = "tex_remap_{}.png"
        base64_to_image(response_data["images"][0], output_image_path.format(str(i)))
        print(f"Saved output image to {output_image_path.format(str(i))}")
    else:
        print(f"Request failed with status code {response.status_code}")