Update app.py

app.py

@@ -316,95 +316,6 @@ def do_ML_MPT(text:str, trycount:int):
 '''Falcon 40 b intruct'''
 
 
-@app.post("/falcon")
-async def get_answer_falcon(request: Request ):
-    data = await request.json()
-    text = data['text']
-    print("recived ",text)
-    res= do_ML_FALCON(text,0)
-    
-    dict={"FALCON":res}
-
-    return JSONResponse(dict)
-
-
-
-
-
-
-def do_ML_FALCON(text:str, trycount:int):
-    starttime=time.time()
-    options = ChromeOptions()
-    options.add_argument('--no-sandbox')
-    options.add_argument('-headless')
-    options.add_argument("start-maximized")
-    service = Service(executable_path=chrome_driver_path)
-    driver = webdriver.Chrome(options= options,service=service)
-    driver.get("https://huggingfaceh4-falcon-chat.hf.space") 
-    try:
-         
-        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 !!"]')
-
-                for line in text.split('\n'):
-                    textarea_element.send_keys(line)
-    # Simulate pressing the 'Shift + Enter' keys to add a newline without triggering submit
-                    textarea_element.send_keys(Keys.SHIFT + Keys.ENTER)
-                textarea_element.send_keys('\n')
-                break
-            except:
-                continue
-
-            
-        prev =""
-        
-        # time.sleep(2)
-        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"
-                
-            # value=""
-            try:
-                div_element = driver.find_element(By.CSS_SELECTOR,'div[data-testid="bot"]')
-                div_text = div_element.text
-                print(div_text)
-                if prev==div_text and prev !="":
-                    driver.quit()
-                    return div_text
-                
-                prev= div_text
-                
-                    
-            except Exception as e:
-                print(e)
-                continue
-                
-            
-            
-        driver.quit()
-        return " --Error Occurred-- "
-            
-            
-
-    except:
-        print("Error")
-        driver.delete_all_cookies()
-        driver.quit()
-        
-        return "Error Occureed"
-        
 
 
 '''Star Code'''
@@ -503,68 +414,6 @@ def do_ML_STARCODE(text:str, trycount:int):
         return "Error Occureed"
 
 
-@app.post("/chatgpt")
-async def get_answer_llama3(request: Request ):
-    data = await request.json()
-    msges= data['messages']
-    print("recievoed , ", msges )
-    return do_chat_gpt(msges)
-    
-
-
-
-
-def do_chat_gpt(msges):
-    starttime=time.time()
-    options = ChromeOptions()
-    options.add_argument('--no-sandbox')
-    options.add_argument('-headless')
-    options.add_argument("start-maximized")
-    service = Service(executable_path=chrome_driver_path)
-    driver = webdriver.Chrome(options= options,service=service)
-    driver.get("https://openskyml-pigeon-chat.hf.space/") 
-    base=""
-    for message in msges:
-            base += "%s: %s  \n" % (message["role"], message["content"])
-    base += "assistant:"
-    print("base ",base)
-    try:
-        while True:
-            curr= time.time()
-            if curr- starttime>20:
-                driver.quit()
-                return "Error -1"
-            try:
-                textarea_element = driver.find_element(By.XPATH,'//textarea[@placeholder="Type a message..."]')
-                for line in base.split('\n'):
-                    textarea_element.send_keys(line)
-                    textarea_element.send_keys(Keys.SHIFT + Keys.ENTER)
-                time.sleep(0.5)
-                textarea_element.send_keys('\n')
-                break
-            except Exception as e:
-                print(e)
-                time.sleep(0.5)
-                continue
-        while True:
-            curr= time.time()
-            if curr- starttime>175:
-                driver.quit()
-                return "Error -1"
-            try:
-                div_element = driver.find_element(By.CSS_SELECTOR,'div[data-testid="bot"]')
-                text = div_element.text
-                print(text)
-                if text=="":
-                    raise ValueError("Empty text")
-                driver.quit()
-                return text
-            except Exception as e:
-                print(e)
-                time.sleep(0.5)
-    except:
-        driver.quit()
-        return "Error"
             
 
 @app.post("/falcon180b")
@@ -587,9 +436,9 @@ async def do_falcon180(text):
     options.add_argument("start-maximized")
     service = Service(executable_path=chrome_driver_path)
     driver = webdriver.Chrome(options= options,service=service)
-    driver.get("https://tiiuae-falcon-180b-demo.hf.space/")
+    # driver.get("https://tiiuae-falcon-180b-demo.hf.space")
     # driver.get("https://ysharma-falcon-180b-demo.hf.space/")
-    # driver.get("https://wffcyrus-falcon-180b-demo.hf.space")
+    driver.get("https://lunarflu-falcon-180b-demo-duplicate.hf.space")
     try:
         while True:
             try: