update

routers/get_transcript.py

@@ -11,8 +11,8 @@ router = APIRouter(prefix="/get-transcript", tags=["transcript"])
 
 # api_key: str = Depends(get_api_key)
 @router.get("/")
-def get_transcript(audio_path: str, model_size: str = "distil-large-v3"):
-     # Run on GPU with FP16
+def get_transcript(audio_path: str, model_size: str = "distil-large-v3", api_key: str = Depends(get_api_key)):
+    # Run on GPU with FP16
     # model = WhisperModel(model_size, device="cuda", compute_type="float16")
 
     # or run on GPU with INT8
@@ -39,10 +39,52 @@ def get_transcript(audio_path: str, model_size: str = "distil-large-v3"):
 
     text = ""
 
+    listSentences = []
+
     for segment in segments:
         text += segment.text
+        listSentences.append({
+            "start_time": segment.start,
+            "end_time": segment.end,
+            "text": segment.text
+        })
+
 
     et = time.time()
     elapsed_time = et - st
-    return {"text": text}
+    return {"text": text,
+            'list_sentence':  listSentences
+            }
+
+#     time.sleep(5)
 
+#     return {
+#     "text": " She has a dimble on her left cheek, it adds charm to her facial features. The dimple is a genetic trait that she inherited from her mother. She's always been proud of it. People compliment her on it. She can't help but smile wider.",
+#     "list_sentence": [
+#         {
+#             "start_time": 0.0,
+#             "end_time": 8.0,
+#             "text": " She has a dimble on her left cheek, it adds charm to her facial features."
+#         },
+#         {
+#             "start_time": 8.0,
+#             "end_time": 16.0,
+#             "text": " The dimple is a genetic trait that she inherited from her mother."
+#         },
+#         {
+#             "start_time": 16.0,
+#             "end_time": 20.0,
+#             "text": " She's always been proud of it."
+#         },
+#         {
+#             "start_time": 20.0,
+#             "end_time": 24.0,
+#             "text": " People compliment her on it."
+#         },
+#         {
+#             "start_time": 24.0,
+#             "end_time": 28.0,
+#             "text": " She can't help but smile wider."
+#         }
+#     ]
+# }