# Copyright (c) Alibaba, Inc. and its affiliates.
import unittest

from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.utils.plugins import PluginsManager
from modelscope.utils.test_utils import test_level


class PluginModelTest(unittest.TestCase):

    def setUp(self):
        self.package = 'adaseq'

    def tearDown(self):
        # make sure uninstalled after installing
        super().tearDown()
        import subprocess
        result = subprocess.run(
            ['pip', 'install', 'adaseq>=0.6.2', '--no-deps'],
            stdout=subprocess.PIPE)
        print(result.stdout.decode('utf-8'))

    @unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
    def test_run_span_based_ner_pipeline(self):
        try:
            pipeline_ins = pipeline(
                Tasks.named_entity_recognition,
                'damo/nlp_nested-ner_named-entity-recognition_chinese-base-med'
            )
            print(
                pipeline_ins(
                    '1、可测量目标： 1周内胸闷缓解。2、下一步诊疗措施：1.心内科护理常规，一级护理，低盐低脂饮食，留陪客。'
                    '2.予“阿司匹林肠溶片”抗血小板聚集，“呋塞米、螺内酯”利尿减轻心前负荷，“瑞舒伐他汀”调脂稳定斑块，“厄贝沙坦片片”降血压抗心机重构'
                ))
        except RuntimeError:
            print(
                'Skip test span_based_ner_pipeline! RuntimeError: Try loading from huggingface and modelscope failed'
            )

    def test_maoe_pipelines(self):
        try:
            pipeline_ins = pipeline(
                Tasks.named_entity_recognition,
                'damo/nlp_maoe_named-entity-recognition_chinese-base-general')
            print(
                pipeline_ins(
                    '刘培强，男，生理年龄40岁（因为在太空中进入休眠状态），实际年龄52岁，领航员国际空间站中的中国航天员，机械工程专家，军人，军衔中校。'
                ))
        except RuntimeError:
            print(
                'Skip test maoe_pipeline! RuntimeError: Try loading from huggingface and modelscope failed'
            )