import modelscope_studio.components.antd as antd
import modelscope_studio.components.base as ms


def Logo(model_display=None):
    """Logo component with optional model display"""
    with antd.Flex(vertical=True, gap="small", align="center", elem_style=dict(padding=8)):
        with antd.Typography.Title(level=1,
                                   elem_style=dict(fontSize=24,
                                                   margin=0)):
            with antd.Flex(align="center", gap="small", justify="center"):
                antd.Image('./assets/m-boxed-rainbow.png',
                           preview=False,
                           alt="logo",
                           width=24,
                           height=24)
                ms.Span("Ministrals Demo")
        
        # Display model name if provided
        if model_display is not None:
            with antd.Flex(justify="center"):
                model_display