Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 import requests
 import random
 
+
 import shutil
 import json
 # from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
@@ -86,267 +87,6 @@ async def startup_event():
 
 
 
-@app.post("/")
-async def get_answer(request: Request ):
-    data = await request.json()
-    
-    text = data['text']
-    host= ''
-
-    temperature=-1
-
-    try:
-        temperature= data['temperature']
-        temperature= float(temperature)
-        temperature= round(temperature,1)
-        
-    except:
-        print("No temperature")
-        
-
-    # N = 20
-    # res = ''.join(random.choices(string.ascii_uppercase +
-    #                          string.digits, k=N))
-    # res= res+ str(time.time())
-
-    id= ''
-    
-    
-    # t = threading.Thread(target=do_ML, args=(id,text,host,0))  
-    # t.start()
-    temperature= temperature*2
-
-    res= do_ML(id,text,host,0,temperature)
-    
-
-    dict={"ChatGPT":res}
-    # dict= {"id":id}
-    
-    
-    return JSONResponse(dict)
-
-
-
-
-
-
-def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
-    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/") 
-    try:
-         
-        if temperature>=0 and temperature<=2:
-            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)
-
-                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)
-            except:
-                print("could not set temperature")
-
-
-        while True:
-            currtime= time.time()
-            if(currtime>starttime+10):
-                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()
-                break
-            except:
-                time.sleep(0.2)
-
-            
-        prev =""
-        
-        # time.sleep(2)
-        while True:
-            time.sleep(0.5)
-            # currtime= time.time()
-            
-            if(currtime>starttime+120):
-                driver.delete_all_cookies()
-                driver.quit()
-                
-                return "Requested Could not be proceed"
-                
-            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:
-                continue
-                
-            
-            
-            driver.delete_all_cookies()
-            driver.quit()
-            return value
-            
-            
-
-    except:
-        print("Error")
-        driver.delete_all_cookies()
-        if trycount>1:
-            
-            return 
-        driver.quit()
-        return do_ML(id,text,host,trycount+1)
-        
-        
-
-
-
-
-@app.post("/image")
-async def get_answer(q: Query ):
-    
-    text = q.text
-    host= q.host
-
-    # N = 20
-    # res = ''.join(random.choices(string.ascii_uppercase +
-    #                          string.digits, k=N))
-    # res= res+ str(time.time())
-
-    id= ''
-    
-    # t = threading.Thread(target=do_ML2, args=(id,text,host,0))  
-    # t.start()
-
-    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):
-
-    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/image/")  
-    try:
-            
-        while True:
-            currtime= time.time()
-            if(currtime>starttime+5):
-                raise ValueError("h")
-                # 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()
-                break
-            except:
-                time.sleep(0.2)
-            
-        # time.sleep(2)
-        while True:
-            currtime= time.time()
-            if(currtime>starttime+40):
-                raise ValueError("j")
-                return "Requested Could not be proceed"
-                
-            time.sleep(0.2)
-            # currtime= time.time()
-            # if(currtime>starttime+18.5):
-            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"
-            except Exception as e:
-                print(e)
-                pass
-
-                
-                # 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.delete_all_cookies()
-                driver.quit()
-                
-                return src
-                break
-            except:
-                continue
-        
-    except:
-        print("Error")
-        driver.delete_all_cookies()
-        if trycount>2:
-        
-            return "Request Could not be proceed"
-        driver.quit()
-        return do_ML2(id,text,host,trycount+1)
-        
-        
-
 # LLAMA
 
 
@@ -371,6 +111,7 @@ def do_ML_LLAMA(text:str, trycount:int):
     options = ChromeOptions()
     options.add_argument('--no-sandbox')
     options.add_argument('-headless')
+    options.add_argument("start-maximized")
     service = Service()
     driver = webdriver.Chrome(options= options,service=service)
     driver.get("https://ysharma-explore-llamav2-with-tgi.hf.space/") 
@@ -386,10 +127,11 @@ def do_ML_LLAMA(text:str, trycount:int):
 
                 textarea_element.send_keys(text)
             
-                button_id = 'component-7' 
+                button_id = 'component-8' 
                 button_element = driver.find_element(By.ID,button_id)
 
                 button_element.click()
+                # print("sent hii")
                 break
             except Exception as e:
                 print(e)
@@ -411,11 +153,10 @@ def do_ML_LLAMA(text:str, trycount:int):
                 
             # value=""
             try:
-                div_id = 'message'  # Replace 'your_div_id' with the actual ID of the div element
-                div_element = driver.find_elements(By.CLASS_NAME,div_id)
-                div_element= div_element[-1]
+                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 = div_element.text
+                text = element.text
                 print("text from llama",text)
                 if  prev!="":
 
@@ -425,7 +166,7 @@ def do_ML_LLAMA(text:str, trycount:int):
                     except:
                         pass
                         
-
+                    driver.delete_all_cookies()
                     driver.quit()
                     return text
                 prev= text