objektify2 / main.py
tripleS-Dev
update V1.3.0L Beta
d10d21b
raw
history blame
15.6 kB
import gradio as gr
import beta_fn
import fns
import generate
import login_discord
import preset_fn
import rekord
import pre
js_func = """
function refresh() {
const url = new URL(window.location);
if (url.searchParams.get('__theme') !== 'dark') {
url.searchParams.set('__theme', 'dark');
window.location.href = url.href;
}
}
"""
theme = gr.themes.Soft(
secondary_hue="violet",
text_size="sm",
spacing_size="sm",
radius_size="lg",
).set(
block_background_fill='*secondary_50'
)
login_uri = pre.login_uri
with gr.Blocks(theme=theme) as demo: # , js=js_func
with gr.Column():
with gr.Row():
gr.Markdown(
"""
# Objektify
""")
login = gr.Button(size="sm", value="Login to Discord", link=f"https://discord.com/oauth2/authorize?client_id=1239473790732472370&response_type=code&redirect_uri={login_uri}&scope=identify", scale=0)
with gr.Row():
gr.Markdown(
"""
### [Join objektify official discord!](https://discord.com/invite/KAuTCUGXpH)
## Input image and Select Options
"""
)
preset = gr.Dropdown(scale=0, label='Preset', interactive=True, allow_custom_value=False, type='value', visible=False)
with gr.Tab("Front"):
with gr.Row():
with gr.Column(): #1
img_type = gr.Radio(["Image", "Gif"], value="Image", scale=1, label="type")
img_static = gr.Image(label="Image", interactive=True, sources=['upload', 'clipboard'], type="pil", visible=True, scale=10)
img_dynamic_file = gr.File(file_types=[".gif"], visible=False, scale=1, interactive=True, label="Gif File", type="filepath")
img_dynamic_show = gr.Image(label="Gif", interactive=False, sources=['upload', 'clipboard'], type="pil", visible=False, scale=10)
img_type.change(fn=fns.change_img_type, inputs=[img_type], outputs=[img_static, img_dynamic_file, img_dynamic_show])
img_static.upload(fn=fns.img2cache_resize, inputs=img_static, outputs=img_static)
img_dynamic_file.upload(fn=fns.gif2cache, inputs=img_dynamic_file, outputs=img_dynamic_show)
img_dynamic_file.clear(fn=fns.gif_clear, outputs=img_dynamic_show)
with gr.Column(): #2
with gr.Accordion(label='Colors'): #objekt color
obj_color = gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Special", "OMA","Other color", "AI Color"], value="Cream01", label='Objekt Color')
with gr.Row():
ai_num = gr.Slider(1, 1000, 500, label="AI Serial", visible=False, step=1)
change_ai = gr.Button(value="Change AI", visible=False)
obj_color_picker = gr.ColorPicker(label="Other Objekt Color", visible=False, interactive=True, value="#ffffff")
obj_color.change(fn=fns.obj_color_update, inputs=obj_color, outputs=[obj_color_picker, ai_num, change_ai])
txt_color = gr.Radio(["White", "Black", "Other"], label="Text Color", value="Black")
txt_color_picker = gr.ColorPicker(label="Other Text Color", value="#ffffff", visible=False, interactive=True)
txt_color.change(fns.txt_color_update, txt_color, txt_color_picker)
with gr.Accordion('Names'):
group_name_radio = gr.Radio(choices=pre.groups_Other, label="Group Name")
group_name_txt = gr.Textbox(label="Other Group name", placeholder="CLASSY", visible=False, interactive=True)
member_name_dropdown = gr.Dropdown(choices=pre.names, label="Member Name", allow_custom_value=True)
with gr.Accordion("Add Objekt Number", open=False):
with gr.Row():
obj_num = gr.Textbox(label="Objekt Number", placeholder="100", interactive=True, visible=True, scale=1)
obj_alphabet = gr.Radio(["Z", "A", "Other"], label="Z or A", interactive=True, visible=True, scale=1, value='Z')
obj_alphabet_txt = gr.Textbox(label="Alphabet", placeholder="S", interactive=True, visible=False, scale=1)
obj_alphabet.change(fns.alphabet_txt_update, obj_alphabet, obj_alphabet_txt)
with gr.Row():
obj_serial = gr.Textbox(label="Objekt Serial", placeholder="00001", interactive=True, visible=True, scale=1)
obj_serial_random = gr.Button(value="Random Serial", interactive=True, visible=True, scale=1)
obj_serial_random.click(fn=fns.serial_random, outputs=obj_serial)
with gr.Group():
rounded = gr.Checkbox(label="Rounded", value=True, interactive=True)
with gr.Group():
generate_btn = gr.Button(value="Generate")
with gr.Column():
edited_img = gr.Image(label="Edited Image", scale=10)
with gr.Tab("Back") as back_tab:
with gr.Row():
preview = gr.Image(label="Image", interactive=False)
with gr.Column():
with gr.Accordion('Contents'):
obj_outline_color = gr.Radio(["White", "Black", "Other"], label="Frame color", value="White")
obj_outline_color_picker = gr.ColorPicker(label="Other Outline Color", value="#ffffff", visible=False, interactive=True)
obj_outline_color.change(fns.frame_color_update, obj_outline_color, obj_outline_color_picker)
logo_radio = gr.Radio(choices=pre.groups, label="Add Logo", interactive=True, value="None")
with gr.Accordion('Texts'):
class_radio = gr.Radio(["Zero", "First", "Double", "Special", "Other"], label="CLASS", visible=True)
class_txt = gr.Textbox(label="Other class", placeholder="Third", visible=False)
class_radio.change(fns.txt_class_visibility, class_radio, class_txt)
season_radio = gr.Radio(["Atom01", "Binary01", "Cream01", "Divine01", "Other"], label="SEASON", type="value")
with gr.Row():
season_txt = gr.Textbox(label="Other SEASON", interactive=True, visible=False)
season_txt_outline = gr.Textbox(label="Other SEASON Outline", interactive=True, visible=False)
season_radio.change(fns.txt_season_visibility, season_radio, [season_txt, season_txt_outline])
sign = gr.Dropdown(choices=pre.names, label="Sign", type='index')
with gr.Accordion('QR Options'):
qr_url_dropdown = gr.Dropdown(["tripleS website", "tripleS youtube", "tripleS 𝕏", "tripleS discord"], label="QR url", info="Can write any url", allow_custom_value=True, value='https://objektify.wav.haus')
qr_Logo_radio = gr.Radio(["tripleS", "None"], label="QR Logo", value="None", interactive=True)
right_btn = gr.Checkbox(label="Add Rights", value=False, interactive=True)
rounded_back = gr.Checkbox(label="Rounded", value=True, interactive=True)
generate_btn_back = gr.Button(value="Generate")
with gr.Column():
outputs2 = gr.Image(label="Edited Image", scale=10)
with gr.Tab("Export") as ex:
with gr.Row():
ex_img_f = gr.Image(label="Front", type="pil", interactive=False)
ex_file_f = gr.File(render=True, visible=False)
ex_img_b = gr.Image(label="Back", type="pil", interactive=False)
with gr.Column():
with gr.Group():
with gr.Row():
ex_hide_blur = gr.Checkbox(label="Blur")
ex_hide_blur_range = gr.Slider(1, 20, 5, label="Blur Range", step=1, interactive=False, scale=10)
ex_hide_blur.select(fns.ex_blur, ex_hide_blur, ex_hide_blur_range)
with gr.Group():
watermark = gr.Textbox(label="Add watermark")
ex_gen_file_count = gr.Radio(["One File", "Two Files"], label="Export Type", value="One File")
ex_btn = gr.Button(value="Export")
with gr.Row():
ex_img_f_out = gr.Image(label="Front", interactive=False)
ex_img_b_out = gr.Image(label="Back", visible=False, interactive=False)
with gr.Tab("Beta", visible=True) as beta:
gr.Markdown(
"""
## This is the beta channel for testing new features.
### Please join [here](https://discord.com/invite/KAuTCUGXpH) to report bugs and provide feedback.
"""
)
with gr.Row():
with gr.Column():
preset_name = gr.Textbox(label="Set preset name", placeholder="aespa winter")
artist_name = gr.Dropdown(label="Artist name", multiselect=True, allow_custom_value=True, info="Press Enter to add more than 1 people (Max 12)", max_choices=12)
group_name = gr.Textbox(label="Group name", placeholder="aespa")
company_name = gr.Textbox(label="Company name", placeholder="SM ent")
top_logo_img = gr.Image(label="Top Logo", type='pil')
qr_logo_img = gr.Image(label="QR Logo", type='pil')
with gr.Accordion('Signs', open=True):
with gr.Row():
sign_img1 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img2 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img3 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img4 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
with gr.Row():
sign_img5 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img6 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img7 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img8 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
with gr.Row():
sign_img9 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img10 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img11 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
sign_img12 = gr.Image(label="Sign", visible=False, interactive=True, type='pil')
save = gr.Button(value="Save and Apply all")
artist_name.change(fn=fns.custom_name, inputs=artist_name, outputs=[sign_img1, sign_img2, sign_img3, sign_img4, sign_img5, sign_img6, sign_img7, sign_img8, sign_img9, sign_img10, sign_img11, sign_img12])
with gr.Tab("ReKord", visible=False) as rec:
rekord_img = gr.Image(type="filepath")
rekord_member = gr.Dropdown(choices=pre.names, label="Sign", type='index')
rekord_submit = gr.Button()
rekord_out = gr.Textbox()
rekord_submit.click(fn=rekord.upload, inputs=[rekord_img, rekord_member], outputs=rekord_out)
with gr.Tab("Dev", visible=False) as dev:
dev_btn = gr.Button(value="refresh", visible=True, interactive=True)
dev_out = gr.Textbox()
dev_drop = gr.Dropdown()
dev_file = gr.File()
dev_btn2 = gr.Button(value="move")
dev_btn3 = gr.Button(value="zip")
dev_btn4 = gr.Button(value="download")
dev_btn5 = gr.Button(value="remove")
dev_btn2.click(fn=fns.dev2, outputs=dev_out)
dev_btn3.click(fn=fns.dev3, inputs=dev_drop, outputs=dev_out)
dev_btn4.click(fn=fns.dev4, inputs=dev_drop, outputs=dev_file)
dev_btn5.click(fn=fns.dev5, inputs=dev_drop, outputs=dev_out)
dev_btn.click(fns.dev, outputs=[dev_out, dev_drop])
gr.Markdown(
"""
## Made by
# Discord : hj_sss
## official discord
### [Click here and Join](https://discord.com/invite/KAuTCUGXpH)
### Members or group presets can be added and advertised. Check the official discord.
#### Check the objektify official discord for licensing and other data processing policies.
""")
userid = gr.Textbox(visible=False)
preset.change(fn=preset_fn.apply, inputs=preset,
outputs=[group_name_radio, member_name_dropdown, logo_radio, sign, qr_Logo_radio, preset])
group_name_radio.change(fns.group_member_name_update, [group_name_radio, artist_name, preset], [group_name_txt, member_name_dropdown])
save.click(fn=beta_fn.save, inputs=[preset_name, artist_name, group_name, company_name, top_logo_img, qr_logo_img, sign_img1, sign_img2, sign_img3, sign_img4, sign_img5, sign_img6, sign_img7, sign_img8, sign_img9, sign_img10, sign_img11, sign_img12, userid, login], outputs=[group_name_radio, member_name_dropdown, logo_radio, sign, qr_Logo_radio, preset])
generate_btn.click(fn=generate.sorting,
inputs=[img_type, img_static, img_dynamic_file, obj_color, obj_color_picker, ai_num, txt_color,
txt_color_picker, member_name_dropdown, group_name_radio, group_name_txt, obj_num,
obj_alphabet, obj_alphabet_txt, obj_serial, rounded], outputs=[edited_img, ex_img_f, ex_file_f])
ex_gen_file_count.change(fns.ex_gen_count, ex_gen_file_count, ex_img_b_out)
rounded.change(fns.rounded_sync, rounded, rounded_back)
rounded_back.change(fns.rounded_sync, rounded_back, rounded)
change_ai.click(fns.change_ran, obj_color, ai_num)
generate_btn_back.click(fn=generate.back,
inputs=[obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group_name_radio,
group_name_txt, member_name_dropdown, obj_num, obj_alphabet, obj_alphabet_txt,
obj_serial, obj_outline_color, obj_outline_color_picker, logo_radio, class_radio,
class_txt, season_radio, season_txt, season_txt_outline, sign, qr_url_dropdown,
qr_Logo_radio, right_btn, rounded_back, preset], outputs=[outputs2, ex_img_b])
back_tab.select(fn=generate.back,
inputs=[obj_color, obj_color_picker, ai_num, txt_color, txt_color_picker, group_name_radio,
group_name_txt, member_name_dropdown, obj_num, obj_alphabet, obj_alphabet_txt, obj_serial,
obj_outline_color, obj_outline_color_picker, logo_radio, class_radio, class_txt,
season_radio, season_txt, season_txt_outline, sign, qr_url_dropdown, qr_Logo_radio,
right_btn, rounded_back, preset], outputs=[preview, ex_img_b])
ex_btn.click(fn=generate.ex_gen, inputs=[ex_img_f, ex_file_f, ex_img_b, ex_gen_file_count, ex_hide_blur, ex_hide_blur_range, watermark], outputs=[ex_img_f_out, ex_img_b_out])
demo.load(fn=login_discord.id_check, outputs=[login, dev, rec, userid, beta, preset])
demo.load(fn=preset_fn.refresh, outputs=preset)
if __name__ == "__main__":
print("http://localhost:7860")
demo.launch(server_name="0.0.0.0")