Update app.py

app.py

@@ -4,30 +4,39 @@ from selenium.common.exceptions import WebDriverException
 from selenium.webdriver.common.by import By
 
 
-resp=""
-def take_screenshot(url):
+def take_screenshots(start, end, step):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
-    #options.add_argument('--no-sandbox')
-    #options.add_argument('--disable-dev-shm-usage')
     wd = webdriver.Chrome(options=options)
-    wd.get(url)
     
-    wd.implicitly_wait(3)
-    #element=wd.find_element(By.TAG_NAME, "body")
-    #button_send = driver.find_element(By.ID, "L2AGLb").click()
-    element = wd.find_element(By.TAG_NAME, "body")
-    body_html = element.get_attribute("outerHTML")
-    #return body_html
+    results = []
+    
+    for i in range(start, end + 1, step):
+        url = f"https://www.google.com/search?q=vereine+bamberg&start={i}"
+        wd.get(url)
+        wd.implicitly_wait(3)
+        
+        try:
+            element = wd.find_element(By.TAG_NAME, "body")
+            body_text = element.text
+            results.append(body_text)
+        except Exception as e:
+            print(f"Error fetching data from {url}: {e}")
+    
+    wd.quit()
+    return "\n".join(results)
 
-    return element.text
 
 iface = gr.Interface(
-    fn=take_screenshot,
-    inputs=gr.Textbox(label="Website URL", value="https://spiegel.de"),
+    fn=take_screenshots,
+    inputs=[
+        gr.Number(label="Start Index", value=10),
+        gr.Number(label="End Index", value=100),
+        gr.Number(label="Step Size", value=10)
+    ],
     outputs=gr.Textbox(),
     title="Website Screenshot",
-    description="Take a screenshot of a website."
+    description="Take screenshots of websites in specified range and step size."
 )
 
-iface.launch()
+iface.launch()