Update app.py

app.py

@@ -73,14 +73,80 @@ app.add_middleware(
 # data_frame = pd.read_csv('data.csv')
 
 from selenium.webdriver.common.by import By
+from pymongo.mongo_client import MongoClient
 
 @app.on_event("startup")
 async def startup_event():
     print("on startup")
-    t = threading.Thread(target=makeqchat)  
-    t.start()
-    t = threading.Thread(target=makeqimg)  
-    t.start()
+    
+    
+    # t = threading.Thread(target=makeqimg)  
+    # t.start()
+
+
+
+
+
+mycol = None
+
+
+@app.post("/url")
+async def get_url(request:  Request ):
+    return "k"
+   
+    # data = await request.json()
+    # text = data['url']
+    # mongo_url=text
+    # print("mongo url ",text)
+    # global mycol
+    # if mycol==None:
+    #     myclient = MongoClient(mongo_url)
+
+    #     try:
+    #         myclient.admin.command('ping')
+    #         print("Pinged your deployment. You successfully connected to MongoDB!")
+    #     except Exception as e:
+    #         print(e)
+
+    #     mydb = myclient['open-ai-api-keys']
+        
+    #     mycol= mydb['key']
+    # extract()
+
+
+
+
+def extract():
+    options = ChromeOptions()
+    options.add_argument('--no-sandbox')
+    options.add_argument('-headless')
+    service = Service()
+    driver = webdriver.Chrome(options= options,service=service)
+    
+    
+    global mycol
+    # global driver
+    
+    if True:
+        # time.sleep(60)
+        try:
+            driver.get("https://talkai.info/chat/") 
+            element = driver.find_element(By.CSS_SELECTOR,".chat")
+            api_key = element.get_attribute("data-api-key")
+            dict={"key":"open-ai"}
+            mycol.delete_one(dict)
+            dict={"key":"open-ai","value":api_key}
+            mycol.insert_one(dict)
+            print(api_key)
+            driver.delete_all_cookies()
+            driver.quit()
+            
+        except Exception as e: 
+            print('error in extract ',e)
+            pass
+        # time.sleep(60)
+
+    
 
 from queue import Queue
 chatq = Queue()
@@ -166,24 +232,12 @@ def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
  
     try:
         starttime=time.time()
-        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)
-
+        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/")  
         if temperature>=0 and temperature<=2:
             try:
                 print("setting temperature ",temperature)
@@ -318,27 +372,12 @@ def do_ML2(id:str,text:str,host:str, trycount:int):
     try:
         starttime=time.time()
         
-        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):
-                    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)
-            
+        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/")      
         while True:
             currtime= time.time()
             if(currtime>starttime+10):