Update app.py

app.py

@@ -90,7 +90,7 @@ imgq= Queue()
 
 def makeqchat():
 
-    while chatq.qsize()<10:
+    while chatq.qsize()<2:
         print("appending in chat queue")
         options = ChromeOptions()
         options.add_argument('--no-sandbox')
@@ -104,7 +104,7 @@ def makeqchat():
 
 def makeqimg():
 
-    while imgq.qsize()<10:
+    while imgq.qsize()<2:
         print("appending in img queue")
         options = ChromeOptions()
         options.add_argument('--no-sandbox')
@@ -121,10 +121,22 @@ def makeqimg():
 
 
 @app.post("/")
-async def get_answer(q: Query ):
+async def get_answer(request: Request ):
+    data = await request.json()
     
-    text = q.text
-    host= q.host
+    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 +
@@ -137,7 +149,7 @@ async def get_answer(q: Query ):
     # t = threading.Thread(target=do_ML, args=(id,text,host,0))  
     # t.start()
 
-    res= do_ML(id,text,host,0)
+    res= do_ML(id,text,host,0,temperature)
     
 
     dict={"ChatGPT":res}
@@ -151,7 +163,8 @@ async def get_answer(q: Query ):
 
 
 
-def do_ML(id:str,text:str,host:str, trycount:int):
+def do_ML(id:str,text:str,host:str, trycount:int,temperature:float):
+    
     # options = ChromeOptions()
     # options.add_argument('--no-sandbox')
     # options.add_argument('-headless')
@@ -178,7 +191,22 @@ def do_ML(id:str,text:str,host:str, trycount:int):
                 t = threading.Thread(target=makeqchat)  
                 t.start()
                 time.sleep(0.5)
-            
+
+        if temperature>=0 and temperature<=2:
+            try:
+                print("setting temperature ",temperature)
+                setting_button = driver.find_element(By.ID, "openSettings")
+                setting_button.click()
+
+                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)
+
+                confirm_button = driver.find_element(By.CLASS_NAME, "settingsButtonConfirm")
+                confirm_button.click()
+            except:
+                print("could not set temperature")
+
         
         textarea = driver.find_element(By.CSS_SELECTOR, "textarea")
         textarea.send_keys(text)
@@ -195,24 +223,19 @@ def do_ML(id:str,text:str,host:str, trycount:int):
                 data=data={"id":id,"result":"","status":"Error"}
                 x= requests.post(host,data= data)
                 return 
-                
-            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...":
+            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
                 
-            # if prev!="":
-            #     if value==prev:
-            #         data={"id":id,"result":value,"status":"Generated"}
-                    
-            #         driver.delete_all_cookies()
-            #         driver.quit()
-            #         requests.post(host, data=data)
-            #         break
-            # prev= value
+            
             
             driver.delete_all_cookies()
             driver.quit()