t

app.py

@@ -17,7 +17,7 @@ def pipeline(video_url):
     sentences = re.split(r"[\.\!\?]\s", punctuated_text)
     classification, probs = predict_from_document(sentences)
     #    return punctuated_text
-    times = match_mask_and_transcript(sentences, transcript, classification)
+    times, timestamps = match_mask_and_transcript(sentences, transcript, classification)
     return [
         {
             "start": "12:05",
@@ -25,6 +25,7 @@ def pipeline(video_url):
             "classification": str(classification),
             "probabilities": probs,
             "times": times,
+            "timestamps": timestamps,
         }
     ]
 

functions/convert_time.py

@@ -47,4 +47,4 @@ def match_mask_and_transcript(split_punct, transcript, classification):
         else:
             times[current] = (times[current][0], elem[0] + elem[1])
             current_time = elem[0] + elem[1]
-    return times
+    return times, timestamps