Update app.py

app.py

@@ -640,8 +640,179 @@ async def do_falcon180(text):
                 continue
     except:
         driver.quit()
+
+
         
+
+@app.post("/wizlm70b")
+async def get_answer_falcon180(request: Request):
+    data = await request.json()
+    text= data['text']
+    print("recievoed , ", text )
+    return do_wizard70b(text)
+    
+
+def do_wizard70b(text):
+    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://chat.lmsys.org/")
+    while True:
+        curr= time.time()
+        if curr- starttime>10:
+            driver.delete_all_cookies()
+            driver.quit()
+            return "Error-- 1"
+        try:
+            button = driver.find_element(By.XPATH, "//button[contains(@class, 'svelte-kqij2n') and text()='Single Model']")
+            button.click()
+            break
+        except Exception as e:
+            print(e)
+            time.sleep(0.5)
+    # return driver.page_source
+
+    # while True:
+    #     curr= time.time()
+    #     if curr- starttime>50:
+    #         driver.delete_all_cookies()
+    #         driver.quit()
+    #         return "Error-- 2"
+    #     try:
+    #         input_element = driver.find_elements(By.CLASS_NAME,'border-none')
+    # # Send text to the <input> element
+    #         input_element[2].clear()
+    #         input_element[2].send_keys('wizardlm-70b\n')
+    #         span_element = driver.find_elements(By.XPATH,'//span[text()="Parameters"]')
+    #         span_element[2].click()
+    #         input_element = driver.find_elements(By.XPATH ,'//input[@data-testid="number-input"]')
+    #         input_element[8].clear()  # Clear any existing value
+    #         input_element[8].send_keys('1024')
+    #         break
+    #     except Exception as e:
+    #         print(e)
+    #         time.sleep(0.5)
+    time.sleep(1)
+    while True:
+        curr= time.time()
+        if curr- starttime>60:
+            driver.delete_all_cookies()
+            driver.quit()
+            return "Error-- 3"
+        try:
+            textarea = driver.find_elements(By.XPATH,'//textarea[@data-testid="textbox"]')  # You can also use other locators like by_id, by_name, by_class_name, etc.
+            textarea[2].clear()
+            for line in text.split('\n'):
+                textarea[2].send_keys(line)
+            textarea[2].send_keys('\n')
+        except Exception as e:
+            print(e)
+            time.sleep(0.5)
+         
+    time.sleep(2)
+    while True:
+        curr= time.time()
+        if curr- starttime>178:
+            driver.delete_all_cookies()
+            driver.quit()
+            return "Error-- 3 -- timeout"
+        try:
+            element = driver.find_element(By.CSS_SELECTOR, '.generating')
+            print("geberating")
+            time.sleep(0.5)
+        except:
+            print("Element with 'generating' keyword in class is not present.")
+            div_element = driver.find_element(By.XPATH,'//div[@data-testid="bot"]')
+            extracted_text = div_element.text
+            driver.delete_all_cookies()
+            driver.quit()
+            return extracted_text
+
+            
+
+
+
+#####PALM 2
+
+@app.post("/palm2")
+async def get_answer_llama(request: Request ):
+    data = await request.json()
+    text = data['text']
+    # print("recived ",text)
+    res= ""
+    res= do_ML_PALM(text,0)
+    dict={"BOT":res}
+    return JSONResponse(dict)
+
+
+def do_ML_PALM(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://chansung-palm-with-gradio-chat.hf.space')
+    try:
+        while True:
+            curr = time.time()
+            if curr-starttime>178:
+                driver.delete_all_cookies()
+                driver.quit()
+                return 'Error -1'
+
+            try:
+                xpath = "//textarea[@data-testid='textbox' and @placeholder='Ask anything']"
+                textarea = driver.find_element(By.XPATH,xpath)
+                textarea.clear()
+                for line in text.split('\n'):
+                    textarea.send_keys(line)
+                textarea.send_keys("\n")
+                break
+            except:
+                continue
+        time.sleep(1.5)
+        while True:
+            if curr-starttime>178:
+                driver.delete_all_cookies()
+                driver.quit()
+                return 'Error -2'
+            try:
+                div_element = driver.find_element(By.CSS_SELECTOR,'div.wrap.default.full.svelte-zlszon.generating')
+                continue
+            except:
+                print('fail')
+            try:
+                div_element = driver.find_elements(By.CLASS_NAME,'md.svelte-9tftx4.chatbot')
+                div_element= div_element[1]
+                # Extract and print the text content of the div element
+                div_text = div_element.text
+                if div_text=="":
+                    continue
+                driver.delete_all_cookies()
+                driver.quit()
+                return div_text
+            except:
+                time.sleep(0.5)
+                continue
+            
+
+    except Exception as e:
+        print(e)
+        driver.delete_all_cookies()
+        driver.quit()
+        return "Error 3"
+
         
+            
+
+
+