Update app.py

app.py

@@ -46,7 +46,7 @@ service = Service()
 
 driver = webdriver.Chrome(options= options,service=service)
 
-driver.get("https://yuntian-deng-chatgpt.hf.space/")
+# driver.get("https://yuntian-deng-chatgpt.hf.space/")
 
 
 class Query(BaseModel):
@@ -116,21 +116,15 @@ async def get_answer(q: Query ):
 def do_ML(id:str,text:str,host:str, trycount:int):
     try:
         starttime=time.time()
-        driver.get("https://yuntian-deng-chatgpt.hf.space/")
+        driver.get("https://talkai.info/chat/")
         time.sleep(3+trycount)
-        button = driver.find_element(By.ID,"component-23")
-        button.click()
-        time.sleep(1+trycount)
-        alert = driver.switch_to.alert
-        alert.accept()
-        print("alert accepted")
-        time.sleep(1+trycount)
-        id_box = driver.find_element(By.XPATH,'//textarea[@data-testid="textbox"]')
-        id_box.send_keys(text)
+        textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
+        textarea.send_keys(text)
         time.sleep(1+trycount)
-        run = driver.find_element(By.ID,"component-9")
-        run.click()
+        button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
+        button.click()
         prev =""
+        
         time.sleep(2)
         while True:
             time.sleep(2+trycount)
@@ -140,11 +134,18 @@ def do_ML(id:str,text:str,host:str, trycount:int):
                 x= requests.post(host,data= data)
                 return 
                 
-            div = driver.find_element(By.XPATH,'//div[@class="message bot svelte-a99nd8 latest"]')
-            value = div.text
+            messages = driver.find_elements(By.CLASS_NAME, 'messageContain')
+            last_message_contain = messages[len(messages)-2]
+            value = last_message_contain.text
+            value = value[8:len(value)]
+
+            if value=="Please, wait...":
+                continue
+                
             if prev!="":
                 if value==prev:
                     data={"id":id,"result":value,"status":"Generated"}
+                    driver.delete_all_cookies()
                     requests.post(host, data=data)
                     break
             prev= value
@@ -154,6 +155,7 @@ def do_ML(id:str,text:str,host:str, trycount:int):
             print(x.text)
 
     except:
+        driver.delete_all_cookies()
         if trycount>3:
             data=data={"id":id,"result":"","status":"Error"}
             x= requests.post(host,data= data)