Update app.py

app.py

@@ -5,9 +5,16 @@ import urllib.request
 # Replace 'YOUR_API_KEY' with your actual YouTube Data API key
 API_KEY = os.getenv('api_key')
 
+def get_youtube_id(youtube_url):
+    if 'youtube.com' in youtube_url:
+        video_id = youtube_url.split('v=')[-1]
+    elif 'youtu.be' in youtube_url:
+        video_id = youtube_url.split('/')[-1].split('?')[0]
+    return video_id
+
 def check_cc_license(youtube_url):
     # Extract video ID from the URL
-    video_id = youtube_url.split('v=')[-1]
+    video_id = get_youtube_id(youtube_url)
     
     # YouTube Data API URL to get video details
     api_url = f'https://www.googleapis.com/youtube/v3/videos?id={video_id}&part=status&key={API_KEY}'