50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
import os
|
|
import folder_paths
|
|
from server import PromptServer
|
|
import numpy as np
|
|
import cv2
|
|
import torch
|
|
|
|
class Doom:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
FUNCTION = "run_doom"
|
|
CATEGORY = "DOOM"
|
|
|
|
def run_doom(self):
|
|
img = np.zeros((512, 768, 3), dtype=np.uint8)
|
|
font = cv2.FONT_HERSHEY_SIMPLEX
|
|
|
|
# Draw header
|
|
header = "RyanOnTheInside"
|
|
header_size = cv2.getTextSize(header, font, 2, 3)[0]
|
|
headerX = (img.shape[1] - header_size[0]) // 2
|
|
headerY = img.shape[0] // 4
|
|
cv2.putText(img, header, (headerX, headerY), font, 2, (255,255,255), 3)
|
|
|
|
# Draw social links
|
|
links = [
|
|
"@RyanOnTheInside everywhere",
|
|
"Node Suite:",
|
|
"github.com/ryanontheinside/ComfyUI_RyanOnTheInside"
|
|
]
|
|
|
|
y_offset = img.shape[0] // 2
|
|
for text in links:
|
|
textsize = cv2.getTextSize(text, font, 0.7, 2)[0]
|
|
textX = (img.shape[1] - textsize[0]) // 2
|
|
cv2.putText(img, text, (textX, y_offset), font, 0.7, (255,255,255), 2)
|
|
y_offset += 50
|
|
|
|
tensor = torch.from_numpy(img).float() / 255.0
|
|
tensor = tensor.unsqueeze(0)
|
|
return (tensor,)
|
|
|
|
|