Update app.py

app.py

@@ -73,15 +73,52 @@ app.add_middleware(
 # db = firestore.client()
 # data_frame = pd.read_csv('data.csv')
 
-
+from selenium.webdriver.common.by import By
 
 @app.on_event("startup")
 async def startup_event():
-   print("on startup")
+    print("on startup")
+    t = threading.Thread(target=makeqchat)  
+    t.start()
+    t = threading.Thread(target=makeqimg)  
+    t.start()
+
+from queue import Queue
+chatq = Queue()
+imgq= Queue()
+
+
+def makeqchat():
+
+    while chatq.qsize()<10:
+        print("appending in chat queue")
+        options = ChromeOptions()
+        options.add_argument('--no-sandbox')
+        options.add_argument('-headless')
+        service = Service()
+        driver = webdriver.Chrome(options= options,service=service)
+        driver.get("https://talkai.info/chat/")  
+        chatq.put(driver)
+        
+        
+
+def makeqimg():
+
+    while imgq.qsize()<10:
+        print("appending in img queue")
+        options = ChromeOptions()
+        options.add_argument('--no-sandbox')
+        options.add_argument('-headless')
+        service = Service()
+        driver = webdriver.Chrome(options= options,service=service)
+        driver.get("https://talkai.info/image/")  
+        imgq.put(driver)
+        
+        
+
+
 
 
-   
-from selenium.webdriver.common.by import By
 
 @app.post("/")
 async def get_answer(q: Query ):
@@ -89,18 +126,22 @@ async def get_answer(q: Query ):
     text = q.text
     host= q.host
 
-    N = 20
-    res = ''.join(random.choices(string.ascii_uppercase +
-                             string.digits, k=N))
-    res= res+ str(time.time())
+    # N = 20
+    # res = ''.join(random.choices(string.ascii_uppercase +
+    #                          string.digits, k=N))
+    # res= res+ str(time.time())
 
-    id= res
+    id= ''
     
-    t = threading.Thread(target=do_ML, args=(id,text,host,0))  
-    t.start()
+    
+    # t = threading.Thread(target=do_ML, args=(id,text,host,0))  
+    # t.start()
 
+    res= do_ML(id,text,host,0)
     
-    dict= {"id":id}
+
+    dict={"ChatGPT":res}
+    # dict= {"id":id}
     
     
     return JSONResponse(dict)
@@ -111,15 +152,34 @@ async def get_answer(q: Query ):
 
 
 def do_ML(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)
+    # 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.get("https://talkai.info/chat/")
+        # time.sleep(2+trycount)
+        driver= None
+        while True:
+            try:
+                currtime= time.time()
+                if(currtime>starttime+10):
+                    return "Error"
+                
+                driver= chatq.get()
+                # chatq.pop()[0]
+                t = threading.Thread(target=makeqchat)  
+                t.start()
+                break;
+            except Exception as error:
+                print("Error in popping ", error )
+                t = threading.Thread(target=makeqchat)  
+                t.start()
+                time.sleep(0.5)
+            
+        
         textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
         textarea.send_keys(text)
         time.sleep(0.1)
@@ -129,9 +189,9 @@ def do_ML(id:str,text:str,host:str, trycount:int):
         
         # time.sleep(2)
         while True:
-            time.sleep(2+trycount)
+            time.sleep(0.2)
             currtime= time.time()
-            if(currtime>starttime+600):
+            if(currtime>starttime+60):
                 data=data={"id":id,"result":"","status":"Error"}
                 x= requests.post(host,data= data)
                 return 
@@ -144,16 +204,20 @@ def do_ML(id:str,text:str,host:str, trycount:int):
             if value=="Please, wait...":
                 continue
                 
-            if prev!="":
-                if value==prev:
-                    data={"id":id,"result":value,"status":"Generated"}
+            # if prev!="":
+            #     if value==prev:
+            #         data={"id":id,"result":value,"status":"Generated"}
                     
-                    driver.delete_all_cookies()
-                    driver.quit()
-                    requests.post(host, data=data)
-                    break
-            prev= value
-
+            #         driver.delete_all_cookies()
+            #         driver.quit()
+            #         requests.post(host, data=data)
+            #         break
+            # prev= value
+            
+            driver.delete_all_cookies()
+            driver.quit()
+            return value
+            
             data=data={"id":id,"result":value,"status":"Generating --keep refreshing"}
             x= requests.post(host,data= data)
             print(x.text)
@@ -166,7 +230,7 @@ def do_ML(id:str,text:str,host:str, trycount:int):
             x= requests.post(host,data= data)
             return 
         driver.quit()
-        do_ML(id,text,host,trycount+1)
+        return do_ML(id,text,host,trycount+1)
         
         
 
@@ -179,18 +243,22 @@ async def get_answer(q: Query ):
     text = q.text
     host= q.host
 
-    N = 20
-    res = ''.join(random.choices(string.ascii_uppercase +
-                             string.digits, k=N))
-    res= res+ str(time.time())
+    # N = 20
+    # res = ''.join(random.choices(string.ascii_uppercase +
+    #                          string.digits, k=N))
+    # res= res+ str(time.time())
 
-    id= res
+    id= ''
     
-    t = threading.Thread(target=do_ML2, args=(id,text,host,0))  
-    t.start()
+    # t = threading.Thread(target=do_ML2, args=(id,text,host,0))  
+    # t.start()
 
+    url = do_ML2(id,text,host,0)
     
-    dict= {"id":id}
+    dict= {"url":url}
+
+    
+    # dict= {"id":id}
     
     
     return JSONResponse(dict)
@@ -200,15 +268,43 @@ 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)
+    # 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/")
-        time.sleep(2+trycount)
+        # driver.get("https://talkai.info/image/")
+
+
+        driver= None
+        while True:
+            try:
+                currtime= time.time()
+                if(currtime>starttime+10):
+                    return "Error"
+                
+                driver= imgq.get()
+                # chatq.pop()[0]
+                t = threading.Thread(target=makeqimg)  
+                t.start()
+                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)
@@ -216,9 +312,9 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
         button.click()
         # time.sleep(2)
         while True:
-            time.sleep(1+trycount)
+            time.sleep(0.2+trycount)
             currtime= time.time()
-            if(currtime>starttime+600):
+            if(currtime>starttime+60):
                 data=data={"id":id,"result":"","status":"Error"}
                 x= requests.post(host,data= data)
                 return 
@@ -227,6 +323,12 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
                 last_message_contain = messages[len(messages)-2]
                 src = last_message_contain.get_attribute("src")
                 print(src)
+
+                driver.delete_all_cookies()
+                driver.quit()
+                return src
+
+                
                 data={"id":id,"result":src,"status":"Generated"}
                 requests.post(host, data=data)
                 driver.delete_all_cookies()
@@ -243,7 +345,7 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
             x= requests.post(host,data= data)
             return 
         driver.quit()
-        do_ML2(id,text,host,trycount+1)
+        return do_ML2(id,text,host,trycount+1)