Update student_functions.py

student_functions.py

@@ -13,7 +13,14 @@ from urllib.parse import urlparse, parse_qs
 from pypdf import PdfReader
 from ai71 import AI71
 import os
-
+chrome_options = webdriver.ChromeOptions()
+chrome_options.add_argument("--headless")
+chrome_options.add_argument("--disable-gpu")
+chrome_options.add_argument("--disable-extensions")
+chrome_options.add_argument("--no-sandbox")
+chrome_options.add_argument("--disable-dev-shm-usage")
+    
+    driver = webdriver.Chrome(options=chrome_options)
 AI71_API_KEY = "api71-api-652e5c6c-8edf-41d0-9c34-28522b07bef9"
 
 
@@ -197,27 +204,21 @@ def spk_msg(user_input, memory):
 
 
 def get_first_youtube_video_link(query):
-    chrome_options = webdriver.ChromeOptions()
-    chrome_options.add_argument("--headless")
-    chrome_options.add_argument("--disable-gpu")
-    chrome_options.add_argument("--disable-extensions")
-    chrome_options.add_argument("--no-sandbox")
-    chrome_options.add_argument("--disable-dev-shm-usage")
-    driver = webdriver.Chrome(options=chrome_options)
+    url = f'https://www.youtube.com/results?search_query={query}'
+
+    # Navigate to the URL
+    driver.get(url)
+
+    # Find the first video link
     try:
-        driver.get('https://www.youtube.com')
-        search_box = driver.find_element(By.NAME, 'search_query')
-        search_box.send_keys(query)
-        search_box.send_keys(Keys.RETURN)
-        WebDriverWait(driver, 10).until(
-            EC.presence_of_element_located((By.CSS_SELECTOR, 'a#video-title')))
-        first_video = driver.find_element(By.CSS_SELECTOR, 'a#video-title')
-        first_video_link = first_video.get_attribute('href')
-        video_title = first_video.get_attribute('title')
-        return first_video_link, video_title
+        # Find the first video element
+        first_video_element = driver.find_element(By.XPATH, '//a[@id="video-title"]')
+        video_link = first_video_element.get_attribute('href')
+        print(video_link)
     finally:
+        # Close the WebDriver
         driver.quit()
-    return
+
 
 
 def content_translate(text):