Update app.py

app.py

@@ -99,200 +99,43 @@ chrome_driver_path = '/usr/local/bin/chromedriver-linux64/chromedriver'
 async def get_answer_llama(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= ""
-    # if currtime-llama_last_error<=3600:
-    #     res = do_ML_LLAMA70b(text,0,proxy)
-    # else:    
-    res= do_ML_LLAMA70b_chori(text,0, proxy)
+    res= do_ML_LLAMA70b(text,0)
     dict={"LLAMA":res}
     return JSONResponse(dict)
 
 
-@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 =""
-        
-        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 do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
+def do_ML_LLAMA70b(text:str, trycount:int):
     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")
-    # 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/")
+    driver.get("https://ysharma-explore-llamav2-with-tgi.hf.space/")
     try:
          
         while True:
             currtime= time.time()
-            if(currtime>starttime+20):
+            if(currtime>starttime+30):
                 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)
+                span_element = driver.find_element(By.CLASS_NAME, 'svelte-s1r2yt')
+                span_element.click()
+                input_element = driver.find_elements(By.CSS_SELECTOR, 'input[data-testid="number-input"]')
+                new_value = "2000"  
+                input_element[1].clear()
+                input_element[1].send_keys(new_value)
+                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(text)
+                    textarea_element.send_keys(Keys.SHIFT + Keys.ENTER)
                 
                 textarea_element.send_keys('\n')
 
@@ -309,25 +152,30 @@ def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
             time.sleep(0.5)
             currtime= time.time()
             
-            if(currtime>starttime+170):
+            if(currtime>starttime+178):
                 driver.delete_all_cookies()
                 driver.quit()
                 
-                return "Requested Could not be proceed"
+                return prev
                 
+            element_selector = '.wrap.default.minimal.svelte-119qaqt.translucent.generating'
+
+            
+# print(x)
             try:
-                class_name = "-ml-1" 
-                svg_elements = driver.find_element(By.CLASS_NAME, class_name)
+                element = driver.find_element(By.CSS_SELECTOR,element_selector)
+                print("generating")
+                element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
+                prev=(element.text)
                 continue
-
             except:
-                print("not found")
+                print("Element is not present.")
 
-            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()
+            element = driver.find_element(By.XPATH,'//div[@data-testid="bot" and contains(@class, "message bot")]')
+            element_text= element.text
+                
             driver.quit()
-            print("llama huugiing chat")
+            
             return element_text
             
             
@@ -348,181 +196,6 @@ def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
 
 
 
-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 "
-
-
-
-
-
-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:
-         
-        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"
-
-            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
-
-            
-        driver.delete_all_cookies()
-        driver.quit()
-            
-        return " --Error Occurred-- "
-            
-            
-
-    except:
-        driver.delete_all_cookies()
-        driver.quit()
-        return "Error Occurred "
-
-
 
 
     ##########  MPT
@@ -938,7 +611,7 @@ async def do_falcon180(text):
                 textarea_element.send_keys('\n')
                 break
             except Exception as e:
-                print(e," --1")
+                print(e)
                 continue
         old_text=""
         while True: