Update app.py

app.py

@@ -30,12 +30,12 @@ def get_concat_v_cut(in1, in2):
     dst = Image.new(
         'RGB', (min(im1.width, im2.width), im1.height + im2.height))
     dst.paste(im1, (0, 0))
-    dst.paste(im2, (0, im1.height+10))
+    dst.paste(im2, (0, im1.height))
     return dst
 
 
 
-def run_script(url: str, height: int, width: int):
+def run_script(url: str, height: int, width: int, check_b):
     mes_box=[]
     out_box=[]
     uid=uuid.uuid4()
@@ -56,23 +56,35 @@ def run_script(url: str, height: int, width: int):
             page = driver.find_element(By.TAG_NAME, "html")
             print (dir(page))
             print (page.text)
-            driver.execute_script("arguments[0].theme = '';",page)
-            #driver.execute_script("arguments[0].theme = 'gab-black';",page)
+            driver.execute_script("arguments[0].style.background = 'blue';",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))
+            mes_blocks=len(messages)%2
+            
+            for ea in check_b:
+                ea = int(ea)
+                ea = (ea*2)-1
+                print (ea)
+                messages[ea].screenshot(f'{ea}-{uid}-tmp.png')
+                messages[ea+1].screenshot(f'{ea+1}-{uid}-tmp.png')
+                out_box.append(get_concat_v_cut(f'{ea}-{uid}-tmp.png',f'{ea+1}-{uid}-tmp.png'))
+                print(f'out_box:: {out_box}')
+                #mes_box.clear()
+            '''
             for i,mes in enumerate(messages):
                 print (mes.text)
                 if not (i%2):
                     messages[i].screenshot(f'{i}-{uid}-tmp.png')
                     messages[i+1].screenshot(f'{i+1}-{uid}-tmp.png')
+                    mes_box.append(())
                     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")            
             #obj = driver.find_element_by_id("in_html")
@@ -115,6 +127,7 @@ with gr.Blocks() as app:
             
         with gr.Column():
             with gr.Row():
+                check_b=gr.CheckboxGroup(label="Chatblocks", choices=[1,2,3,4,5,6,7,8,9,10])
                 height=gr.Number(label="Height", value=4096)
                 width=gr.Number(label="Width",value=800)
     message=gr.HTML('<center>Enter URL')
@@ -123,5 +136,5 @@ with gr.Blocks() as app:
             outgal=gr.Gallery()
         outim = gr.Image()
 
-    btn.click(run_script,[inp,height,width],[outim,message,outgal])
+    btn.click(run_script,[inp,height,width,check_b],[outim,message,outgal])
 app.launch()