Update app.py

app.py

@@ -132,8 +132,27 @@ class SuperSmartAgent:
         ###################
         
         def contains_youtube_url(text):
+            # This regex is good for identifying the presence of a YouTube URL
             return re.search(r"(https?://)?(www\.)?(youtube\.com|youtu\.be)/", text) is not None
         
+        def extract_youtube_url(text):
+            # A more focused regex to *extract* the full YouTube URL
+            # This pattern attempts to match common YouTube URL formats and capture the entire URL
+            youtube_regex = (
+                r'(https?://)?(www\.)?'
+                '(?:youtube\.com|youtu\.be)/'  # Non-capturing group for domain
+                '(?:watch\?v=|embed/|v/|.+\?v=)?' # Non-capturing group for path
+                '([a-zA-Z0-9_-]{11})' # Capturing group for the 11-character video ID
+                '(?:[&?].*)?' # Optional: matches any query parameters after the video ID
+            )
+            match = re.search(youtube_regex, text)
+            if match:
+                # Reconstruct the standard YouTube watch URL from the video ID
+                video_id = match.group(1) # The 11-character video ID
+                return f"https://www.youtube.com/watch?v={video_id}"
+            return None
+
+        
         def check_special_tools(state):
             q = state["question"].lower()