Update app.py

app.py

@@ -14,7 +14,9 @@ from selenium.webdriver.common.by import By
 driver_type = 'chromedriver'
 driver=False
 
-def get_concat_h_cut(im1, im2):
+def get_concat_h_cut(in1, in2):
+    im1=Image.open(in1)
+    im2=Image.open(in2)
     dst = Image.new('RGB', (im1.width + im2.width,
                             min(im1.height, im2.height)))
     dst.paste(im1, (0, 0))
@@ -33,6 +35,7 @@ def get_concat_v_cut(im1, im2):
 
 def run_script(url: str, height: int, width: int):
     mes_box=[]
+    out_box=[]
     uid=uuid.uuid4()
     is_url=True
     if is_url:
@@ -61,6 +64,10 @@ def run_script(url: str, height: int, width: int):
                 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()
+                
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")            
             #obj = driver.find_element_by_id("in_html")
@@ -84,15 +91,15 @@ def run_script(url: str, height: int, width: int):
             screenshot = obj.screenshot(f'{uid}-tmp.png')
             #screenshot = obj.get_screenshot_as_png()
         except WebDriverException as e:
-            return Image.new('RGB', (1, 1)), f'<center>{e}',mes_box
+            return Image.new('RGB', (1, 1)), f'<center>{e}',out_box
         finally:
             if driver:
                 driver.quit()
 
         #return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
-        return Image.open(f'{uid}-tmp.png'), '<center>operation success.',mes_box
+        return Image.open(f'{uid}-tmp.png'), '<center>operation success.',out_box
     else:
-        return None, '<center>Please enter a valid URL of a website/host.',mes_box
+        return None, '<center>Please enter a valid URL of a website/host.',out_box
 
 with gr.Blocks() as app: