from pyapp.application import PyApplication
from runners.task_generator import TaskGenerator
from runners.data_generator import DataGenerator
from runners.data_recorder import DataRecorder
from runners.task_templates_divider import TaskTemplatesDivider

@PyApplication("agent")
class AgentApp():
    @staticmethod
    def start():
        TaskGenerator(config_path="configs/task_generator.json").run()
        DataGenerator(config_path="configs/data_generator.json").run()
        DataRecorder(config_path="configs/data_recorder.json").run()

@PyApplication("divide_task")
class DivideTaskApp():
    @staticmethod
    def start():
        TaskTemplatesDivider(config_path="configs/divide_task_config.yaml").run()

@PyApplication("generate_task")
class GenerateTaskApp():
    @staticmethod
    def start():
        TaskGenerator(config_path="configs/task_generator.json").run()

@PyApplication("generate_data")
class GenerateDataApp():
    @staticmethod
    def start():
        DataGenerator(config_path="configs/data_generator.json").run()

@PyApplication("record_data")
class RecordDataApp():
    @staticmethod
    def start():
        DataRecorder(config_path="configs/data_recorder.json").run()