Update app.py

app.py

@@ -36,7 +36,7 @@ def run_script(url: str, height: int, width: int):
             #driver.execute_script("arguments[0].theme = 'gab-black';",page)
             
             obj = driver.find_element(By.CLASS_NAME, "main")            
-            messages = driver.find_element(By.CLASS_NAME, "message")
+            messages = driver.find_elements(By.CLASS_NAME, "message")
             print (messages)
             print (messages.text)
             #obj = driver.find_element(By.ID, "app-container")