Mbonea commited on
Commit
f200120
·
1 Parent(s): a068f90
App/Transcription/Schemas.py CHANGED
@@ -19,7 +19,7 @@ class BaseTranscription(BaseModel):
19
  tl_file_id: Optional[str] = "-"
20
  youtubeLink: Optional[str] = "-"
21
  task_id: str
22
- state: str = Field(default="PENDING", alias="transcription_state")
23
  content: list = Field(default=[], alias="result")
24
 
25
 
@@ -27,7 +27,7 @@ class TranscriptionResult(BaseModel):
27
  created_at: datetime = Field(default_factory=datetime.utcnow)
28
  duration: int = 0
29
  language: str = "-"
30
- transcription_state: str = "SUCCESS"
31
  transcript: list = []
32
 
33
  @property
 
19
  tl_file_id: Optional[str] = "-"
20
  youtubeLink: Optional[str] = "-"
21
  task_id: str
22
+ status: str = Field(default="PENDING", alias="transcription_state")
23
  content: list = Field(default=[], alias="result")
24
 
25
 
 
27
  created_at: datetime = Field(default_factory=datetime.utcnow)
28
  duration: int = 0
29
  language: str = "-"
30
+ status: str = Field(default="PENDING", alias="transcription_state")
31
  transcript: list = []
32
 
33
  @property
App/Transcription/TranscriptionRoutes.py CHANGED
@@ -133,8 +133,8 @@ async def create_file(
133
  async def get_status(task_id):
134
  entry: Transcriptions = await Transcriptions.objects.filter(task_id=task_id).first()
135
 
136
- result = BaseTranscription(**entry.__dict__)
137
- if result.transcription_state == "SUCCESS":
138
  return result
139
  task_result = AsyncResult(task_id)
140
 
@@ -158,5 +158,5 @@ async def get_status(task_id):
158
  _trans = TranscriptionMetadata(**task_result.result)
159
  await entry.update(**_trans.dict(exclude={"transcription"}))
160
 
161
- result = BaseTranscription(**entry.__dict__)
162
  return result
 
133
  async def get_status(task_id):
134
  entry: Transcriptions = await Transcriptions.objects.filter(task_id=task_id).first()
135
 
136
+ result = TranscriptionResult(**entry.__dict__)
137
+ if result.status == "SUCCESS":
138
  return result
139
  task_result = AsyncResult(task_id)
140
 
 
158
  _trans = TranscriptionMetadata(**task_result.result)
159
  await entry.update(**_trans.dict(exclude={"transcription"}))
160
 
161
+ result = TranscriptionResult(**entry.__dict__)
162
  return result