Update app.py

app.py

@@ -1,10 +1,12 @@
 import io
 from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, Request, Response, BackgroundTasks
 
 import subprocess
 import os
 import requests
 import random
+# import undetected_chromedriver as uc
 
 import shutil
 import json
@@ -23,6 +25,7 @@ from selenium.webdriver.chrome.service import Service
 import threading
 import random
 import string
+from selenium.webdriver.common.keys import Keys
 import time
 
 
@@ -77,150 +80,463 @@ 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
 
+llama_last_error=0
 
+# LLAMA
 
+chrome_driver_path = '/usr/local/bin/chromedriver-linux64/chromedriver'
 
+@app.post("/llama")
+async def get_answer_llama(request: Request ):
+    data = await request.json()
+    text = data['text']
+    text = text.replace('\n', '  :  ')
+    # print("recived ",text)
+    proxy=None
 
-mycol = None
+    try:
+        proxy = data['proxy']
+    except:
+        pass
+    print('proxy ip ', proxy )
+    currtime= time.time()
+    res= ""
+    # if currtime-llama_last_error<=3600:
+    #     res = do_ML_LLAMA70b(text,0,proxy)
+    # else:    
+    res= do_ML_LLAMA70b_chori(text,0, proxy)
+    dict={"LLAMA":res}
+    return JSONResponse(dict)
 
 
-@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']
+@app.post("/llama2")
+async def get_answer_llama2(request: Request ):
+    data = await request.json()
+    text = data['text']
+    text = text.replace('\n', '  :  ')
+    # print("recived ",text)
+    proxy=None
+
+    try:
+        proxy = data['proxy']
+    except:
+        pass
+    print('proxy ip ', proxy )
+    currtime= time.time()
+    res= ""
+    
+    res = do_ML_LLAMA70b_chori(text,0,proxy)
+    
+    
+    dict={"LLAMA":res}
+
+    return JSONResponse(dict)
+
+@app.post("/llama3")
+async def get_answer_llama3(request: Request ):
+    data = await request.json()
+    text = data['text']
+    text = text.replace('\n', '  :  ')
+    # print("recived ",text)
+    proxy=None
+
+    # try:
+    #     proxy = data['proxy']
+    # except:
+    #     pass
+    # print('proxy ip ', proxy )
+    # currtime= time.time()
+    # res= ""
+    
+    res = do_ML_LLAMA70b(text,0,'')
+    
+    
+    dict={"LLAMA":res}
+
+    return JSONResponse(dict)
+
+
+
+
+def do_ML_LLAMA70b_chori(text:str, trycount:int, proxy:str):
+    starttime=time.time()
+    options = ChromeOptions()
+    global llama_last_error
+    options.add_argument('--no-sandbox')
+    options.add_argument('-headless')
+    options.add_argument("start-maximized")
+    # options.add_argument("--ignore-certificate-errors")
+    service = Service(executable_path=chrome_driver_path)
+    driver = webdriver.Chrome(options= options,service=service)
+    driver.get("https://chansung-llama2-with-gradio-chat.hf.space/")
+    try:
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                llama_last_error= currtime
+                driver.quit()
+                return "Requested Could not be proceed"
+            try:
+                control_panel_span = driver.find_element(By.XPATH,'//span[@class="svelte-s1r2yt" and contains(text(), "Control Panel")]')
+                control_panel_span.click()
+                time.sleep(0.5)
+                input_element = driver.find_elements(By.CSS_SELECTOR,'[data-testid="number-input"]')
+                new_value = "3500"  # Replace this with the value you want to set
+                input_element[3].clear()  # Clear the current value if needed
+                input_element[3].send_keys(new_value)
+                xpath = "//textarea[@data-testid='textbox' and @placeholder='Ask anything']"
+                textarea = driver.find_element(By.XPATH,xpath)
+                textarea.clear()
+                textarea.send_keys(text)
+                textarea.send_keys("\n")
+
+                break
+            except Exception as e:
+                print(e)
+                continue
+
+            
+        prev =""
         
-    #     mycol= mydb['key']
-    # extract()
+        time.sleep(2)
+        while True:
+            time.sleep(0.5)
+            currtime= time.time()
+            
+            if(currtime>starttime+170):
+                driver.delete_all_cookies()
+                driver.quit()
+                llama_last_error= currtime
+                return "Requested Could not be proceed"
+                
+            try:
+                div_element = driver.find_element(By.CSS_SELECTOR,'div.wrap.default.full.svelte-zlszon.generating')
+                print("generating")
+                continue;
+            except:
+              print('done gen ')
+
+            while True:
+                
+                try:
+                    div_element = driver.find_elements(By.CLASS_NAME,'md.svelte-9tftx4.chatbot')
+                    div_element= div_element[len(div_element)-1]
+                    # Extract and print the text content of the div element
+                    div_text = div_element.text
+                    print(div_text)
+                    driver.delete_all_cookies()
+                    driver.quit()
+                    print("Lllama chori")
+                    return div_text
+                except Exception as e:
+                    return "Error Occureed"
+            
+            
+        driver.delete_all_cookies()
+        driver.quit()
+        return " --Error Occurred-- "
+            
+            
 
+    except Exception as e:
+        print(e)
+        driver.delete_all_cookies()
+        driver.quit()
+        currtime= time.time()
+        llama_last_error= currtime
+        return "Error Occurred "
 
 
 
-def extract():
+def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
+    starttime=time.time()
     options = ChromeOptions()
+    global llama_last_error
     options.add_argument('--no-sandbox')
     options.add_argument('-headless')
-    service = Service()
+    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)
-    
-    
-    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.get("https://huggingface.co/chat/")
+    try:
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
+                llama_last_error= currtime
+                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(text)
+                # textarea_element.send_keys(Keys.SHIFT + Keys.ENTER)
+                
+                textarea_element.send_keys('\n')
+
+                break
+            except Exception as e:
+                print(e)
+                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
             driver.delete_all_cookies()
             driver.quit()
+            print("llama huugiing chat")
+            return element_text
+            
+            
+        driver.delete_all_cookies()
+        driver.quit()
+        llama_last_error= time.time()
+        return " --Error Occurred-- "
+            
             
-        except Exception as e: 
-            print('error in extract ',e)
-            pass
-        # time.sleep(60)
 
-    
+    except:
+        driver.delete_all_cookies()
+        driver.quit()
+        llama_last_error= time.time()
+        return "Error Occurred "
 
-from queue import Queue
-chatq = Queue()
-imgq= Queue()
 
 
-def makeqchat():
 
-    while chatq.qsize()<2:
-        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()<2:
-        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)
+def do_ML_LLAMA7b(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://huggingface-projects-llama-2-7b-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:
+                xpath_expression = '//textarea[@data-testid="textbox"]'
+                textarea_element = driver.find_element(By.XPATH, xpath_expression)
+
+                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)
+                
+                span_element = driver.find_element(By.CLASS_NAME, 'svelte-s1r2yt')
+
+                span_element.click()
+                input_element = driver.find_element(By.CSS_SELECTOR, 'input[data-testid="number-input"]')
+                new_value = "2048"  # Replace this with the value you want to set
+                input_element.clear()
+                input_element.send_keys(new_value)
+
+                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"
+                
+            # value=""
+            try:
+                div_element = driver.find_element(By.CLASS_NAME, 'wrap.default.minimal.svelte-zlszon.translucent.hide')
+                element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
         
+                x=(element.text)
+                driver.delete_all_cookies()
+                driver.quit()
+                return x
+            except:
+                continue
 
+            
+        
+        driver.delete_all_cookies()
+        driver.quit()    
+        return " --Error Occurred-- "
+            
+            
 
+    except:
+        driver.delete_all_cookies()
+        driver.quit()
+        return "Error Occurred "
 
 
 
-@app.post("/")
-async def get_answer(request: Request ):
-    data = await request.json()
-    
-    text = data['text']
-    host= ''
 
-    temperature=-1
 
