File size: 4,105 Bytes
54639e8 f20025d 9d54dfd 5f6fd2a 54639e8 b936660 54639e8 b936660 54639e8 2df2397 b936660 54639e8 b936660 54639e8 b936660 503d4ac 54639e8 2df2397 503d4ac 2df2397 503d4ac 2df2397 503d4ac 2df2397 54639e8 2df2397 54639e8 503d4ac 54639e8 503d4ac 54639e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 |
import os
import asyncio
from src.logger_config import logger
from src.utils import clean_tts_script
from src.config import get_config_value, set_config_value
from src.pipelines.factory import get_automation_pipeline
from src.asset_manager import get_asset_downloader
from google_src.gcs_utils import list_gcs_files
import hashlib
def configure_row_settings(row: dict, row_index: int = 0) -> str:
"""
Configure all row-related settings in the global config.
Args:
row: The row dict containing TTS Script, prompts, etc.
row_index: The current row/audio index for tracking.
Returns:
The original tts_script value (for workflow tracking/logging).
"""
# Determine tts_script based on on_screen_text setting
tts_script = row.get("TTS Script (AI Avatar)", "")
if get_config_value("on_screen_text", False):
tts_script = row.get("On-Screen Text", "").strip()
# Get downloaded videos from singleton
asset_downloader = get_asset_downloader()
visual_assets = get_config_value("visual_assets", {})
visual_assets["all_videos"] = asset_downloader.downloaded_videos
set_config_value("visual_assets", visual_assets)
# Set per-row configuration
set_config_value("row_index", row_index)
set_config_value("current_audio_index", row_index)
set_config_value("original_tts_script", tts_script)
set_config_value("tts_script", clean_tts_script(tts_script))
set_config_value("hash_tts_script", hashlib.sha256(get_config_value("tts_script").encode('utf-8')).hexdigest())
set_config_value("gemini_prompt", row.get("Gemini Imagen4 Ultra Prompt (specific)", ""))
set_config_value("runway_prompt", row.get("Runway Prompt Gen4 Turbo", ""))
set_config_value("runway_veo_prompt", row.get("Veo-3.1 Fast Prompt (Text-to-Video)", ""))
set_config_value("captions", row.get("Captions", ""))
return tts_script
async def process_single_row(row: dict) -> dict:
"""Process one CSV row using the main pipeline."""
tts_script = get_config_value("original_tts_script", "")
logger.debug(f"▶️ Executing: {tts_script}")
original_merge_type = get_config_value("video_merge_type")
merge_process = get_config_value("video_merge_process")
# Handle comma-separated merge types
merge_types = [t.strip() for t in str(original_merge_type).split(",")] if original_merge_type else [None]
if merge_process == "sameeachtype":
# Process all types for this row
pass
elif len(merge_types) > 1:
# Round Robin / Alternative behavior
current_index = get_config_value("row_index", 0)
selected_type = merge_types[current_index % len(merge_types)]
logger.debug(f"🔄 Round Robin: Row {current_index} selected '{selected_type}' from {merge_types}")
merge_types = [selected_type]
results = []
try:
for merge_type in merge_types:
if merge_type:
logger.debug(f"🔄 Switching VIDEO_MERGE_TYPE to: {merge_type}")
set_config_value("video_merge_type", merge_type)
pipeline = get_automation_pipeline()
result = await pipeline.run_pipeline()
results.append(result)
status_icon = "✅" if result.get('success', False) else "❌"
logger.debug(f"{status_icon} Completed {tts_script[:20]}... [{merge_type or 'Default'}]: success={result.get('success', False)}")
finally:
# Restore original config
if original_merge_type:
set_config_value("video_merge_type", original_merge_type)
list_gcs_files()
return results
async def download_all_library_videos():
"""Download all library videos once using singletons."""
# Get the asset downloader singleton
asset_downloader = get_asset_downloader()
# Download all videos using the singleton
logger.debug("📥 Pre-downloading all library videos...")
videos = await asset_downloader.download_all_videos()
logger.debug(f"✓ Downloaded {len(videos)} library videos")
return videos
|