Update app.py

app.py

@@ -148,15 +148,15 @@ async def get_answer_llama3(request: Request ):
     # print("recived ",text)
     proxy=None
 
-    try:
-        proxy = data['proxy']
-    except:
-        pass
-    print('proxy ip ', proxy )
-    currtime= time.time()
-    res= ""
+    # try:
+    #     proxy = data['proxy']
+    # except:
+    #     pass
+    # print('proxy ip ', proxy )
+    # currtime= time.time()
+    # res= ""
     
-    res = do_ML_LLAMA70b(text,0,proxy)
+    res = do_ML_LLAMA70b(text,0,'')
     
     
     dict={"LLAMA":res}
@@ -225,17 +225,21 @@ def do_ML_LLAMA70b_chori(text:str, trycount:int, proxy:str):
                 continue;
             except:
               print('done gen ')
-            time.sleep(1)
-            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
+            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()
@@ -244,7 +248,8 @@ def do_ML_LLAMA70b_chori(text:str, trycount:int, proxy:str):
             
             
 
-    except:
+    except Exception as e:
+        print(e)
         driver.delete_all_cookies()
         driver.quit()
         currtime= time.time()
@@ -261,13 +266,12 @@ def do_ML_LLAMA70b(text:str, trycount:int, proxy:str):
     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}')
-
+    # 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/")
@@ -827,4 +831,76 @@ def do_ML_STARCODE(text:str, trycount:int):
         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"
+            
+
+            
+        
+
+        
+    
+