Update transforming/whispertransform.py

transforming/whispertransform.py

@@ -40,22 +40,27 @@ class WhisperTransform(Transform):
         print(f"Video title and url: {video.title} {video.url}")
         audio_file = self._get_audio_from_video(yt)
 
-        result = self.model.transcribe(audio_file, 
-                                       without_timestamps=self.without_timestamps)
-        transcription = result["text"]
-        
-        data = []
-        for seg in result['segments']:
-            data.append(OrderedDict({'start': seg['start'], 'end': seg['end'],'text': seg['text']}))
+        try:
+            result = self.model.transcribe(audio_file, 
+                                           without_timestamps=self.without_timestamps)
+        except Exception as e:
+            print(f"Audio exception print: {e}")
+            pass
+        else:
+            transcription = result["text"]
+
+            data = []
+            for seg in result['segments']:
+                data.append(OrderedDict({'start': seg['start'], 'end': seg['end'],'text': seg['text']}))
 
-        os.remove(audio_file)
+            os.remove(audio_file)
 
-        return YoutubeVideo(channel_name = video.channel_name,
-                            url = video.url,
-                            title = video.title,
-                            description = video.description,
-                            transcription = transcription,
-                            segments = data)
+            return YoutubeVideo(channel_name = video.channel_name,
+                                url = video.url,
+                                title = video.title,
+                                description = video.description,
+                                transcription = transcription,
+                                segments = data)
         
     def _get_audio_from_video(self, yt: Any) -> Path:
         # TODO: Add credits