Update app.py

app.py

@@ -15,38 +15,42 @@ import socket
 
 app = Flask(__name__)
 
-# Mobile emulation settings
 mobile_emulation = {
     "deviceMetrics": {"width": 360, "height": 640, "pixelRatio": 3.0},
     "userAgent": "Mozilla/5.0 (Linux; Android 6.0; Nexus 5 Build/MRA58N) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Mobile Safari/537.36"
 }
 
+def get_instagram_ip():
+    try:
+        # Resolve instagram.com to IP using Python (which usually works better in containers)
+        ip = socket.gethostbyname("www.instagram.com")
+        print(f"DEBUG: Resolved www.instagram.com to {ip}")
+        return ip
+    except Exception as e:
+        print(f"DEBUG: DNS Resolution failed: {e}")
+        return None
+
 def get_video_stream(target_url):
     chrome_options = Options()
     
-    # --- CRITICAL DOCKER FLAGS ---
+    # 1. Resolve IP manually
+    ig_ip = get_instagram_ip()
+    if ig_ip:
+        # Force Chrome to map www.instagram.com to the resolved IP
+        # This bypasses Chrome's internal DNS resolver
+        chrome_options.add_argument(f'--host-resolver-rules=MAP www.instagram.com {ig_ip}, MAP .instagram.com {ig_ip}')
+    
+    # 2. Standard Docker Flags
     chrome_options.add_argument('--headless=new')
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
     chrome_options.add_argument('--disable-gpu')
-    chrome_options.add_argument('--disable-software-rasterizer')
-    chrome_options.add_argument('--disable-extensions')
     
-    # --- NETWORK STABILITY FLAGS ---
-    # This specific combo helps resolving DNS in strict containers
+    # 3. Network Flags
     chrome_options.add_argument('--dns-prefetch-disable')
     chrome_options.add_argument('--disable-features=NetworkService')
-    chrome_options.add_argument('--force-device-scale-factor=1')
     
-    # --- CERTIFICATE & SECURITY BYPASS ---
-    chrome_options.add_argument('--ignore-certificate-errors')
-    chrome_options.add_argument('--allow-running-insecure-content')
-    chrome_options.add_argument('--disable-web-security')
-    
-    # --- DEBUGGING ---
-    chrome_options.add_argument('--remote-debugging-port=9222')
-
-    # User data dir
+    # 4. User Data
     user_data_dir = f'/tmp/chrome-user-data-{int(time.time())}'
     chrome_options.add_argument(f'--user-data-dir={user_data_dir}')
     
@@ -56,33 +60,9 @@ def get_video_stream(target_url):
     driver = webdriver.Chrome(service=service, options=chrome_options)
     
     try:
-        # Debug 1: Check Python DNS
-        try:
-            ip = socket.gethostbyname("instagram.com")
-            print(f"DEBUG: Python resolved instagram.com to {ip}")
-        except Exception as e:
-            print(f"DEBUG: Python DNS Failed: {e}")
-
-        # Debug 2: Check Python HTTP
-        try:
-            r = requests.get("https://www.instagram.com", timeout=5)
-            print(f"DEBUG: Python HTTP Connection Status: {r.status_code}")
-        except Exception as e:
-            print(f"DEBUG: Python HTTP Failed: {e}")
-
         print(f"Navigating to: {target_url}")
-        
-        # Set page load timeout
-        driver.set_page_load_timeout(30)
-        
-        try:
-            driver.get(target_url)
-        except Exception as e:
-            # If navigation fails, try one more time with simple refresh strategy
-            print(f"Navigation error: {e}. Retrying...")
-            driver.get(target_url)
+        driver.get(target_url)
 
-        # Wait for video
         wait = WebDriverWait(driver, 20)
         video_element = wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "video")))
         
@@ -90,23 +70,21 @@ def get_video_stream(target_url):
         print(f"Found Video SRC: {video_src}")
         
         if not video_src:
-            raise Exception("Video element found but no src attribute")
+            raise Exception("No video src found")
 
-        # Use Session for download to mimic the browser context
+        # Use Python requests to download (it handles DNS better)
         session = requests.Session()
         headers = {
             "User-Agent": mobile_emulation["userAgent"],
             "Referer": "https://www.instagram.com/",
-            "Range": "bytes=0-" # Ask for whole file
         }
         
-        # Sometimes src is a blob: or similar, but for IG Reels mobile it's usually a URL
         response = session.get(video_src, headers=headers, stream=True)
         
         if response.status_code in [200, 206]:
             return io.BytesIO(response.content)
         else:
-            raise Exception(f"Failed to download. Status: {response.status_code}")
+            raise Exception(f"Download failed: {response.status_code}")
 
     finally:
         try:
@@ -122,7 +100,7 @@ def download_route(link):
         link = f'https://{link}'
         
     try:
-        print(f"Received Request: {link}")
+        print(f"Processing: {link}")
         video_stream = get_video_stream(link)
         
         return send_file(
@@ -132,12 +110,12 @@ def download_route(link):
             download_name=f'insta_{int(time.time())}.mp4'
         )
     except Exception as e:
-        print(f"SERVER ERROR: {e}")
-        return jsonify({"error": str(e), "logs": "Check Space logs for DNS debug info"}), 500
+        print(f"ERROR: {e}")
+        return jsonify({"error": str(e), "logs": "Check logs"}), 500
 
 @app.route('/')
 def home():
-    return "Instagram Downloader Ready. Use /download/YOUR_LINK"
+    return "Instagram Downloader (DNS Bypass Mode)"
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)