93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
import os
|
|
import requests
|
|
import time
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
from heygen.AvatarEnum import AvatarEnum
|
|
|
|
load_dotenv()
|
|
|
|
# Get HeyGen token
|
|
TOKEN = os.getenv("HEY_GEN_TOKEN")
|
|
|
|
# POST TO CREATE VIDEO
|
|
CREATE_VIDEO_URL = 'https://api.heygen.com/v1/template.generate'
|
|
GET_VIDEO_URL = 'https://api.heygen.com/v1/video_status.get'
|
|
POST_HEADER = {
|
|
'X-Api-Key': TOKEN,
|
|
'Content-Type': 'application/json'
|
|
}
|
|
GET_HEADER = {
|
|
'X-Api-Key': TOKEN
|
|
}
|
|
|
|
MATTHEW_NOAH = "11b234e504e44bfda9bc6b7aac3c8f81"
|
|
VERA_CERISE = "9bf2f27009cd403ab4ba4e22629b27bb"
|
|
EDWARD_TONY = "d3333e37952946059b45efc8482a2b6c"
|
|
TANYA_MOLLY = "07c75076b3f94df4ac658c6de72be83a"
|
|
KAYLA_ABBI = "d688099f8db9472cb4890b0561e81793"
|
|
JEROME_RYAN = "ad41feb2a5c4483085525e3d8907f512"
|
|
TYLER_CHRISTOPHER = "03c796f8ed274bb38f19e893bcbc6121"
|
|
|
|
|
|
def create_video(text, avatar: AvatarEnum):
|
|
# POST TO CREATE VIDEO
|
|
data = {
|
|
"template_id": avatar,
|
|
"title": "video_title",
|
|
"test": False,
|
|
"variables": [
|
|
{
|
|
"properties": {
|
|
"text": text
|
|
},
|
|
"name": "avatar_0"
|
|
}
|
|
]
|
|
}
|
|
response = requests.post(CREATE_VIDEO_URL, headers=POST_HEADER, json=data)
|
|
print(response.status_code)
|
|
print(response.json())
|
|
|
|
# GET TO CHECK STATUS AND GET VIDEO WHEN READY
|
|
video_id = response.json()["data"]["video_id"]
|
|
params = {
|
|
'video_id': response.json()["data"]["video_id"]
|
|
}
|
|
response = {}
|
|
status = "processing"
|
|
error = None
|
|
|
|
while status != "completed" and error is None:
|
|
response = requests.get(GET_VIDEO_URL, headers=GET_HEADER, params=params)
|
|
response_data = response.json()
|
|
|
|
status = response_data["data"]["status"]
|
|
error = response_data["data"]["error"]
|
|
|
|
if status != "completed" and error is None:
|
|
print(f"Status: {status}")
|
|
time.sleep(5) # Wait for 5 second before the next request
|
|
|
|
print(response.status_code)
|
|
print(response.json())
|
|
|
|
# DOWNLOAD VIDEO
|
|
download_url = response.json()['data']['video_url']
|
|
output_directory = 'download-video/'
|
|
output_filename = video_id + '.mp4'
|
|
|
|
response = requests.get(download_url)
|
|
|
|
if response.status_code == 200:
|
|
os.makedirs(output_directory, exist_ok=True) # Create the directory if it doesn't exist
|
|
output_path = os.path.join(output_directory, output_filename)
|
|
with open(output_path, 'wb') as f:
|
|
f.write(response.content)
|
|
print(f"File '{output_filename}' downloaded successfully.")
|
|
return output_filename
|
|
else:
|
|
print(f"Failed to download file. Status code: {response.status_code}")
|
|
return None
|