Update app.py

app.py

@@ -88,23 +88,7 @@ async def startup_event():
     # t.start()
 proxy =None
 
-
-@app.post("/proxy")
-async def get_proxy(request: Request ):
-    data = await request.json()
-    print('data ', data)
-    try:
-        text = data['url']
-
-        global proxy
-        proxy = text
-        ##done
-        print('proxy recived ',proxy)
-    except Exception as e:
-        print(e)
-
-
-
+llama_last_error=0
 
 # LLAMA
 
@@ -114,23 +98,113 @@ 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)
+    text = text.replace('\n', '  :  ')
+    # print("recived ",text)
     proxy=None
 
-    try:
-        proxy = data['proxy']
-    except:
-        pass
-    print('proxy ip ', proxy )
-    
-    res= do_ML_LLAMA70b(text,0, proxy)
+    # try:
+    #     proxy = data['proxy']
+    # except:
+    #     pass
+    # print('proxy ip ', proxy )
+    currtime= time.time()
+    res= ""
+    if currtime-llama_last_error<=3600:
+        res = do_ML_LLAMA7b(text,0)
+    else:    
+        res= do_ML_LLAMA70b_chori(text,0, proxy)
     
     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_element(By.CSS_SELECTOR,'[data-testid="number-input"]')
+                new_value = "4000"  # Replace this with the value you want to set
+                input_element.clear()  # Clear the current value if needed
+                input_element.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(0.6)
+        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 ')
+
+            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
+            print(div_text)
+
+            driver.delete_all_cookies()
+            driver.quit()
+            return div_text
+            
+            
+        driver.delete_all_cookies()
+        driver.quit()
+        return " --Error Occurred-- "
+            
+            
+
+    except:
+        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):
     starttime=time.time()
     options = ChromeOptions()