Update app.py

app.py

@@ -76,7 +76,8 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             
             obj = driver.find_element(By.CLASS_NAME, "main")            
             messages = driver.find_elements(By.CLASS_NAME, "message")
-            driver.execute_script(f"arguments[0].style='{css_bub}';",messages)
+            message_css = driver.find_element(By.CSS_SELECTOR, ".message")
+            driver.execute_script(f"arguments[0].style='{css_bub}';",message_css)
 
             #print (len(messages))
             #mes_blocks=len(messages)%2