Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import io
 from fastapi import FastAPI, File, UploadFile
 
@@ -164,16 +163,9 @@ async def get_answer(request: Request ):
 
 
 def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
-    
-    # options = ChromeOptions()
-    # options.add_argument('--no-sandbox')
-    # options.add_argument('-headless')
-    # service = Service()
-    # driver = webdriver.Chrome(options= options,service=service)
+ 
     try:
         starttime=time.time()
-        # driver.get("https://talkai.info/chat/")
-        # time.sleep(2+trycount)
         driver= None
         while True:
             try:
@@ -195,34 +187,68 @@ def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
         if temperature>=0 and temperature<=2:
             try:
                 print("setting temperature ",temperature)
-                setting_button = driver.find_element(By.ID, "openSettings")
-                setting_button.click()
+                while True:
+                    currtime= time.time()
+                    if(currtime>starttime+10):
+                        return "Requested Could not be proceed"
+            
+                    try:
+                        setting_button = driver.find_element(By.ID, "openSettings")
+                        setting_button.click()
+                        break
+                    except:
+                        time.sleep(0.2)
+
+                while True:
+                    currtime= time.time()
+                    if(currtime>starttime+10):
+                        return "Requested Could not be proceed"
+                    try:
+                        input_element = driver.find_element(By.CLASS_NAME,"styled-slider")
+                        new_value = temperature
+                        driver.execute_script("arguments[0].value = arguments[1]", input_element, new_value)
+                        break
+                    except:
+                        time.sleep(0.2)
+                while True:
+                    currtime= time.time()
+                    if(currtime>starttime+10):
+                        return "Requested Could not be proceed"
+                    try:
+                        confirm_button = driver.find_element(By.CLASS_NAME, "settingsButtonConfirm")
+                        confirm_button.click()
+                        break
+                    except:
+                        time.sleep(0.2)
+            except:
+                print("could not set temperature")
 
-                input_element = driver.find_element(By.CLASS_NAME,"styled-slider")
-                new_value = temperature
-                driver.execute_script("arguments[0].value = arguments[1]", input_element, new_value)
 
-                confirm_button = driver.find_element(By.CLASS_NAME, "settingsButtonConfirm")
-                confirm_button.click()
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+10):
+                return "Requested Could not be proceed"
+            try:
+                textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
+                textarea.send_keys(text)
+            
+                button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
+                button.click()
+                break
             except:
-                print("could not set temperature")
+                time.sleep(0.2)
 
-        
-        textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
-        textarea.send_keys(text)
-        time.sleep(0.1)
-        button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
-        button.click()
+            
         prev =""
         
         # time.sleep(2)
         while True:
             time.sleep(0.2)
             currtime= time.time()
-            if(currtime>starttime+60):
-                data=data={"id":id,"result":"","status":"Error"}
-                x= requests.post(host,data= data)
-                return 
+            if(currtime>starttime+18.5):
+                
+                return "Requested Could not be proceed"
+                
             value=""
             try:
                 messages = driver.find_elements(By.CLASS_NAME, 'messageContain')
@@ -241,16 +267,13 @@ def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
             driver.quit()
             return value
             
-            data=data={"id":id,"result":value,"status":"Generating --keep refreshing"}
-            x= requests.post(host,data= data)
-            print(x.text)
+            
 
     except:
         print("Error")
         driver.delete_all_cookies()
         if trycount>3:
-            data=data={"id":id,"result":"","status":"Error"}
-            x= requests.post(host,data= data)
+            
             return 
         driver.quit()
         return do_ML(id,text,host,trycount+1)
@@ -291,18 +314,15 @@ async def get_answer(q: Query ):
 
 
 def do_ML2(id:str,text:str,host:str, trycount:int):
-    # options = ChromeOptions()
-    # options.add_argument('--no-sandbox')
-    # options.add_argument('-headless')
-    # service = Service()
-    # driver = webdriver.Chrome(options= options,service=service)
+   
     try:
         starttime=time.time()
-        # driver.get("https://talkai.info/image/")
-
-
+        
         driver= None
         while True:
+            currtime= time.time()
+            if(currtime>starttime+10):
+                return "Requested Could not be proceed"
             try:
                 currtime= time.time()
                 if(currtime>starttime+10):
@@ -312,35 +332,38 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
                 # chatq.pop()[0]
                 t = threading.Thread(target=makeqimg)  
                 t.start()
-                break;
+                break
             except Exception as error:
                 print("Error in popping ", error )
                 t = threading.Thread(target=makeqimg)  
                 t.start()
                 time.sleep(0.5)
             
-        
-
-
-
-
-
-
-        
-        # time.sleep(2+trycount)
-        textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
-        textarea.send_keys(text)
-        time.sleep(0.1)
-        button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
-        button.click()
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+10):
+                return "Requested Could not be proceed"
+            try:
+                textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
+                textarea.send_keys(text)
+                time.sleep(0.1)
+                button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
+                button.click()
+                break
+            except:
+                time.sleep(0.2)
+            
         # time.sleep(2)
         while True:
-            time.sleep(0.2+trycount)
             currtime= time.time()
-            if(currtime>starttime+60):
-                data=data={"id":id,"result":"","status":"Error"}
-                x= requests.post(host,data= data)
-                return 
+            if(currtime>starttime+10):
+                return "Requested Could not be proceed"
+                
+            time.sleep(0.2)
+            currtime= time.time()
+            if(currtime>starttime+18.5):
+                
+                return "Request Could not be proceed"
             try:
                 messages = driver.find_elements(By.XPATH, "//div[@class='messageContain']/p/img")
                 last_message_contain = messages[len(messages)-2]
@@ -349,13 +372,8 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
 
                 driver.delete_all_cookies()
                 driver.quit()
-                return src
-
                 
-                data={"id":id,"result":src,"status":"Generated"}
-                requests.post(host, data=data)
-                driver.delete_all_cookies()
-                driver.quit()
+                return src
                 break
             except:
                 continue
@@ -363,10 +381,9 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
     except:
         print("Error")
         driver.delete_all_cookies()
-        if trycount>3:
-            data=data={"id":id,"result":"","status":"Error"}
-            x= requests.post(host,data= data)
-            return 
+        if trycount>1:
+        
+            return "Request Could not be proceed"
         driver.quit()
         return do_ML2(id,text,host,trycount+1)