Update app.py

app.py

@@ -79,11 +79,29 @@ from pymongo.mongo_client import MongoClient
 
 @app.on_event("startup")
 async def startup_event():
+    x=requests.get('https://open-ai-ping-eight.vercel.app/proxy')
+    print("response ", x.text)
     print("on startup")
     
     
     # t = threading.Thread(target=makeqimg)  
     # t.start()
+proxy =None
+
+
+@app.post("/proxy")
+async def get_proxy(request: Request ):
+    data = await request.json()
+    print('data ', data)
+    try:
+        text = data['url']
+
+        global proxy
+        proxy = text
+        ##done
+        print('proxy recived ',proxy)
+    except Exception as e:
+        print(e)
 
 
 
@@ -97,25 +115,96 @@ async def get_answer_llama(request: Request ):
     data = await request.json()
     text = data['text']
     print("recived ",text)
+    proxy=None
 
-    # try:
-    #     model = data['model']
-    #     if model == '13b':
-    #         print('using 13b')
-    #         res= do_ML_LLAMA13b(text,0)
-    #         dict={"LLAMA":res}
-    #         return JSONResponse(dict)
-    # except:
-    #     print("Using 7b")
-        
+    try:
+        proxy = data['proxy']
+    except:
+        pass
+    print('proxy ip ', proxy )
     
-    res= do_ML_LLAMA7b(text,0)
+    res= do_ML_LLAMA70b(text,0, proxy)
     
     dict={"LLAMA":res}
 
     return JSONResponse(dict)
 
 
+def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
+    starttime=time.time()
+    options = ChromeOptions()
+    options.add_argument('--no-sandbox')
+    options.add_argument('-headless')
+    options.add_argument("start-maximized")
+    options.add_argument("--ignore-certificate-errors")
+    proxy_port = "8080"
+    proxy_ip= proxy
+    print("pxoxy ip ",proxy_ip)
+    proxy_ip = f"{proxy_ip}:{proxy_port}"
+    print("proxy final ", proxy_ip)
+    options.add_argument(f'--proxy-server=http://{proxy_ip}')
+
+    service = Service(executable_path=chrome_driver_path)
+    driver = webdriver.Chrome(options= options,service=service)
+    driver.get("https://huggingface.co/chat/")
+    try:
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                return "Requested Could not be proceed"
+            try:
+                # xpath_expression = '//textarea[@data-testid="tex
+                textarea_element = driver.find_element(By.XPATH, "//textarea[@placeholder='Ask anything']")
+
+
+                for line in text.split('\n'):
+                    textarea_element.send_keys(line)
+                    textarea_element.send_keys(Keys.SHIFT + Keys.ENTER)
+                
+                textarea_element.send_keys('\n')
+
+                break
+            except:
+                continue
+
+            
+        prev =""
+        
+        time.sleep(0.6)
+        while True:
+            time.sleep(0.5)
+            currtime= time.time()
+            
+            if(currtime>starttime+170):
+                driver.delete_all_cookies()
+                driver.quit()
+                
+                return "Requested Could not be proceed"
+                
+            try:
+                class_name = "-ml-1" 
+                svg_elements = driver.find_element(By.CLASS_NAME, class_name)
+                continue
+
+            except:
+                print("not found")
+
+            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
+            return element_text
+            
+            
+            
+        return " --Error Occurred-- "
+            
+            
+
+    except:
+        return "Error Occurred "
+
+
+
 
 
 def do_ML_LLAMA7b(text:str, trycount:int):