Update app.py

app.py

@@ -3,32 +3,38 @@ 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
 
-
-def take_screenshots(ort, start, end, step):
+def selenium(message):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
     wd = webdriver.Chrome(options=options)
-    
-    results = []
-    
-    for i in range(start, end + 1, step):
-        #url = f"https://www.google.com/search?q=vereine+{ort}&start={i}"
-        url =f"https://www.google.com/search?q=vereine+{ort}&udm=1&start={i}"
-
+    text_input = WebDriverWait(driver, 15).until(
+        EC.presence_of_element_located((By.XPATH, '/html/body/main/div/div/div[2]/div[1]/div/div[3]/div/div/div[2]/div[2]/div/div/span/div/div/div/div[1]/textarea'))
+    )
+    text_input.send_keys(prompt)
+    button_send = driver.find_element(By.XPATH, '/html/body/main/div/div/div[2]/div[1]/div/div[3]/div/div/div[2]/div[2]/div/div/span/div/div/div/div[3]/button').click()
+    time.sleep(5)
 
-        
-        #url = f"https://www.google.com/search&start={i}?client=ubuntu-sn&hs=IfI&sca_esv=87428cc8fad3a684&channel=fs&sxsrf=ADLYWILS7FFOYHO5lGqnnazhj3oUCD5brg:1732777961189&q=vereine+coburg&udm=1&fbs=AEQNm0A6bwEop21ehxKWq5cj-cHa02QUie7apaStVTrDAEoT1GgxBwUH1JWFUZ5PObPyzN6ZUsA_o5eBQ6q3XCohkYS3xwvahKNlHLvb7ik7d2FuZj0TOof95P-zQ_8CiAOTe_v8hfAibflUmrJL4FWa_eOcMb26glFo9Yxba4fVczAb7MDRkvPj2OyJbBpnWvmNz0EzxJ0sKYEcdGCvmX8E2KcIAu46WQ&sa=X&ved=2ahUKEwjLsef2vP6JAxXH1QIHHQD_OwQQs6gLegQIFRAB&biw=1558&bih=753&dpr=1"
-        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}")
+    txtelements = driver.find_elements(By.XPATH, '/html/body/main/div/div/div[2]/div[1]/div/div[3]/div/div[1]/div/div[2]/div/div[1]/div[2]/div[2]/div[2]')
+    #text_el =  txtelements.text
+    time.sleep(5)
+    txtcode = ""
+    #with open("datei.txt", "w") as f:
+        #f.write(code)
+    #response_code = {"result": code}
+    #json.dump(response_code, open('code.json', 'w'), indent=4, sort_keys=True)
+    for e in txtelements:
+        txtcode += e.text
+    #response_txt = {"result": txtcode}
+    #print(code)
+    print(txtcode)
     
+    #json.dump(response_txt, open('text.json', 'w'), indent=4, sort_keys=True)
+    #json.dump(response_code, open('code.json', 'w'), indent=4, sort_keys=True)
+    return txtcode
     wd.quit()
     #erg ="\n".join(results)
     #return erg
@@ -52,13 +58,10 @@ iface = gr.Interface(
     fn=take_screenshots,
     inputs=[
         gr.Textbox(label="Ort", value=""),
-        gr.Number(label="Start Index", value=10),
-        gr.Number(label="Anzahl Vereine", value=100),
-        gr.Number(label="Step", value=10),
     ],
     outputs=gr.Textbox(),
-    title="Website Screenshot",
-    description="Take screenshots of websites in specified range and step size for a given location."
+    title="perplexity.ai",
+    description="get answer from perplexity"
 )
 
 iface.launch()