app.py

import urllib.parse
import gradio as gr

# ---------------------------
# 배지 URL 생성 함수 정의
# ---------------------------
def generate_static_badge(label, message, color, label_color, logo, logo_color, style, link):
    base = "https://img.shields.io/static/v1"
    params = []
    if label:
        params.append(f"label={urllib.parse.quote(label, safe='')}")
    if message:
        params.append(f"message={urllib.parse.quote(message, safe='')}")
    if color:
        params.append(f"color={urllib.parse.quote(color, safe='')}")
    if label_color:
        params.append(f"labelColor={urllib.parse.quote(label_color, safe='')}")
    if logo:
        params.append(f"logo={urllib.parse.quote(logo, safe='')}")
    if logo_color:
        params.append(f"logoColor={urllib.parse.quote(logo_color, safe='')}")
    if style:
        params.append(f"style={urllib.parse.quote(style, safe='')}")

    badge_url = base + ("?" + "&".join(params) if params else "")
    if link:
        html_code = f'<a href="{link}" target="_blank"><img src="{badge_url}" alt="badge"></a>'
    else:
        html_code = f'<img src="{badge_url}" alt="badge">'

    badge_preview = f"""
    <div style='padding:10px; text-align:center;'>
        {html_code}
    </div>
    """
    return html_code, badge_preview

# ---------------------------
# Gradio UI 구성
# ---------------------------
with gr.Blocks(theme=gr.themes.Default()) as demo:
    gr.Markdown("""
    # BadgeCraft - Badge Generator 🛠️
    생성된 배지를 상단에서 바로 확인하고, 아래 폼을 통해 실시간으로 옵션을 조정하세요.
    """)

    out_code = gr.Code(label="HTML Snippet", language="html")
    out_preview = gr.HTML(label="Badge Preview")

    with gr.Row():
        label = gr.Textbox(label="Label", placeholder="예: build")
        message = gr.Textbox(label="Message", placeholder="예: passing")
        logo = gr.Textbox(label="Logo", placeholder="예: github")

    with gr.Row():
        color = gr.ColorPicker(label="Color", value="#007ec6")
        label_color = gr.ColorPicker(label="Label Color", value="#555")
        logo_color = gr.ColorPicker(label="Logo Color", value="#ffffff")

    style = gr.Dropdown(label="Style", choices=["flat", "flat-square", "plastic", "for-the-badge", "social"], value="flat")
    link = gr.Textbox(label="Link (배지 클릭 시 이동할 URL)", placeholder="https://yourlink.com")

    inputs = [label, message, color, label_color, logo, logo_color, style, link]
    for inp in inputs:
        inp.change(fn=generate_static_badge, inputs=inputs, outputs=[out_code, out_preview])

    # 초기값 한 번 출력
    generate_static_badge("build", "passing", "#007ec6", "#555", "github", "#ffffff", "flat", "https://shields.io")

# 실행
# demo.launch()