Update video generation to use elai.
This commit is contained in:
24
app.py
24
app.py
@@ -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. "
|
||||
|
||||
Reference in New Issue
Block a user