Update app.py

app.py

@@ -3,23 +3,24 @@ from selenium import webdriver
 from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.by import By
 from gradio_client import Client
-#from selenium.webdriver.support.wait import WebDriverWait
-#from selenium.webdriver.support import expected_conditions as EC
-#from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.support.wait import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.chrome.options import Options
 
 
-def selenium(message):   
+def take_screenshot(url):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
-    wd = webdriver.Chrome(options=options)
+
     try:
-        #url = 'https://www.google.com/search?q=zugverbindung+bad+kissingen+euerdorf'
-        url = 'https://www.spiegel.de'
+        wd = webdriver.Chrome(options=options)
+        url = 'https://www.google.com/search?q=zugverbindung+bad+kissingen+euerdorf'
+        #url = 'https://www.spiegel.de'
         #<ol class="AmbQnf">
         wd.get(url)
-        wd.implicitly_wait(10)
-        element = driver.find_element(By.TAG_NAME, "body")
-        time.sleep(5)
+        wd.implicitly_wait(5)
+        element = driver.find_element(By.ID, "exp0")
+        time.sleep(2)
         return element.text