Add "marketing" to block.py BlockCategory + remove comments

This commit is contained in:
Bentlybro
2025-05-13 17:22:30 +01:00
parent 957706a941
commit b1e6161a7d
2 changed files with 1 additions and 22 deletions

View File

@@ -143,11 +143,6 @@ class VisualMediaType(str, Enum):
logger = logging.getLogger(__name__)
# ---------------------------------------------------------------------------
# Base mixin providing shared helper methods for the Revid.ai workflow
# ---------------------------------------------------------------------------
class _RevidMixin:
"""Utility mixin that bundles the shared webhook / polling helpers."""
@@ -280,8 +275,6 @@ class AIShortformVideoCreatorBlock(Block, _RevidMixin):
test_credentials=TEST_CREDENTIALS,
)
# create_webhook, create_video, check_video_status & wait_for_video live in _RevidMixin
def run(
self, input_data: Input, *, credentials: APIKeyCredentials, **kwargs
) -> BlockOutput:
@@ -337,11 +330,6 @@ class AIShortformVideoCreatorBlock(Block, _RevidMixin):
yield "video_url", video_url
# ---------------------------------------------------------------------------
# 2. AdMaker (text & image → 30s vertical AI video)
# ---------------------------------------------------------------------------
class AIAdMakerVideoCreatorBlock(Block, _RevidMixin):
"""Generates a 30second vertical AI advert using optional usersupplied imagery."""
@@ -473,11 +461,6 @@ class AIAdMakerVideoCreatorBlock(Block, _RevidMixin):
yield "video_url", video_url
# ---------------------------------------------------------------------------
# 3. PrompttoVideo (single prompt → fullyAI video)
# ---------------------------------------------------------------------------
class AIPromptToVideoCreatorBlock(Block, _RevidMixin):
"""Turns a single creative prompt into a fully AIgenerated video."""
@@ -587,11 +570,6 @@ class AIPromptToVideoCreatorBlock(Block, _RevidMixin):
yield "video_url", video_url
# ---------------------------------------------------------------------------
# 4. ScreenshottoVideo Advert (image + voice + avatar)
# ---------------------------------------------------------------------------
class AIScreenshotToVideoAdBlock(Block, _RevidMixin):
"""Creates an advert where the supplied screenshot is narrated by an AI avatar."""

View File

@@ -76,6 +76,7 @@ class BlockCategory(Enum):
PRODUCTIVITY = "Block that helps with productivity"
ISSUE_TRACKING = "Block that helps with issue tracking"
MULTIMEDIA = "Block that interacts with multimedia content"
MARKETING = "Block that helps with marketing"
def dict(self) -> dict[str, str]:
return {"category": self.name, "description": self.value}