jaredjoss commited on
Commit
3a83ffe
·
1 Parent(s): 0e59933

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -157,8 +157,15 @@ def quantizeEffectiveness(url):
157
  # 10. Vote on video effectiveness
158
  result, confidence = vote(test_point, _test)
159
  return result, confidence
 
 
 
 
160
 
161
  def greet(url):
 
 
 
162
  result, confidence = quantizeEffectiveness(url)
163
 
164
  return f"The video (ID: {getID(url)}) is {result} with a confidence of {round(confidence*100,2)}%"
 
157
  # 10. Vote on video effectiveness
158
  result, confidence = vote(test_point, _test)
159
  return result, confidence
160
+
161
+ def is_valid_youtube_url(text):
162
+ youtube_regex = re.compile(r"^(https?://)?(www\.)?(youtube\.com/watch\?v=|youtu\.be/|youtube\.com/embed/|youtube\.com/v/|youtube\.com/user/[^/]+/u/)?([^/&?=\s]{11})$")
163
+ return bool(youtube_regex.match(text))
164
 
165
  def greet(url):
166
+ if not is_valid_youtube_url(text):
167
+ return "Please input a valid YouTube URL"
168
+
169
  result, confidence = quantizeEffectiveness(url)
170
 
171
  return f"The video (ID: {getID(url)}) is {result} with a confidence of {round(confidence*100,2)}%"