Update app.py

app.py

@@ -22,12 +22,19 @@ mobile_emulation = {
 
 def get_video_stream(target_url):
     chrome_options = Options()
-    chrome_options.add_argument('--headless')
+    
+    # Core Headless Settings
+    chrome_options.add_argument('--headless=new')  # Updated headless syntax
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
     chrome_options.add_argument('--disable-gpu')
     chrome_options.add_argument('--window-size=360,640')
     
+    # Network & DNS fixes for Docker/Hugging Face
+    chrome_options.add_argument('--dns-prefetch-disable')
+    chrome_options.add_argument('--disable-features=NetworkService')
+    chrome_options.add_argument('--disable-features=VizDisplayCompositor')
+    
     # Critical for Docker: Write user data to a writable temporary location
     user_data_dir = f'/tmp/chrome-user-data-{int(time.time())}'
     chrome_options.add_argument(f'--user-data-dir={user_data_dir}')
@@ -40,23 +47,26 @@ def get_video_stream(target_url):
     driver = webdriver.Chrome(service=service, options=chrome_options)
     
     try:
-        # Go directly to the target URL (No login/cookies)
+        # Go directly to the target URL
+        print(f"Navigating to: {target_url}")
         driver.get(target_url)
 
-        # Wait for video element (Timeout increased to 15s in case of slow loading)
+        # Wait for video element
         wait = WebDriverWait(driver, 15)
         video_element = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "video")))
         
         video_src = video_element.get_attribute('src')
         if not video_src:
             raise Exception("No video src found")
+            
+        print(f"Found video SRC: {video_src}")
 
         # Download video content into memory
         response = requests.get(video_src)
         if response.status_code == 200:
             return io.BytesIO(response.content)
         else:
-            raise Exception("Failed to download video file")
+            raise Exception(f"Failed to download video file: {response.status_code}")
 
     finally:
         driver.quit()
@@ -81,6 +91,7 @@ def download_route(link):
             download_name=f'instagram_video_{int(time.time())}.mp4'
         )
     except Exception as e:
+        print(f"Error: {e}")
         return jsonify({"error": str(e)}), 500
 
 @app.route('/')