Mbonea commited on
Commit
a068f90
·
1 Parent(s): 051c4d0

possible fixes

Browse files
App/Transcription/Schemas.py CHANGED
@@ -20,7 +20,7 @@ class BaseTranscription(BaseModel):
20
  youtubeLink: Optional[str] = "-"
21
  task_id: str
22
  state: str = Field(default="PENDING", alias="transcription_state")
23
- # content: list = []
24
 
25
 
26
  class TranscriptionResult(BaseModel):
 
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
 
26
  class TranscriptionResult(BaseModel):
App/Transcription/TranscriptionRoutes.py CHANGED
@@ -131,6 +131,11 @@ async def create_file(
131
 
132
  @transcription_router.get("/tasks/{task_id}")
133
  async def get_status(task_id):
 
 
 
 
 
134
  task_result = AsyncResult(task_id)
135
 
136
  if task_result.result == None:
@@ -140,7 +145,6 @@ async def get_status(task_id):
140
  "task_result": task_result.result,
141
  }
142
 
143
- entry: Transcriptions = await Transcriptions.objects.filter(task_id=task_id).first()
144
  if task_result.status == "SUCCESS":
145
  trans = TranscriptionResult(**task_result.result)
146
  trans
 
131
 
132
  @transcription_router.get("/tasks/{task_id}")
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
 
141
  if task_result.result == None:
 
145
  "task_result": task_result.result,
146
  }
147
 
 
148
  if task_result.status == "SUCCESS":
149
  trans = TranscriptionResult(**task_result.result)
150
  trans