87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
import os
|
|
import requests
|
|
import time
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
# Get HeyGen token
|
|
TOKEN = os.getenv("HEY_GEN_TOKEN")
|
|
|
|
# POST TO CREATE VIDEO
|
|
CREATE_VIDEO_URL = 'https://api.heygen.com/v1/video.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
|
|
}
|
|
|
|
def create_video(text):
|
|
# POST TO CREATE VIDEO
|
|
data = {
|
|
"background": "#ffffff",
|
|
"clips": [
|
|
{
|
|
"avatar_id": "Mido-lite-20221128",
|
|
"avatar_style": "normal",
|
|
"input_text": text,
|
|
"offset": {
|
|
"x": 0,
|
|
"y": 0
|
|
},
|
|
"scale": 1,
|
|
"voice_id": "ccb30e87c6b34ca8941f88352c71612d"
|
|
}
|
|
],
|
|
"ratio": "16:9",
|
|
"test": True,
|
|
"version": "v1alpha"
|
|
}
|
|
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 |