Update app.py

app.py

@@ -22,6 +22,7 @@ def Transcribe(audio="temp_audio.wav"):
         spl = timeStr.split(":")
         s = (int)((int(spl[0]) * 60 * 60 + int(spl[1]) * 60 + float(spl[2]) )* 1000)
         return s
+    
     def preprocess(audio):
         t1 = 0 * 1000 
         t2 = 20 * 60 * 1000
@@ -32,6 +33,7 @@ def Transcribe(audio="temp_audio.wav"):
         newAudio = spacer.append(a, crossfade=0)
         newAudio.export(audio, format="wav")
         return spacermilli, spacer
+    
     def diarization(audio):
         as_audio = AudioSegment.from_wav(audio)
         DEMO_FILE = {'uri': 'blabal', 'audio': audio}
@@ -109,31 +111,37 @@ def VideoTranscribe(video):
     subprocess.call(command, shell=True)
     return Transcribe()
 
-def YoutubeTranscribe(url):
-    try:
-        os.remove("temp_audio.wav")
-    except OSError:
-        pass
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'outtmpl': 'temp_audio.%(ext)s',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'wav',
-        }],
-    }
-    try:
-      with youtube_dl.YoutubeDL(ydl_opts) as ydl:
-          ydl.download([url])
-    except:
-        return YoutubeTranscribe(url)
-    stream = ffmpeg.input('temp_audio.m4a')
-    stream = ffmpeg.output(stream, 'temp_audio.wav')
-    try:
-        os.remove("temp_audio.m4a")
-    except OSError:
-        pass
-    return Transcribe()
+def YoutubeTranscribe(URL, retries = 5):
+    if retries:
+        if "youtu" not in URL.lower():
+            raise Exception(f"{url} is not a valid youtube URL.")
+        else:
+            try:
+                os.remove("temp_audio.wav")
+            except OSError:
+                pass
+            ydl_opts = {
+                'format': 'bestaudio/best',
+                'outtmpl': 'temp_audio.%(ext)s',
+                'postprocessors': [{
+                    'key': 'FFmpegExtractAudio',
+                    'preferredcodec': 'wav',
+                }],
+            }
+            try:
+              with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+                  ydl.download([url])
+            except:
+                return YoutubeTranscribe(url, retries-1)
+            stream = ffmpeg.input('temp_audio.m4a')
+            stream = ffmpeg.output(stream, 'temp_audio.wav')
+            try:
+                os.remove("temp_audio.m4a")
+            except OSError:
+                pass
+            return Transcribe()
+    else:
+        raise Exception(f"Unable to get video from {url}")
 
 with gr.Blocks() as i:
     video = gr.Video()