devops/encoach_backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix speaking by using heygen api v2.

Browse Source
This commit is contained in:
Cristiano Ferreira
2024-01-03 16:23:26 +00:00
parent 479620116d
commit 9b3997f65e
2 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
helper/heygen_api.py
View File

@@ -92,23 +92,26 @@ def create_videos_and_save_to_db(exercises, template, id):
template["exercises"][2]["title"] = exercises[2]["topic"] template["exercises"][2]["title"] = exercises[2]["topic"]
save_to_db_with_id("speaking", template, id) save_to_db_with_id("speaking", template, id)
app.app.logger.info('Saved speaking to DB with id ' + id + " : " + str(template))
def create_video(text, avatar: AvatarEnum): def create_video(text, avatar: AvatarEnum):
# POST TO CREATE VIDEO # POST TO CREATE VIDEO
create_video_url = 'https://api.heygen.com/v2/template/' + avatar.value + '/generate'
data = { data = {
"template_id": avatar.value,
"title": "video_title",
"test": False, "test": False,
"variables": [ "caption": False,
{ "title": "video_title",
"variables": {
"script_here": {
"name": "script_here",
"type": "text",
"properties": { "properties": {
"text": text "content": text
}, }
"name": "avatar_0"
} }
] }
} }
response = requests.post(CREATE_VIDEO_URL, headers=POST_HEADER, json=data) response = requests.post(create_video_url, headers=POST_HEADER, json=data)
app.app.logger.info(response.status_code) app.app.logger.info(response.status_code)
app.app.logger.info(response.json()) app.app.logger.info(response.json())
@@ -130,7 +133,7 @@ def create_video(text, avatar: AvatarEnum):
if status != "completed" and error is None: if status != "completed" and error is None:
app.app.logger.info(f"Status: {status}") app.app.logger.info(f"Status: {status}")
time.sleep(5) # Wait for 5 second before the next request time.sleep(10) # Wait for 10 second before the next request
app.app.logger.info(response.status_code) app.app.logger.info(response.status_code)
app.app.logger.info(response.json()) app.app.logger.info(response.json())

14
heygen/AvatarEnum.py
View File

@@ -2,10 +2,10 @@ from enum import Enum
class AvatarEnum(Enum): class AvatarEnum(Enum):
MATTHEW_NOAH = "11b234e504e44bfda9bc6b7aac3c8f81" MATTHEW_NOAH = "5912afa7c77c47d3883af3d874047aaf"
VERA_CERISE = "9bf2f27009cd403ab4ba4e22629b27bb" VERA_CERISE = "9e58d96a383e4568a7f1e49df549e0e4"
EDWARD_TONY = "d3333e37952946059b45efc8482a2b6c" EDWARD_TONY = "d2cdd9c0379a4d06ae2afb6e5039bd0c"
TANYA_MOLLY = "07c75076b3f94df4ac658c6de72be83a" TANYA_MOLLY = "045cb5dcd00042b3a1e4f3bc1c12176b"
KAYLA_ABBI = "d688099f8db9472cb4890b0561e81793" KAYLA_ABBI = "1ae1e5396cc444bfad332155fdb7a934"
JEROME_RYAN = "ad41feb2a5c4483085525e3d8907f512" JEROME_RYAN = "0ee6aa7cc1084063a630ae514fccaa31"
TYLER_CHRISTOPHER = "03c796f8ed274bb38f19e893bcbc6121" TYLER_CHRISTOPHER = "5772cff935844516ad7eeff21f839e43"