youtube mobile link

app.py

@@ -17,39 +17,11 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
 
 def extract_video_id(url):
-    if "youtu.be" in url:  # Mobile YouTube URL
-        return get_redirected_url_video_id(url)
-    else:  # Desktop YouTube URL
+    # Handle both mobile and desktop URLs
+    if "youtu.be" in url:
+        video_id = re.search(r"youtu\.be/([^&]+)", url).group(1)
+    else:
         video_id = re.search(r"v=([^&]+)", url).group(1)
-        return video_id
-
-def get_redirected_url_video_id(mobile_url):
-    # Set up Chrome options to run headless (without opening a visible browser window)
-    chrome_options = Options()
-    chrome_options.add_argument("--headless")
-    chrome_options.add_argument("--disable-gpu")
-
-    # Specify the path to the ChromeDriver executable
-    chrome_driver_path = 'chromedriver'  # Update this path
-
-    # Create a new Chrome session
-    service = Service(chrome_driver_path)
-    driver = webdriver.Chrome(service=service, options=chrome_options)
-
-    # Open the mobile YouTube link
-    driver.get(mobile_url)
-
-    # Wait for a few seconds to ensure the page has loaded
-    time.sleep(1)
-
-    # Get the current URL (which should be the full YouTube URL)
-    current_url = driver.current_url
-
-    # Close the browser
-    driver.quit()
-
-    # Extract the video ID from the current URL
-    video_id = re.search(r"v=([^&]+)", current_url).group(1)
     return video_id
 
 def get_youtube_comments(video_id):