Update app.py

app.py

@@ -90,6 +90,7 @@ async def startup_event():
 
 # LLAMA
 
+chrome_driver_path = '/usr/local/bin/chromedriver'
 
 @app.post("/llama")
 async def get_answer_llama(request: Request ):
@@ -105,17 +106,15 @@ async def get_answer_llama(request: Request ):
 
 
 
-
-
 def do_ML_LLAMA(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()
+    service = Service(executable_path=chrome_driver_path)
     driver = webdriver.Chrome(options= options,service=service)
-    driver.get("https://ysharma-explore-llamav2-with-tgi.hf.space/") 
+    driver.get("https://huggingface-projects-llama-2-7b-chat.hf.space") 
     try:
          
         while True:
@@ -123,42 +122,37 @@ def do_ML_LLAMA(text:str, trycount:int):
             if(currtime>starttime+20):
                 return "Requested Could not be proceed"
             try:
-                textarea_xpath = "//textarea[@data-testid='textbox' and @class='scroll-hide svelte-1kcgrqr' and @dir='ltr' and @placeholder='Type a message...']"
-                textarea_element = driver.find_element(By.XPATH,textarea_xpath)
+                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
-        while True:
-            currtime= time.time()
-            if(currtime>starttime+20):
-                return "Requested Could not be proceed"
-                    
-            try:
-                button_id = 'component-10' 
-                button_element = driver.find_element(By.ID,button_id)
-
-                button_element.click()
-                # print("sent hii")
-                break
-            except Exception as e:
-                print(e)
-                time.sleep(0.2)
 
             
         prev =""
         
-        # time.sleep(2)
+        time.sleep(0.6)
         while True:
             time.sleep(0.5)
             currtime= time.time()
             
-            if(currtime>starttime+120):
+            if(currtime>starttime+170):
                 driver.delete_all_cookies()
                 driver.quit()
                 
@@ -166,35 +160,14 @@ def do_ML_LLAMA(text:str, trycount:int):
                 
             # 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")]')
-                print(element.text)
-# Extract the text from the div element
-                text = element.text
-                # print("text from llama",text)
-
-                
-                
-                if  prev!="":
-
-                    try:
-                        element_translucent = driver.find_element(By.XPATH,'//div[contains(@class, "translucent") and contains(@class, "generating")]')
-                        continue
-                    except:
-                        pass
-                        
-                    driver.delete_all_cookies()
-                    driver.quit()
-                    return text
-                prev= text
+        
+                x=(element.text)
+                return x
+            except:
                 continue
-                
-                    
-                    
 
-            except Exception as e:
-                print(e)
-                continue
-                
             
             
             
@@ -203,16 +176,11 @@ def do_ML_LLAMA(text:str, trycount:int):
             
 
     except:
-        print("Error")
-        
-        if trycount>1:
-            
-            return 
-        driver.quit()
-        return do_ML_LLAMA(text,trycount+1)
+        return "Error Occurred "
         
 
 
+
     ##########  MPT