Update functions/punctuation.py

functions/punctuation.py

@@ -49,10 +49,13 @@ def parse_output(output, comb):
 
 
 def punctuate(video_id):
-    transcript = retrieve_transcript(video_id)
-    splits = split_transcript(
-        transcript
-    )  # Get the transcript from the YoutubeTranscriptApi
-    resp = query_punctuation(splits)  # Get the response from the Inference API
-    punctuated_transcript = parse_output(resp, splits)
-    return punctuated_transcript, transcript
+    try:
+        transcript = retrieve_transcript(video_id)
+        splits = split_transcript(
+            transcript
+        )  # Get the transcript from the YoutubeTranscriptApi
+        resp = query_punctuation(splits)  # Get the response from the Inference API
+        punctuated_transcript = parse_output(resp, splits)
+        return punctuated_transcript, transcript
+    except Exception as e:
+        return None, str(e)