Update app.py

app.py

@@ -31,6 +31,11 @@ def download_video(url):
     )
     print("Downloaded")
     return local_file
+
+import json
+import re
+from pytube import YouTube
+
 def validate_youtube(url):
     try:
         yt = YouTube(url)
@@ -38,21 +43,33 @@ def validate_youtube(url):
         print(f"Invalid URL or network issue: {e}")
         return True
 
-    # Fetch video details using multiple fallback methods
     try:
-        # Try getting length from `yt.length`
+        # First try: Direct access (most reliable)
         video_length = yt.length
-        if video_length is None:
-            raise ValueError("yt.length returned None")
-
-    except:
-        # Fallback: Extract video details from `yt.watch_html`
-        match = re.search(r'"lengthSeconds":"(\d+)"', yt.watch_html)
-        if match:
-            video_length = int(match.group(1))
-        else:
-            print("Error: Could not retrieve video length.")
-            return True
+
+    except Exception as e:
+        try:
+            # Fallback 1:  Try extracting from embed HTML (more robust than watch_html)
+            embed_html = yt.embed_html
+            match = re.search(r'"lengthSeconds":"(\d+)"', embed_html)
+            if match:
+                video_length = int(match.group(1))
+            else:
+                raise ValueError("Could not find length in embed HTML")  # Explicitly raise to trigger next fallback
+
+        except Exception as e:
+            try:
+                # Fallback 2:  Try watch_html (less reliable, but a backup)
+                match = re.search(r'"lengthSeconds":"(\d+)"', yt.watch_html)
+                if match:
+                    video_length = int(match.group(1))
+                else:
+                    raise ValueError("Could not find length in watch HTML") # Explicitly raise to trigger final return
+
+            except Exception as e:
+                print(f"Error: Could not retrieve video length: {e}") # Print the actual exception
+                return True  # Return True indicating failure
+
 
     # Check video length limit
     if video_length > 600:
@@ -61,6 +78,8 @@ def validate_youtube(url):
     else:
         print("Your video is less than 10 minutes")
         return False
+
+
 def validate_url(url):
     import validators
     if not validators.url(url):