with open('components/label-management-dialog.tsx', 'r') as f:
    content = f.read()

# Add language to useLanguage()
content = content.replace(
    'const { t } = useLanguage()',
    'const { t, language } = useLanguage()'
)

# Add dir to Dialog
content = content.replace(
    '<Dialog>',
    '<Dialog dir={language === \'fa\' || language === \'ar\' ? \'rtl\' : \'ltr\'}>'
)

content = content.replace(
    '<Dialog open={open} onOpenChange={onOpenChange}>',
    '<Dialog open={open} onOpenChange={onOpenChange} dir={language === \'fa\' || language === \'ar\' ? \'rtl\' : \'ltr\'}>'
)

with open('components/label-management-dialog.tsx', 'w') as f:
    f.write(content)