+def do_ML_LLAMA13b(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://huggingface-projects-llama-2-13b-chat--mkxgl.hf.space/") 
     try:
-        temperature= data['temperature']
-        temperature= float(temperature)
-        temperature= round(temperature,1)
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
+                return "Requested Could not be proceed"
+            try:
+                xpath_expression = '//textarea[@data-testid="textbox"]'
+                textarea_element = driver.find_element(By.XPATH, xpath_expression)
+
+                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)
+                
+                span_element = driver.find_element(By.CLASS_NAME, 'svelte-s1r2yt')
+
+                span_element.click()
+                input_element = driver.find_element(By.CSS_SELECTOR, 'input[data-testid="number-input"]')
+                new_value = "2048"  # Replace this with the value you want to set
+                input_element.clear()
+                input_element.send_keys(new_value)
+
+                textarea_element.send_keys('\n')
+
+                break
+            except:
+                continue
+
+            
+        prev =""
         
-    except:
-        print("No temperature")
+        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:
+                queue_element = driver.find_element(By.CLASS_NAME,'progress-text')
+                queue_text = queue_element.text
+                print(queue_text)
+                if "queue" in queue_text:
+                    print("Queue is present")
+                    driver.delete_all_cookies()
+                    driver.quit()
+                    return do_ML_LLAMA7b(text, trycount)
+            except:
+                pass
+                
+            # value=""
+            try:
+                div_element = driver.find_element(By.CLASS_NAME, 'wrap.default.minimal.svelte-zlszon.translucent.hide')
+                element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
         
+                x=(element.text)
+                driver.delete_all_cookies()
+                driver.quit()
+                return x
+            except:
+                continue
 
-    # N = 20
-    # res = ''.join(random.choices(string.ascii_uppercase +
-    #                          string.digits, k=N))
-    # res= res+ str(time.time())
+            
+        driver.delete_all_cookies()
+        driver.quit()
+            
+        return " --Error Occurred-- "
+            
+            
 
-    id= ''
-    
-    
-    # t = threading.Thread(target=do_ML, args=(id,text,host,0))  
-    # t.start()
+    except:
+        driver.delete_all_cookies()
+        driver.quit()
+        return "Error Occurred "
 
-    res= do_ML(id,text,host,0,temperature)
-    
 
-    dict={"ChatGPT":res}
-    # dict= {"id":id}
-    
+
+
+    ##########  MPT
+
+
+@app.post("/mpt")
+async def get_answer_mpt(request: Request ):
+    data = await request.json()
+    text = data['text']
+    print("recived ",text)
+    res= do_ML_MPT(text,0)
     
+    dict={"MPT":res}
+
     return JSONResponse(dict)
 
 
@@ -228,68 +544,52 @@ async def get_answer(request: Request ):
 
 
 
-def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
- 
+def do_ML_MPT(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://mosaicml-mpt-30b-chat.hf.space") 
     try:
-        starttime=time.time()
-        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:
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
+                return "Requested Could not be proceed"
             try:
-                print("setting temperature ",temperature)
-                while True:
-                    currtime= time.time()
-                    if(currtime>starttime+10):
-                        return "Requested Could not be proceed"
-            
-                    try:
-                        setting_button = driver.find_element(By.ID, "openSettings")
-                        setting_button.click()
-                        break
-                    except:
-                        time.sleep(0.2)
+                textarea_xpath = "//textarea[@data-testid='textbox' and @class='scroll-hide svelte-1pie7s6' and @placeholder='Chat Message Box']"
+                textarea_element = driver.find_element(By.XPATH,textarea_xpath)
 
-                while True:
-                    currtime= time.time()
-                    if(currtime>starttime+10):
-                        return "Requested Could not be proceed"
-                    try:
-                        input_element = driver.find_element(By.CLASS_NAME,"styled-slider")
-                        new_value = temperature
-                        driver.execute_script("arguments[0].value = arguments[1]", input_element, new_value)
-                        break
-                    except:
-                        time.sleep(0.2)
-                while True:
-                    currtime= time.time()
-                    if(currtime>starttime+10):
-                        return "Requested Could not be proceed"
-                    try:
-                        confirm_button = driver.find_element(By.CLASS_NAME, "settingsButtonConfirm")
-                        confirm_button.click()
-                        break
-                    except:
-                        time.sleep(0.2)
+                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)
+                break
             except:
-                print("could not set temperature")
-
+                continue
 
         while True:
             currtime= time.time()
-            if(currtime>starttime+10):
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
                 return "Requested Could not be proceed"
             try:
-                textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
-                textarea.send_keys(text)
-            
-                button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
-                button.click()
+                    
+
+                button_id = 'component-9'  # Replace 'comonent-7' with the actual ID
+                button_element = driver.find_element(By.ID,button_id)
+
+# Perform actions on the element (e.g., clicking the button)
+                button_element.click()
                 break
-            except:
+            except Exception as e:
+                print(e)
                 time.sleep(0.2)
 
             
@@ -297,152 +597,386 @@ def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
         
         # time.sleep(2)
         while True:
-            time.sleep(0.2)
-            # currtime= time.time()
-            # if(currtime>starttime+18.5):
+            time.sleep(0.5)
+            currtime= time.time()
+            
+            if(currtime>starttime+120):
+                driver.delete_all_cookies()
+                driver.quit()
                 
-                # return "Requested Could not be proceed"
+                return "Requested Could not be proceed"
                 
-            value=""
+            # value=""
             try:
-                messages = driver.find_elements(By.CLASS_NAME, 'messageContain')
-                last_message_contain = messages[len(messages)-2]
-                value = last_message_contain.text
-                value = value[8:len(value)]
-                print(value)
-                if value=="Please, wait...":
-                    continue
-            except:
+                element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
+        
+                x=(element.text)
+
+                print("From text ",x)
+                if x=="":
+                    raise ValueError(" k")
+                driver.quit()
+                return x
+
+                    
+                    
+
+            except Exception as e:
+                print(e)
                 continue
                 
             
             
-            driver.delete_all_cookies()
-            driver.quit()
-            return value
+        driver.quit()
+        return " --Error Occurred-- "
             
             
 
     except:
         print("Error")
-        driver.delete_all_cookies()
-        if trycount>3:
+        
+        if trycount>1:
             
-            return 
+            driver.delete_all_cookies()
+            driver.quit()
+            return "Error"
         driver.quit()
-        return do_ML(id,text,host,trycount+1)
-        
+        return do_ML_MPT(text,trycount+1)
         
 
+'''Falcon 40 b intruct'''
 
 
-
-@app.post("/image")
-async def get_answer(q: Query ):
+@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)
     
-    text = q.text
-    host= q.host
-
-    # N = 20
-    # res = ''.join(random.choices(string.ascii_uppercase +
-    #                          string.digits, k=N))
-    # res= res+ str(time.time())
+    dict={"FALCON":res}
 
-    id= ''
-    
-    # t = threading.Thread(target=do_ML2, args=(id,text,host,0))  
-    # t.start()
+    return JSONResponse(dict)
 
-    url = do_ML2(id,text,host,0)
-    
-    dict= {"url":url}
 
-    
-    # dict= {"id":id}
-    
-    
-    return JSONResponse(dict)
 
-                
-            
 
 
-def do_ML2(id:str,text:str,host:str, trycount:int):
 
+def do_ML_FALCON(text:str, trycount:int):
     starttime=time.time()
-        
     options = ChromeOptions()
     options.add_argument('--no-sandbox')
     options.add_argument('-headless')
-    service = Service()
+    options.add_argument("start-maximized")
+    service = Service(executable_path=chrome_driver_path)
     driver = webdriver.Chrome(options= options,service=service)
-    driver.get("https://talkai.info/image/")  
+    driver.get("https://huggingfaceh4-falcon-chat.hf.space") 
     try:
-            
+         
         while True:
             currtime= time.time()
-            if(currtime>starttime+5):
-                raise ValueError("h")
-                # return "Requested Could not be proceed"
+            if(currtime>starttime+20):
+                driver.delete_all_cookies()
+                driver.quit()
+                return "Requested Could not be proceed"
             try:
-                textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
-                textarea.send_keys(text)
-                time.sleep(0.1)
-                button = driver.find_element(By.CLASS_NAME, "sectionChatFormButton")
-                button.click()
+                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:
-                time.sleep(0.2)
+                continue
+
             
+        prev =""
+        
         # time.sleep(2)
         while True:
+            time.sleep(0.5)
             currtime= time.time()
-            if(currtime>starttime+40):
-                raise ValueError("j")
+            
+            if(currtime>starttime+170):
+                driver.delete_all_cookies()
+                driver.quit()
+                
                 return "Requested Could not be proceed"
                 
-            time.sleep(0.2)
-            # currtime= time.time()
-            # if(currtime>starttime+18.5):
+            # value=""
             try:
-                messages = driver.find_elements(By.CLASS_NAME,"messageContain")
-                x= messages[len(messages)-2]
-# Extract the text from the 'p' element within the 'div' with class 'messageContain'
-                message_text = x.find_element(By.TAG_NAME,"p").text
-                if message_text=='The request could not be processed':
-                    return "The request could not be processed"
+                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)
-                pass
-
+                continue
                 
-                # return "Request Could not be proceed"
-            try:
-                messages = driver.find_elements(By.XPATH, "//div[@class='messageContain']/p/img")
-                last_message_contain = messages[len(messages)-2]
-                src = last_message_contain.get_attribute("src")
-                print(src)
+            
+            
+        driver.quit()
+        return " --Error Occurred-- "
+            
+            
+
+    except:
+        print("Error")
+        driver.delete_all_cookies()
+        driver.quit()
+        
+        return "Error Occureed"
+        
+
 
+'''Star Code'''
+
+
+@app.post("/starcode")
+async def get_answer_falcon(request: Request ):
+    data = await request.json()
+    text = data['text']
+    print("recived ",text)
+    res= do_ML_STARCODE(text,0)
+    
+    dict={"RESULT":res}
+
+    return JSONResponse(dict)
+
+
+
+
+
+
+def do_ML_STARCODE(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-starchat-playground.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="Enter your message here"]')
                 
-                return src
+                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)
+                time.sleep(0.5)
+                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"
+
+            try:
+                div_element = driver.find_element(By.CLASS_NAME,'svelte-j1gjts.generating')
+                continue
+            except:
+                pass
+            # value=""
+            try:
+                div_element = driver.find_element(By.CSS_SELECTOR,'div[data-testid="bot"]')
+                text_content = div_element.text
+                print(text_content)
+                driver.delete_all_cookies()
+                driver.quit()
+                return text_content
+                
+            except Exception as e:
+                print(e)
+                continue
+                
+            
+            
+        driver.quit()
+        return " --Error Occurred-- "
+            
+            
+
     except:
         print("Error")
-        driver.delete_all_cookies()
-        if trycount>2:
         
-            return "Request Could not be proceed"
+        driver.delete_all_cookies()
+        driver.quit()
+        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")
+async def get_answer_falcon180(request: Request, background_tasks: BackgroundTasks ):
+    data = await request.json()
+    text= data['text']
+    print("recievoed , ", text )
+    return StreamingResponse(do_falcon180(text), media_type="text/plain")
+    
+
+
+import asyncio
+from fastapi.responses import StreamingResponse
+
+async def do_falcon180(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://ysharma-falcon-180b-demo.hf.space/") 
+    try:
+        while True:
+            try:
+                curr= time.time()
+                if curr- starttime>20:
+                    break
+                xpath_expression = "//span[@class='svelte-s1r2yt' and contains(text(), 'Additional Inputs')]"
+                element = driver.find_element(By.XPATH, xpath_expression)
+                element.click()
+                element = driver.find_elements(By.CSS_SELECTOR, 'input[data-testid="number-input"]')
+                element[1].clear()  # Clear any existing value
+                element[1].send_keys('4000')
+                textarea_element = driver.find_element(By.XPATH,'//textarea[@placeholder="Type a message..."]')
+                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 Exception as e:
+                print(e," --1")
+                continue
+        old_text=""
+        while True:
+            # await asyncio.sleep(0.1)
+            curr= time.time()
+            if curr- starttime>180:
+                driver.quit()
+                break
+            done = False
+            try:
+                xpath_expression = "//span[contains(@class, 'chatbot')]"
+                element = driver.find_elements(By.XPATH, xpath_expression)
+                text = element[1].text
+                # print(text)
+                while True:
+                    text = element[1].text
+                    send_text= text[len(old_text):]
+                    old_text= text
+                    yield send_text.encode()
+                    try:
+                        element = driver.find_element(By.CLASS_NAME,"generating")
+                        # print("generating")
+                    except:
+                        # print("generated")
+                        driver.quit()
+                        done= True
+                        break
+                if done:
+                    break
+            except Exception as e:
+                print(e)
+                continue
+    except:
         driver.quit()
-        return do_ML2(id,text,host,trycount+1)
         
         
 
         
-       
-
+    
 
+