to_json fix

src/auto_archiver/enrichers/whisper_enricher.py

@@ -100,7 +100,19 @@ class WhisperEnricher(Enricher):
             r_res = requests.get(f'{self.api_endpoint}/jobs/{job_id}/artifacts', headers={'Authorization': f'Bearer {self.api_key}'})
             assert r_res.status_code == 200, f"Job artifacts did not respond with 200, instead with: {r_res.status_code}"
             logger.success(r_res.json())
-            return [artifact.get("data").get("text", "") for artifact in r_res.json()]
+            result = []
+            for artifact in r_res.json():
+                subtitle = []
+                full_text = []
+                for i, d in enumerate(artifact.get("data")):
+                    subtitle.append(f"{i+1}\n{d.get('start')} --> {d.get('end')}\n{d.get('text').strip()}")
+                    full_text.append(d.get('text').strip())
+                if not len(subtitle): continue
+                result.append({
+                    "subtitle": "\n".join(subtitle),
+                    "full_text": "\n".join(full_text),
+                })
+            return result
         return False
 
     def _get_s3_storage(self) -> S3Storage: