Update app.py

app.py

@@ -482,4 +482,91 @@ def do_ML_FALCON(text:str, trycount:int):
         
 
 
+'''Star Code'''
 
+
+@app.post("/starcode")
+async def get_answer_falcon(request: Request ):
+    data = await request.json()
+    text = data['text']
+    print("recived ",text)
+    res= do_ML_STARCODE(text,0)
+    
+    dict={"RESULT":res}
+
+    return JSONResponse(dict)
+
+
+
+
+
+
+def do_ML_STARCODE(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://huggingfaceh4-starchat-playground.hf.space/") 
+    try:
+         
+        while True:
+            currtime= time.time()
+            if(currtime>starttime+20):
+                return "Requested Could not be proceed"
+            try:
+                textarea_element = driver.find_element(By.CSS_SELECTOR,'textarea[placeholder="Enter your message here"]')
+                
+                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)
+                textarea_element.send_keys('\n')
+                break
+            except:
+                continue
+
+            
+        prev =""
+        
+        # time.sleep(2)
+        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:
+                div_element = driver.find_element(By.CLASS_NAME,'svelte-j1gjts.generating')
+                continue
+            except:
+                pass
+            # value=""
+            try:
+                div_element = driver.find_element(By.CSS_SELECTOR,'div[data-testid="bot"]')
+                text_content = div_element.text
+                print(text_content)
+                return text_content
+                
+            except Exception as e:
+                print(e)
+                continue
+                
+            
+            
+        driver.quit()
+        return " --Error Occurred-- "
+            
+            
+
+    except:
+        print("Error")
+        
+        return "Error Occureed"
+