Tools / src /automation.py
jebin2's picture
refactor: Pipeline architecture revamp with proper separation
54639e8
from pipelines.base import ContentAutomationBase
from pipelines.ai_pipeline import AIContentAutomationBase
from pipelines.standard_ai_pipeline import StandardAIPipeline
from pipelines.avatar_ai_pipeline import AvatarAIPipeline
from pipelines.non_ai_pipeline import NonAIContentAutomationBase
from pipelines.hard_cut_pipeline import HardCutPipeline
from pipelines.beats_cut_pipeline import BeatsCutPipeline
from pipelines.factory import get_automation_pipeline
# Re-export ContentAutomation for backward compatibility,
# but note that directly instantiating this specific class might be deprecated
# if the logic was split.
# However, if 'ContentAutomation' was used as the MAIN entry point, it should probably
# point to the Factory or the Base?
# The original code did `ContentAutomation().run_pipeline()`.
# The refactored code uses `get_automation_pipeline().run_pipeline()`.
# Allowing old imports to work if they just want the class definitions
ContentAutomation = ContentAutomationBase
__all__ = [
"ContentAutomationBase",
"AIContentAutomationBase",
"StandardAIPipeline",
"AvatarAIPipeline",
"NonAIContentAutomationBase",
"HardCutPipeline",
"BeatsCutPipeline",
"get_automation_pipeline",
"ContentAutomation"
]