Update app.py

app.py

@@ -45,7 +45,7 @@ def run_script(url: str, height: int, width: int, check_b):
         options.add_argument('--headless')
         options.add_argument('--no-sandbox')
         options.add_argument('--disable-dev-shm-usage')
-
+        mes='<center>operation success'
         try:
             driver = webdriver.Chrome(options=options)
             driver.get(url)
@@ -64,16 +64,18 @@ def run_script(url: str, height: int, width: int, check_b):
             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()
+            try:
+                for ea in check_b:
+                    ea = int(ea)
+                    ea = (ea*2)-2
+                    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()
+            except Exception:
+                mes="<center>Selected blocks returned an error"
             '''
             for i,mes in enumerate(messages):
                 print (mes.text)
@@ -114,7 +116,7 @@ def run_script(url: str, height: int, width: int, check_b):
                 driver.quit()
 
         #return [Image.open(BytesIO(screenshot)), 'operation success.',cookie_jar,html]
-        return Image.open(f'{uid}-tmp.png'), '<center>operation success.',out_box
+        return Image.open(f'{uid}-tmp.png'), mes,out_box
     else:
         return None, '<center>Please enter a valid URL of a website/host.',out_box