Update app.py

app.py

@@ -59,7 +59,7 @@ def run_script(url: str, height: int, width: int):
             print (len(messages))
             for i,mes in enumerate(messages):
                 print (mes.text)
-                obj.screenshot(f'{i}-{uid}-tmp.png')
+                mes.screenshot(f'{i}-{uid}-tmp.png')
                 mes_box.append(f'{i}-{uid}-tmp.png')
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")