Update app.py

app.py

@@ -75,9 +75,9 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')
             
             obj = driver.find_element(By.CLASS_NAME, "main")            
-            messages = driver.find_elements(By.CLASS_NAME, "message")
-            message_css = driver.find_element(By.CSS_SELECTOR, ".message")
-            driver.execute_script(f"arguments[0].style='{css_bub}';",message_css)
+            messages = driver.find_elements(By.CLASS_NAME, "message-bubble")
+            #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