import gradio as gr
import os, sys, subprocess

if not __package__:
    from __init__ import Separator
else:
    from .. import Separator


class Plugin(Separator):
    def __init__(self):
        self.name = "Тестовый плагин"
        self.requirements = []
        self.install_requirements(self.requirements)

    def install_requirements(self, requirements: list):
        if requirements:
            cmd = [os.sys.executable, "-m", "pip", "install"]
            for pkg in requirements:
                cmd.append(pkg)
            result = subprocess.run(cmd, text=True, capture_output=True)

    def test(self):
        print("Тест")
        print(self.model_manager.get_mt())

    def UI(self):
        with gr.Column():
            gr.Markdown("<h3><center>Пример рабочего плагина</center></h3>")
            gr.Button("Показать все типы моделей", variant="primary").click(self.test)