Update app.py

app.py

@@ -1,35 +1,35 @@
-import gradio as gr
 from selenium import webdriver
-from selenium.common.exceptions import WebDriverException
-from PIL import Image
-from io import BytesIO
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
 
-def take_screenshot(url):
-    options = webdriver.ChromeOptions()
-    options.add_argument('--headless')
-    options.add_argument('--no-sandbox')
-    options.add_argument('--disable-dev-shm-usage')
+# Set Chrome to run in headless mode to save memory
+chrome_options = Options()
+chrome_options.add_argument("--headless")  # Headless mode
+chrome_options.add_argument("--no-sandbox")  # Necessary for running in some environments
+chrome_options.add_argument("--disable-dev-shm-usage")  # Overcome limited resource problems
 
-    try:
-        wd = webdriver.Chrome(options=options)
-        wd.set_window_size(1080, 720)  # Adjust the window size here
-        wd.get(url)
-        wd.implicitly_wait(30)
-        screenshot = wd.get_screenshot_as_png()
-    except WebDriverException as e:
-        return Image.new('RGB', (1, 1))
-    finally:
-        if wd:
-            wd.quit()
+# Initialize the Chrome WebDriver
+driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
 
-    return Image.open(BytesIO(screenshot))
+# Go to the provided URL
+url = "https://genyoutube.online/url=https://www.youtube.com/watch?v=-2RAq5o5pwc&list=RD-2RAq5o5pwc&start_radio=1"
+driver.get(url)
 
-iface = gr.Interface(
-    fn=take_screenshot,
-    inputs=gr.Textbox(label="Website URL", value="https://kargaranamir.github.io"),
-    outputs=gr.Image(type="pil", label="Screenshot", height=360, width=540),  # Adjust the image size here
-    title="Website Screenshot",
-    description="Take a screenshot of a website."
-)
+# Print the page title to verify success
+print(driver.title)
 
-iface.launch()
+# Use WebDriverWait to wait for the element to be present
+try:
+    my_tab_content = WebDriverWait(driver, 10).until(
+        EC.presence_of_element_located((By.ID, "myTabContent"))
+    )
+    print(my_tab_content.text)  # Print the text content
+except Exception as e:
+    print("Error locating the element:", e)
+
+# Close the browser session
+driver.quit()