Add gradio error handling for incorrect video urls

app.py

@@ -40,17 +40,21 @@ def get_video_similarity(video1_url, video2_url):
         with_transcript = False
     else:
         with_transcript = True
-    dataset = RRUMDataset(df, with_transcript=with_transcript, label_col=None,
-                          cross_encoder_model_name_or_path=cross_encoder_model_name_or_path)
-    data_loader = DataLoader(dataset.test_dataset, shuffle=False,
-                             batch_size=1, num_workers=0, pin_memory=False)
-
-    with torch.inference_mode():
-        if with_transcript:
-            pred = model_wt(next(iter(data_loader)))
-        else:
-            pred = model_nt(next(iter(data_loader)))
-    pred = torch.special.expit(pred).squeeze().tolist()
+    try:
+        dataset = RRUMDataset(df, with_transcript=with_transcript, label_col=None,
+                              cross_encoder_model_name_or_path=cross_encoder_model_name_or_path)
+        data_loader = DataLoader(dataset.test_dataset, shuffle=False,
+                                 batch_size=1, num_workers=0, pin_memory=False)
+
+        with torch.inference_mode():
+            if with_transcript:
+                pred = model_wt(next(iter(data_loader)))
+            else:
+                pred = model_nt(next(iter(data_loader)))
+        pred = torch.special.expit(pred).squeeze().tolist()
+    except:
+        raise gr.Error(
+            f'There was error in getting a prediction from the model, please try again.')
     return f'YouTube videos are {pred:.0%} similar'
 
 

utils/helper_funcs.py

@@ -2,6 +2,7 @@ import itertools
 import random
 import requests
 import pandas as pd
+import gradio as gr
 from pytube import YouTube
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
@@ -62,7 +63,10 @@ def update_youtube_embedded_html(video_url, video_position):
 
 
 def get_youtube_video_data(url):
-    video = YouTube(url)
+    try:
+        video = YouTube(url)
+    except:
+        raise gr.Error(f'Could not find YouTube video with the URL {url}')
     channel_id = video.channel_id
     video_title = video.title
     video_description = video.description