SS-2005's picture
Added Multi-lingual naration in users preferred language
5576989
Raw
History Blame Contribute Delete
1.22 kB
# # test_main.py
# import os
# import json
# # The session ID whose state JSON already exists in S3:
# # s3://tech-learn-state/video_states/{SESSION_ID}/{SESSION_ID}.json
# SESSION_ID = "sess_08d00cc5"
# # main.py reads INPUT_PAYLOAD from env var
# input_payload = json.dumps({
# "session_id": SESSION_ID
# })
# os.environ["INPUT_PAYLOAD"] = input_payload
# # Now import and run
# import main
# main.main()
# test_consultation_pipeline.py
import logging
logging.basicConfig(level=logging.INFO)
from src.state import VideoGenerationState
from src.nodes.consultation.consultation_pipeline import ConsultationPipelineNode
state = VideoGenerationState(
session_id="mock_user_arjun_007",
topic="consultation",
target_language="english",
optional_params={"script_type": "consultation"}
)
node = ConsultationPipelineNode()
result = node.generate_consultation_video(state)
print(f"Status: {result.status}")
print(f"Video: {result.slide_video_path}")
"""
mock_user_priya_001
mock_user_rahul_002
mock_user_anjali_003
mock_user_kiran_004
mock_user_deepak_005
mock_user_meera_006
mock_user_arjun_007
mock_user_sunita_008
mock_user_vikram_009
"""