Update video generation to use elai.

This commit is contained in:
Cristiano Ferreira
2024-10-01 18:12:56 +01:00
parent 164f47994b
commit 5289f33599
14 changed files with 32267 additions and 27613 deletions

24
app.py
View File

@@ -14,11 +14,11 @@ from helper.exercises import *
from helper.file_helper import delete_files_older_than_one_day
from helper.firebase_helper import *
from helper.gpt_zero import GPTZero
from helper.heygen_api import create_video, create_videos_and_save_to_db
from helper.elai_api import create_video, create_videos_and_save_to_db
from helper.openai_interface import *
from helper.question_templates import *
from helper.speech_to_text_helper import *
from heygen.AvatarEnum import AvatarEnum
from elai.AvatarEnum import AvatarEnum
from modules import GPT
from modules.training_content import TrainingContentService, TrainingContentKnowledgeBase
from modules.upload_level import UploadLevelService
@@ -984,7 +984,7 @@ def generate_video_1():
try:
data = request.get_json()
sp1_questions = []
avatar = data.get("avatar", random.choice(list(AvatarEnum)).value)
avatar = data.get("avatar", random.choice(list(AvatarEnum)).name)
request_id = str(uuid.uuid4())
logging.info("POST - generate_video_1 - Received request to generate video 1. "
@@ -992,13 +992,13 @@ def generate_video_1():
request.get_json()))
id_to_name = {
"5912afa7c77c47d3883af3d874047aaf": "MATTHEW",
"9e58d96a383e4568a7f1e49df549e0e4": "VERA",
"d2cdd9c0379a4d06ae2afb6e5039bd0c": "EDWARD",
"045cb5dcd00042b3a1e4f3bc1c12176b": "TANYA",
"1ae1e5396cc444bfad332155fdb7a934": "KAYLA",
"0ee6aa7cc1084063a630ae514fccaa31": "JEROME",
"5772cff935844516ad7eeff21f839e43": "TYLER",
"VADIM_BUSINESS": "MATTHEW",
"GIA_BUSINESS": "VERA",
"ORHAN_BUSINESS": "EDWARD",
"FLORA_BUSINESS": "TANYA",
"SCARLETT_BUSINESS": "KAYLA",
"ETHAN_BUSINESS": "JEROME",
"PARKER_CASUAL": "TYLER",
}
@@ -1052,7 +1052,7 @@ def generate_video_1():
def generate_video_2():
try:
data = request.get_json()
avatar = data.get("avatar", random.choice(list(AvatarEnum)).value)
avatar = data.get("avatar", random.choice(list(AvatarEnum)).name)
prompts = data.get("prompts", [])
question = data.get("question")
suffix = data.get("suffix", "")
@@ -1109,7 +1109,7 @@ def generate_video_3():
try:
data = request.get_json()
sp3_questions = []
avatar = data.get("avatar", random.choice(list(AvatarEnum)).value)
avatar = data.get("avatar", random.choice(list(AvatarEnum)).name)
request_id = str(uuid.uuid4())
logging.info("POST - generate_video_3 - Received request to generate video 3. "