Update app.py

app.py

@@ -153,6 +153,8 @@ def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 # xpath_expression = '//textarea[@data-testid="tex
@@ -194,15 +196,20 @@ def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
 
             element = driver.find_element(By.CLASS_NAME, "prose")  # Find the element by its class name
             element_text = element.text  # Extract the text from the element
+            driver.delete_all_cookies()
+            driver.quit()
             return element_text
             
             
-            
+        driver.delete_all_cookies()
+        driver.quit()
         return " --Error Occurred-- "
             
             
 
     except:
+        driver.delete_all_cookies()
+        driver.quit()
         return "Error Occurred "
 
 
@@ -222,7 +229,9 @@ def do_ML_LLAMA7b(text:str, trycount:int):
          
         while True:
             currtime= time.time()
-            if(currtime>starttime+20):
+            if(currtime>starttime+20): 
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 xpath_expression = '//textarea[@data-testid="textbox"]'
@@ -267,18 +276,23 @@ def do_ML_LLAMA7b(text:str, trycount:int):
                 element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
         
                 x=(element.text)
+                driver.delete_all_cookies()
+                driver.quit()
                 return x
             except:
                 continue
 
             
-            
-            
+        
+        driver.delete_all_cookies()
+        driver.quit()    
         return " --Error Occurred-- "
             
             
 
     except:
+        driver.delete_all_cookies()
+        driver.quit()
         return "Error Occurred "
 
 
@@ -299,6 +313,8 @@ def do_ML_LLAMA13b(text:str, trycount:int):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 xpath_expression = '//textarea[@data-testid="textbox"]'
@@ -343,6 +359,8 @@ def do_ML_LLAMA13b(text:str, trycount:int):
                 print(queue_text)
                 if "queue" in queue_text:
                     print("Queue is present")
+                    driver.delete_all_cookies()
+                    driver.quit()
                     return do_ML_LLAMA7b(text, trycount)
             except:
                 pass
@@ -353,18 +371,23 @@ def do_ML_LLAMA13b(text:str, trycount:int):
                 element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
         
                 x=(element.text)
+                driver.delete_all_cookies()
+                driver.quit()
                 return x
             except:
                 continue
 
             
-            
+        driver.delete_all_cookies()
+        driver.quit()
             
         return " --Error Occurred-- "
             
             
 
     except:
+        driver.delete_all_cookies()
+        driver.quit()
         return "Error Occurred "
 
 
@@ -403,6 +426,8 @@ def do_ML_MPT(text:str, trycount:int):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 textarea_xpath = "//textarea[@data-testid='textbox' and @class='scroll-hide svelte-1pie7s6' and @placeholder='Chat Message Box']"
@@ -419,6 +444,8 @@ def do_ML_MPT(text:str, trycount:int):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                     
@@ -478,7 +505,9 @@ def do_ML_MPT(text:str, trycount:int):
         
         if trycount>1:
             
-            return 
+            driver.delete_all_cookies()
+            driver.quit()
+            return "Error"
         driver.quit()
         return do_ML_MPT(text,trycount+1)
         
@@ -516,6 +545,8 @@ def do_ML_FALCON(text:str, trycount:int):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 textarea_element = driver.find_element(By.CSS_SELECTOR,'textarea[placeholder="Hello Falcon !!"]')
@@ -568,6 +599,8 @@ def do_ML_FALCON(text:str, trycount:int):
 
     except:
         print("Error")
+        driver.delete_all_cookies()
+        driver.quit()
         
         return "Error Occureed"
         
@@ -606,6 +639,8 @@ def do_ML_STARCODE(text:str, trycount:int):
         while True:
             currtime= time.time()
             if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
                 textarea_element = driver.find_element(By.CSS_SELECTOR,'textarea[placeholder="Enter your message here"]')
@@ -644,6 +679,8 @@ def do_ML_STARCODE(text:str, trycount:int):
                 div_element = driver.find_element(By.CSS_SELECTOR,'div[data-testid="bot"]')
                 text_content = div_element.text
                 print(text_content)
+                driver.delete_all_cookies()
+                driver.quit()
                 return text_content
                 
             except Exception as e:
@@ -660,5 +697,7 @@ def do_ML_STARCODE(text:str, trycount:int):
     except:
         print("Error")
         
+        driver.delete_all_cookies()
+        driver.quit()
         return "Error Occureed"