Update app.py

app.py

@@ -19,6 +19,13 @@ from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
 from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
+wait = WebDriverWait(driver, 15)
+img_el = wait.until(EC.presence_of_element_located((By.TAG_NAME, "img")))
+
 
 # === Globals & Configuration ===
 stop_event = threading.Event()
@@ -68,8 +75,8 @@ def download_image(url, out_path, fail_log, code, suf):
     from webdriver_manager.chrome import ChromeDriverManager
 
     def log(msg):
-    entry = f"{code}{suf}: {msg}"
-    print(entry)  # See logs in console/output
+        entry = f"{code}{suf}: {msg}"
+        print(entry)  # See logs in console/output
     with open(fail_log, "a") as f:
         f.write(entry + "\n")