import bpy
from .main_panel import AvatarToolKit_PT_AvatarToolkitPanel, CATEGORY_NAME
from ..functions.viseme import AvatarToolKit_OT_AutoVisemeButton
from ..core.translations import t
from ..core.common import get_selected_armature

class AvatarToolkitVisemePanel(bpy.types.Panel):
    bl_label = t("VisemePanel.label")
    bl_idname = "OBJECT_PT_avatar_toolkit_viseme"
    bl_space_type = 'VIEW_3D'
    bl_region_type = 'UI'
    bl_category = CATEGORY_NAME
    bl_parent_id = AvatarToolKit_PT_AvatarToolkitPanel.bl_idname
    bl_order = 7

    def draw(self, context: bpy.types.Context) -> None:
        layout = self.layout
        
        armature = get_selected_armature(context)
        if armature:
            layout.label(text=t("VisemePanel.label"), icon='SOUND')
            
            layout.separator(factor=0.5)
            
            layout.prop(context.scene.avatar_toolkit, "selected_mesh", text=t("VisemePanel.select_mesh"), icon='OUTLINER_OB_MESH')
            
            mesh = bpy.data.objects.get(context.scene.avatar_toolkit.selected_mesh)
            if mesh and mesh.type == 'MESH':
                if mesh.data.shape_keys:
                    box = layout.box()
                    col = box.column(align=True)
                    col.prop_search(context.scene.avatar_toolkit, "mouth_a", mesh.data.shape_keys, "key_blocks", text=t('VisemePanel.mouth_a.label'), icon='SHAPEKEY_DATA')
                    col.prop_search(context.scene.avatar_toolkit, "mouth_o", mesh.data.shape_keys, "key_blocks", text=t('VisemePanel.mouth_o.label'), icon='SHAPEKEY_DATA')
                    col.prop_search(context.scene.avatar_toolkit, "mouth_ch", mesh.data.shape_keys, "key_blocks", text=t('VisemePanel.mouth_ch.label'), icon='SHAPEKEY_DATA')

                    layout.separator(factor=0.5)

                    layout.prop(context.scene.avatar_toolkit, 'shape_intensity', text=t('VisemePanel.shape_intensity'), icon='FORCE_LENNARDJONES')

                    layout.separator(factor=1.0)

                    row = layout.row()
                    row.scale_y = 1.5
                    row.operator(AvatarToolKit_OT_AutoVisemeButton.bl_idname, text=t('VisemePanel.create_visemes'), icon='TRIA_RIGHT')
                else:
                    layout.label(text=t('VisemePanel.error.noShapekeys'), icon='ERROR')
            else:
                layout.label(text=t('VisemePanel.error.selectMesh'), icon='INFO')
        else:
            layout.label(text=t('VisemePanel.error.noArmature'), icon='ERROR')

        layout.separator(factor=1.0)
        layout.label(text=t('VisemePanel.info.selectMesh'), icon='HELP')