Update app.py

app.py

@@ -132,7 +132,7 @@ def run_script(chat: list, height: int, width: int, check_b,check_h,wait_time):
             #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, "panel")
-            messages = driver.find_elements(By.CLASS_NAME, "message")
+            messages = driver.find_elements(By.CLASS_NAME, "message-row")
             for i,ea in enumerate(check_b):
                 try:    
                     ea = int(ea)