Update app.py

app.py

@@ -58,17 +58,24 @@ def run_script(url: str, height: int, width: int):
             print (page.text)
             driver.execute_script("arguments[0].theme = 'gab-black';",page)
             #driver.execute_script("arguments[0].theme = 'gab-black';",page)
+            main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
+            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")
             print (len(messages))
             for i,mes in enumerate(messages):
                 print (mes.text)
-                mes.screenshot(f'{i}-{uid}-tmp.png')
-                mes_box.append(f'{i}-{uid}-tmp.png')
-                if (i%2):
-                    out_box.append(get_concat_v_cut(mes_box[0],mes_box[1]))
-                    mes_box.clear()
+                if not (i%2):
+                    messages[i].screenshot(f'{i}-{uid}-tmp.png')
+                    
+                    
+                    messages[i+1].screenshot(f'{i+1}-{uid}-tmp.png')
+                  
+   
+                    out_box.append(get_concat_v_cut(f'{i}-{uid}-tmp.png'f'{i+1}-{uid}-tmp.png'))
+                    print(f'out_box:: {out_box}')
+                    #mes_box.clear()
                 
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")