Update app.py

app.py

@@ -34,7 +34,7 @@ def get_concat_v_cut(in1, in2):
     return dst
 
 css_="""--solid_color_primary: #132241;--text_color_primary: #ffffff;--solid_color_bubble: #2c3d7d;--solid_color_tertiary: #223e69;--solid_color_block: #1a1f23;"""
-css_bub="""margin-top:10px!important;"""
+css_bub="""padding: 20px!important;"""
 def run_script(url: str, height: int, width: int, check_b,check_h):
     mes_box=[]
     out_box=[]
@@ -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-bubble")
-            #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")
+            message_css = driver.find_element(By.CSS_SELECTOR, ".message-bubble")
+            driver.execute_script(f"arguments[0].style='{css_bub}';",message_css)
 
             #print (len(messages))
             #mes_blocks=len(messages)%2