Thought I had staged it

app/dtos/speaking.py

@@ -1,9 +1,9 @@
 import random
-from typing import List, Dict
+from typing import List, Dict, Optional
 
 from pydantic import BaseModel
 
-from app.configs.constants import MinTimers, ELAIAvatars
+from app.configs.constants import MinTimers
 
 
 class SaveSpeakingDTO(BaseModel):
@@ -21,22 +21,7 @@ class GradeSpeakingAnswersDTO(BaseModel):
 
 
 class GenerateVideo1DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
+    avatar: str = Optional[str]
     questions: List[str]
     first_topic: str
     second_topic: str
-
-
-class GenerateVideo2DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
-    prompts: List[str] = []
-    suffix: str = ""
-    question: str
-    topic: str
-
-
-class GenerateVideo3DTO(BaseModel):
-    avatar: str = (random.choice(list(ELAIAvatars))).name
-    questions: List[str]
-    topic: str
-

app/dtos/video.py

@@ -0,0 +1,15 @@
+import random
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel
+
+class TaskStatus(Enum):
+    STARTED = "STARTED"
+    IN_PROGRESS = "IN_PROGRESS"
+    COMPLETED = "COMPLETED"
+    ERROR = "ERROR"
+
+class Task(BaseModel):
+    status: TaskStatus
+    result: Optional[str] = None