feat: test api passport checker

visa_availability_scraper_playwright.py → app/util/visa_availability_scraper_playwright.py

@@ -1,397 +1,447 @@
-import asyncio
-import json
-from typing import Dict, Optional, List
-from playwright.async_api import async_playwright
-
-class PassportIndexVisaScraper:
-    def __init__(self, debug: bool = True):
-        """
-        Initialize the Passport Index visa scraper using Playwright
-        
-        Args:
-            debug: Enable debug output
-        """
-        self.base_url = "https://www.passportindex.org/travel-visa-checker/"
-        self.api_url = "https://www.passportindex.org/core/visachecker.php"
-        self.debug = debug
-        self.browser = None
-        self.context = None
-        self.page = None
-        
-    async def __aenter__(self):
-        """Initialize browser with stealth mode"""
-        self.playwright = await async_playwright().start()
-        
-        # Launch browser with stealth settings
-        self.browser = await self.playwright.chromium.launch(
-            headless=True,  # Using headless mode
-            args=[
-                '--disable-blink-features=AutomationControlled',
-                '--disable-dev-shm-usage',
-                '--no-sandbox',
-                '--disable-setuid-sandbox',
-                '--disable-web-security',
-                '--disable-features=IsolateOrigins,site-per-process'
-            ]
-        )
-        
-        # Create context with realistic settings
-        self.context = await self.browser.new_context(
-            viewport={'width': 1920, 'height': 1080},
-            user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36',
-            locale='en-US',
-            timezone_id='America/New_York'
-        )
-        
-        self.page = await self.context.new_page()
-
-        # Add stealth JavaScript to avoid detection
-        await self.page.add_init_script("""
-            // Override the navigator.webdriver property
-            Object.defineProperty(navigator, 'webdriver', {
-                get: () => undefined
-            });
-
-            // Override chrome property
-            window.chrome = {
-                runtime: {}
-            };
-
-            // Override permissions
-            const originalQuery = window.navigator.permissions.query;
-            window.navigator.permissions.query = (parameters) => (
-                parameters.name === 'notifications' ?
-                    Promise.resolve({ state: Notification.permission }) :
-                    originalQuery(parameters)
-            );
-        """)
-        
-        if self.debug:
-            print("🚀 Browser initialized with stealth mode")
-        
-        return self
-    
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
-        """Clean up browser resources"""
-        if self.page:
-            await self.page.close()
-        if self.context:
-            await self.context.close()
-        if self.browser:
-            await self.browser.close()
-        if self.playwright:
-            await self.playwright.stop()
-        
-        if self.debug:
-            print("🔒 Browser closed")
-    
-    async def initialize_session(self) -> bool:
-        """
-        Navigate to the website and wait for it to load properly
-        """
-        try:
-            if self.debug:
-                print("📱 Initializing session...")
-            
-            # Navigate to the page
-            try:
-                response = await self.page.goto(
-                    self.base_url,
-                    wait_until='domcontentloaded',
-                    timeout=30000
-                )
-                await self.page.wait_for_timeout(3000)
-                
-                # Get the cl value from the page
-                cl_value = await self.page.evaluate("""
-                    () => {
-                        const clInput = document.querySelector('#cl');
-                        return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
-                    }
-                """)
-                
-                if self.debug:
-                    print(f"✅ Page loaded, session ID: {cl_value}")
-                    
-                return True
-                
-            except Exception as e:
-                if self.debug:
-                    print(f"⚠️ Page load issue: {e}, continuing anyway...")
-                return True
-                
-        except Exception as e:
-            print(f"❌ Error initializing session: {e}")
-            return False
-    
-    async def check_visa_requirement_browser(self, passport_country: str, destination_country: str) -> Optional[Dict]:
-        """
-        Check visa requirements using browser automation
-        
-        Args:
-            passport_country: Two-letter country code for passport
-            destination_country: Two-letter country code for destination
-        
-        Returns:
-            Dictionary with visa information or None if failed
-        """
-        try:
-            if self.debug:
-                print(f"🌐 Checking {passport_country.upper()} → {destination_country.upper()}")
-            
-            # Get the current session ID from the page
-            cl_value = await self.page.evaluate("""
-                () => {
-                    const clInput = document.querySelector('#cl');
-                    return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
-                }
-            """)
-            
-            # Make the API request through the browser with proper argument passing
-            result = await self.page.evaluate("""
-                async (args) => {
-                    const [passport, destination, sessionId] = args;
-                    const formData = new URLSearchParams();
-                    formData.append('d', destination);
-                    formData.append('s', passport);
-                    formData.append('cl', sessionId);
-                    
-                    try {
-                        const response = await fetch('https://www.passportindex.org/core/visachecker.php', {
-                            method: 'POST',
-                            headers: {
-                                'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
-                                'X-Requested-With': 'XMLHttpRequest',
-                                'Accept': 'application/json, text/javascript, */*; q=0.01'
-                            },
-                            body: formData.toString(),
-                            credentials: 'include'
-                        });
-                        
-                        if (!response.ok) {
-                            throw new Error(`HTTP ${response.status}`);
-                        }
-                        
-                        const data = await response.json();
-                        return data;
-                    } catch (error) {
-                        return { error: error.message };
-                    }
-                }
-            """, [passport_country.lower(), destination_country.lower(), cl_value])
-            
-            if result and 'error' not in result:
-                if self.debug:
-                    print(f"✅ Got result: {result}")
-                return result
-            elif result and 'error' in result:
-                print(f"❌ API Error: {result['error']}")
-                return None
-            else:
-                return None
-                
-        except Exception as e:
-            print(f"❌ Error checking visa requirement: {e}")
-            return None
-    
-    async def check_visa_interactive(self, passport_country: str, destination_country: str) -> Optional[Dict]:
-        """
-        Alternative method: Use the interactive UI to check visa requirements
-        """
-        try:
-            if self.debug:
-                print(f"🖱️ Using interactive method for {passport_country.upper()} → {destination_country.upper()}")
-            
-            # Click on the passport selector
-            await self.page.click('.vch-select-pass')
-            await self.page.wait_for_timeout(500)
-            
-            # Find and click the country in the list
-            passport_selector = f'.vch-passports .s-div[data-ccode="{passport_country.lower()}"]'
-            await self.page.wait_for_selector(passport_selector, timeout=5000)
-            await self.page.click(passport_selector)
-            await self.page.wait_for_timeout(500)
-            
-            # Click on the destination selector
-            await self.page.click('.vch-select-des')
-            await self.page.wait_for_timeout(500)
-            
-            # Find and click the destination country
-            dest_selector = f'.vch-destinations .s-div[data-ccode="{destination_country.lower()}"]'
-            await self.page.wait_for_selector(dest_selector, timeout=5000)
-            await self.page.click(dest_selector)
-            await self.page.wait_for_timeout(1000)
-            
-            # Get the result from the page
-            result = await self.page.evaluate("""
-                () => {
-                    const resultElement = document.querySelector('.vch-result');
-                    if (resultElement) {
-                        const text = resultElement.querySelector('.text');
-                        const days = resultElement.querySelector('.days');
-                        return {
-                            text: text ? text.textContent : '',
-                            days: days ? days.textContent : '',
-                            pass: '""" + passport_country.lower() + """',
-                            dest: '""" + destination_country.upper() + """'
-                        };
-                    }
-                    return null;
-                }
-            """)
-            
-            return result
-            
-        except Exception as e:
-            if self.debug:
-                print(f"❌ Interactive method failed: {e}")
-            return None
-    
-    async def check_multiple_destinations(self, passport_country: str, destinations: List[str], delay: float = 2.0) -> Dict:
-        """
-        Check visa requirements for multiple destinations
-        
-        Args:
-            passport_country: Two-letter country code for passport
-            destinations: List of two-letter country codes for destinations
-            delay: Delay between requests in seconds
-        
-        Returns:
-            Dictionary mapping destination codes to visa information
-        """
-        results = {}
-        
-        for i, dest in enumerate(destinations, 1):
-            print(f"\n[{i}/{len(destinations)}] Checking {passport_country.upper()} → {dest.upper()}...")
-            
-            # Try API method first
-            result = await self.check_visa_requirement_browser(passport_country, dest)
-            
-            # If API fails, try interactive method
-            if not result:
-                result = await self.check_visa_interactive(passport_country, dest)
-            
-            if result:
-                results[dest] = result
-                text = result.get('text', 'No text available')
-                print(f"  ✅ Result: {text}")
-            else:
-                results[dest] = None
-                print(f"  ❌ Failed to get result")
-            
-            # Rate limiting
-            if i < len(destinations):
-                print(f"  ⏳ Waiting {delay} seconds...")
-                await asyncio.sleep(delay)
-        
-        return results
-    
-    def format_result(self, result: Dict) -> str:
-        """Format a single result for display"""
-        if not result:
-            return "No information available"
-        
-        text = result.get('text', 'N/A')
-        dest = result.get('dest', 'N/A')
-        passport = result.get('pass', 'N/A')
-        
-        return f"{passport.upper()} → {dest.upper()}: {text}"
-
-
-async def main():
-    """Main function to demonstrate usage"""
-    print("="*60)
-    print(" Passport Index Visa Checker (Playwright)")
-    print("="*60)
-    
-    async with PassportIndexVisaScraper(debug=True) as scraper:
-        # Initialize session
-        if not await scraper.initialize_session():
-            print("❌ Failed to initialize session")
-            return
-        
-        print("\n" + "="*60)
-        print(" Testing visa requirements...")
-        print("="*60)
-        
-        # Test single visa requirement
-        print("\n📍 Single visa check: US → GB")
-        print("-" * 40)
-        result = await scraper.check_visa_requirement_browser('us', 'gb')
-        if result:
-            print(f"Result: {scraper.format_result(result)}")
-        else:
-            print("Trying interactive method...")
-            result = await scraper.check_visa_interactive('us', 'gb')
-            if result:
-                print(f"Result: {scraper.format_result(result)}")
-        
-        # Test multiple destinations
-        print("\n📍 Multiple destinations for US passport:")
-        print("-" * 40)
-        destinations = ['ca', 'mx', 'jp', 'au']  # Canada, Mexico, Japan, Australia
-        results = await scraper.check_multiple_destinations('us', destinations, delay=2.0)
-        
-        print("\n📊 Summary:")
-        for dest, result in results.items():
-            if result:
-                print(f"  ✅ {scraper.format_result(result)}")
-            else:
-                print(f"  ❌ US → {dest.upper()}: Failed")
-
-
-# Country codes reference (partial list)
-COUNTRY_CODES = {
-    'af': 'Afghanistan', 'al': 'Albania', 'dz': 'Algeria', 'ad': 'Andorra',
-    'ao': 'Angola', 'ag': 'Antigua and Barbuda', 'ar': 'Argentina', 'am': 'Armenia',
-    'au': 'Australia', 'at': 'Austria', 'az': 'Azerbaijan', 'bs': 'Bahamas',
-    'bh': 'Bahrain', 'bd': 'Bangladesh', 'bb': 'Barbados', 'by': 'Belarus',
-    'be': 'Belgium', 'bz': 'Belize', 'bj': 'Benin', 'bt': 'Bhutan',
-    'bo': 'Bolivia', 'ba': 'Bosnia and Herzegovina', 'bw': 'Botswana', 'br': 'Brazil',
-    'bn': 'Brunei', 'bg': 'Bulgaria', 'bf': 'Burkina Faso', 'bi': 'Burundi',
-    'kh': 'Cambodia', 'cm': 'Cameroon', 'ca': 'Canada', 'cv': 'Cape Verde',
-    'cf': 'Central African Republic', 'td': 'Chad', 'cl': 'Chile', 'cn': 'China',
-    'co': 'Colombia', 'km': 'Comoros', 'cg': 'Congo', 'cr': 'Costa Rica',
-    'hr': 'Croatia', 'cu': 'Cuba', 'cy': 'Cyprus', 'cz': 'Czech Republic',
-    'dk': 'Denmark', 'dj': 'Djibouti', 'dm': 'Dominica', 'do': 'Dominican Republic',
-    'ec': 'Ecuador', 'eg': 'Egypt', 'sv': 'El Salvador', 'gq': 'Equatorial Guinea',
-    'er': 'Eritrea', 'ee': 'Estonia', 'et': 'Ethiopia', 'fj': 'Fiji',
-    'fi': 'Finland', 'fr': 'France', 'ga': 'Gabon', 'gm': 'Gambia',
-    'ge': 'Georgia', 'de': 'Germany', 'gh': 'Ghana', 'gr': 'Greece',
-    'gd': 'Grenada', 'gt': 'Guatemala', 'gn': 'Guinea', 'gw': 'Guinea-Bissau',
-    'gy': 'Guyana', 'ht': 'Haiti', 'hn': 'Honduras', 'hu': 'Hungary',
-    'is': 'Iceland', 'in': 'India', 'id': 'Indonesia', 'ir': 'Iran',
-    'iq': 'Iraq', 'ie': 'Ireland', 'il': 'Israel', 'it': 'Italy',
-    'jm': 'Jamaica', 'jp': 'Japan', 'jo': 'Jordan', 'kz': 'Kazakhstan',
-    'ke': 'Kenya', 'ki': 'Kiribati', 'kp': 'North Korea', 'kr': 'South Korea',
-    'kw': 'Kuwait', 'kg': 'Kyrgyzstan', 'la': 'Laos', 'lv': 'Latvia',
-    'lb': 'Lebanon', 'ls': 'Lesotho', 'lr': 'Liberia', 'ly': 'Libya',
-    'li': 'Liechtenstein', 'lt': 'Lithuania', 'lu': 'Luxembourg', 'mk': 'Macedonia',
-    'mg': 'Madagascar', 'mw': 'Malawi', 'my': 'Malaysia', 'mv': 'Maldives',
-    'ml': 'Mali', 'mt': 'Malta', 'mh': 'Marshall Islands', 'mr': 'Mauritania',
-    'mu': 'Mauritius', 'mx': 'Mexico', 'fm': 'Micronesia', 'md': 'Moldova',
-    'mc': 'Monaco', 'mn': 'Mongolia', 'me': 'Montenegro', 'ma': 'Morocco',
-    'mz': 'Mozambique', 'mm': 'Myanmar', 'na': 'Namibia', 'nr': 'Nauru',
-    'np': 'Nepal', 'nl': 'Netherlands', 'nz': 'New Zealand', 'ni': 'Nicaragua',
-    'ne': 'Niger', 'ng': 'Nigeria', 'no': 'Norway', 'om': 'Oman',
-    'pk': 'Pakistan', 'pw': 'Palau', 'pa': 'Panama', 'pg': 'Papua New Guinea',
-    'py': 'Paraguay', 'pe': 'Peru', 'ph': 'Philippines', 'pl': 'Poland',
-    'pt': 'Portugal', 'qa': 'Qatar', 'ro': 'Romania', 'ru': 'Russia',
-    'rw': 'Rwanda', 'kn': 'Saint Kitts and Nevis', 'lc': 'Saint Lucia',
-    'vc': 'Saint Vincent and the Grenadines', 'ws': 'Samoa', 'sm': 'San Marino',
-    'st': 'Sao Tome and Principe', 'sa': 'Saudi Arabia', 'sn': 'Senegal',
-    'rs': 'Serbia', 'sc': 'Seychelles', 'sl': 'Sierra Leone', 'sg': 'Singapore',
-    'sk': 'Slovakia', 'si': 'Slovenia', 'sb': 'Solomon Islands', 'so': 'Somalia',
-    'za': 'South Africa', 'es': 'Spain', 'lk': 'Sri Lanka', 'sd': 'Sudan',
-    'sr': 'Suriname', 'sz': 'Swaziland', 'se': 'Sweden', 'ch': 'Switzerland',
-    'sy': 'Syria', 'tw': 'Taiwan', 'tj': 'Tajikistan', 'tz': 'Tanzania',
-    'th': 'Thailand', 'tl': 'Timor-Leste', 'tg': 'Togo', 'to': 'Tonga',
-    'tt': 'Trinidad and Tobago', 'tn': 'Tunisia', 'tr': 'Turkey', 'tm': 'Turkmenistan',
-    'tv': 'Tuvalu', 'ug': 'Uganda', 'ua': 'Ukraine', 'ae': 'United Arab Emirates',
-    'gb': 'United Kingdom', 'us': 'United States', 'uy': 'Uruguay', 'uz': 'Uzbekistan',
-    'vu': 'Vanuatu', 've': 'Venezuela', 'vn': 'Vietnam', 'ye': 'Yemen',
-    'zm': 'Zambia', 'zw': 'Zimbabwe'
-}
-
-if __name__ == "__main__":
-    asyncio.run(main())
+import asyncio
+import json
+from typing import Dict, Optional, List
+from playwright.async_api import async_playwright
+
+class PassportIndexVisaScraper:
+    def __init__(self, debug: bool = True):
+        """
+        Initialize the Passport Index visa scraper using Playwright
+        
+        Args:
+            debug: Enable debug output
+        """
+        self.base_url = "https://www.passportindex.org/travel-visa-checker/"
+        self.api_url = "https://www.passportindex.org/core/visachecker.php"
+        self.debug = debug
+        self.browser = None
+        self.context = None
+        self.page = None
+        
+    async def __aenter__(self):
+        """Initialize browser with stealth mode"""
+        self.playwright = await async_playwright().start()
+        
+        self.browser = await self.playwright.chromium.launch(
+            headless=True,  # Using headless mode
+            args=[
+                '--disable-blink-features=AutomationControlled',
+                '--disable-dev-shm-usage',
+                '--no-sandbox',
+                '--disable-setuid-sandbox',
+                '--disable-web-security',
+                '--disable-features=IsolateOrigins,site-per-process'
+            ]
+        )
+        
+        # Create context with realistic settings
+        self.context = await self.browser.new_context(
+            viewport={'width': 1920, 'height': 1080},
+            user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/140.0.0.0 Safari/537.36',
+            locale='en-US',
+            timezone_id='America/New_York'
+        )
+        
+        self.page = await self.context.new_page()
+
+        # Add stealth JavaScript to avoid detection
+        await self.page.add_init_script("""
+            // Override the navigator.webdriver property
+            Object.defineProperty(navigator, 'webdriver', {
+                get: () => undefined
+            });
+
+            // Override chrome property
+            window.chrome = {
+                runtime: {}
+            };
+
+            // Override permissions
+            const originalQuery = window.navigator.permissions.query;
+            window.navigator.permissions.query = (parameters) => (
+                parameters.name === 'notifications' ?
+                    Promise.resolve({ state: Notification.permission }) :
+                    originalQuery(parameters)
+            );
+        """)
+        
+        if self.debug:
+            print("🚀 Browser initialized with stealth mode")
+        
+        return self
+    
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """Clean up browser resources"""
+        if self.page:
+            await self.page.close()
+        if self.context:
+            await self.context.close()
+        if self.browser:
+            await self.browser.close()
+        if self.playwright:
+            await self.playwright.stop()
+        
+        if self.debug:
+            print("🔒 Browser closed")
+    
+    async def initialize_session(self) -> bool:
+        """
+        Navigate to the website and wait for it to load properly
+        """
+        try:
+            if self.debug:
+                print("📱 Initializing session...")
+            
+            # Navigate to the page
+            try:
+                response = await self.page.goto(
+                    self.base_url,
+                    wait_until='domcontentloaded',
+                    timeout=30000
+                )
+                await self.page.wait_for_timeout(3000)
+                
+                # Get the cl value from the page
+                cl_value = await self.page.evaluate("""
+                    () => {
+                        const clInput = document.querySelector('#cl');
+                        return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
+                    }
+                """)
+                
+                if self.debug:
+                    print(f"✅ Page loaded, session ID: {cl_value}")
+                    
+                return True
+                
+            except Exception as e:
+                if self.debug:
+                    print(f"⚠️ Page load issue: {e}, continuing anyway...")
+                return True
+                
+        except Exception as e:
+            print(f"❌ Error initializing session: {e}")
+            return False
+    
+    async def check_visa_requirement_browser(self, passport_country: str, destination_country: str) -> Optional[Dict]:
+        """
+        Check visa requirements using browser automation
+        
+        Args:
+            passport_country: Two-letter country code for passport
+            destination_country: Two-letter country code for destination
+        
+        Returns:
+            Dictionary with visa information or None if failed
+        """
+        try:
+            if self.debug:
+                print(f"🌐 Checking {passport_country.upper()} → {destination_country.upper()}")
+            
+            # Get the current session ID from the page
+            cl_value = await self.page.evaluate("""
+                () => {
+                    const clInput = document.querySelector('#cl');
+                    return clInput ? clInput.value : 'bc2140a2d83928ce1112d01e610bad89';
+                }
+            """)
+            
+            # Make the API request through the browser with proper argument passing
+            result = await self.page.evaluate("""
+                async (args) => {
+                    const [passport, destination, sessionId] = args;
+                    const formData = new URLSearchParams();
+                    formData.append('d', destination);
+                    formData.append('s', passport);
+                    formData.append('cl', sessionId);
+                    
+                    try {
+                        const response = await fetch('https://www.passportindex.org/core/visachecker.php', {
+                            method: 'POST',
+                            headers: {
+                                'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
+                                'X-Requested-With': 'XMLHttpRequest',
+                                'Accept': 'application/json, text/javascript, */*; q=0.01'
+                            },
+                            body: formData.toString(),
+                            credentials: 'include'
+                        });
+                        
+                        if (!response.ok) {
+                            throw new Error(`HTTP ${response.status}`);
+                        }
+                        
+                        const data = await response.json();
+                        return data;
+                    } catch (error) {
+                        return { error: error.message };
+                    }
+                }
+            """, [passport_country.lower(), destination_country.lower(), cl_value])
+            
+            if result and 'error' not in result:
+                if self.debug:
+                    print(f"✅ Got result: {result}")
+                return result
+            elif result and 'error' in result:
+                print(f"❌ API Error: {result['error']}")
+                return None
+            else:
+                return None
+                
+        except Exception as e:
+            print(f"❌ Error checking visa requirement: {e}")
+            return None
+    
+    async def check_visa_interactive(self, passport_country: str, destination_country: str) -> Optional[Dict]:
+        """
+        Alternative method: Use the interactive UI to check visa requirements
+        """
+        try:
+            if self.debug:
+                print(f"🖱️ Using interactive method for {passport_country.upper()} → {destination_country.upper()}")
+            
+            # Click on the passport selector
+            await self.page.click('.vch-select-pass')
+            await self.page.wait_for_timeout(500)
+            
+            # Find and click the country in the list
+            passport_selector = f'.vch-passports .s-div[data-ccode="{passport_country.lower()}"]'
+            await self.page.wait_for_selector(passport_selector, timeout=5000)
+            await self.page.click(passport_selector)
+            await self.page.wait_for_timeout(500)
+            
+            # Click on the destination selector
+            await self.page.click('.vch-select-des')
+            await self.page.wait_for_timeout(500)
+            
+            # Find and click the destination country
+            dest_selector = f'.vch-destinations .s-div[data-ccode="{destination_country.lower()}"]'
+            await self.page.wait_for_selector(dest_selector, timeout=5000)
+            await self.page.click(dest_selector)
+            await self.page.wait_for_timeout(1000)
+            
+            # Get the result from the page
+            result = await self.page.evaluate("""
+                () => {
+                    const resultElement = document.querySelector('.vch-result');
+                    if (resultElement) {
+                        const text = resultElement.querySelector('.text');
+                        const days = resultElement.querySelector('.days');
+                        return {
+                            text: text ? text.textContent : '',
+                            days: days ? days.textContent : '',
+                            pass: '""" + passport_country.lower() + """',
+                            dest: '""" + destination_country.upper() + """'
+                        };
+                    }
+                    return null;
+                }
+            """)
+            
+            return result
+            
+        except Exception as e:
+            if self.debug:
+                print(f"❌ Interactive method failed: {e}")
+            return None
+    
+    async def check_multiple_destinations(self, passport_country: str, destinations: List[str], delay: float = 2.0) -> Dict:
+        """
+        Check visa requirements for multiple destinations
+        
+        Args:
+            passport_country: Two-letter country code for passport
+            destinations: List of two-letter country codes for destinations
+            delay: Delay between requests in seconds
+        
+        Returns:
+            Dictionary mapping destination codes to visa information
+        """
+        results = {}
+        
+        for i, dest in enumerate(destinations, 1):
+            print(f"\n[{i}/{len(destinations)}] Checking {passport_country.upper()} → {dest.upper()}...")
+            
+            # Try API method first
+            result = await self.check_visa_requirement_browser(passport_country, dest)
+            
+            # If API fails, try interactive method
+            if not result:
+                result = await self.check_visa_interactive(passport_country, dest)
+            
+            if result:
+                results[dest] = result
+                text = result.get('text', 'No text available')
+                print(f"  ✅ Result: {text}")
+            else:
+                results[dest] = None
+                print(f"  ❌ Failed to get result")
+            
+            # Rate limiting
+            if i < len(destinations):
+                print(f"  ⏳ Waiting {delay} seconds...")
+                await asyncio.sleep(delay)
+        
+        return results
+    
+    def format_result(self, result: Dict) -> str:
+        """Format a single result for display"""
+        if not result:
+            return "No information available"
+        
+        text = result.get('text', 'N/A')
+        dest = result.get('dest', 'N/A')
+        passport = result.get('pass', 'N/A')
+        
+        return f"{passport.upper()} → {dest.upper()}: {text}"
+
+
+async def main():
+    """Main function to demonstrate usage"""
+    print("="*60)
+    print(" Passport Index Visa Checker (Playwright)")
+    print("="*60)
+    
+    async with PassportIndexVisaScraper(debug=True) as scraper:
+        # Initialize session
+        if not await scraper.initialize_session():
+            print("❌ Failed to initialize session")
+            return
+        
+        print("\n" + "="*60)
+        print(" Testing visa requirements...")
+        print("="*60)
+        
+        # Test single visa requirement
+        print("\n📍 Single visa check: US → GB")
+        print("-" * 40)
+        result = await scraper.check_visa_requirement_browser('us', 'gb')
+        if result:
+            print(f"Result: {scraper.format_result(result)}")
+        else:
+            print("Trying interactive method...")
+            result = await scraper.check_visa_interactive('us', 'gb')
+            if result:
+                print(f"Result: {scraper.format_result(result)}")
+        
+        # Test multiple destinations
+        print("\n📍 Multiple destinations for US passport:")
+        print("-" * 40)
+        destinations = ['ca', 'mx', 'jp', 'au']  # Canada, Mexico, Japan, Australia
+        results = await scraper.check_multiple_destinations('us', destinations, delay=2.0)
+        
+        print("\n📊 Summary:")
+        for dest, result in results.items():
+            if result:
+                print(f"  ✅ {scraper.format_result(result)}")
+            else:
+                print(f"  ❌ US → {dest.upper()}: Failed")
+
+
+COUNTRY_CODES = {
+    'af': 'Afghanistan', 'al': 'Albania', 'dz': 'Algeria', 'ad': 'Andorra',
+    'ao': 'Angola', 'ag': 'Antigua and Barbuda', 'ar': 'Argentina', 'am': 'Armenia',
+    'au': 'Australia', 'at': 'Austria', 'az': 'Azerbaijan', 'bs': 'Bahamas',
+    'bh': 'Bahrain', 'bd': 'Bangladesh', 'bb': 'Barbados', 'by': 'Belarus',
+    'be': 'Belgium', 'bz': 'Belize', 'bj': 'Benin', 'bt': 'Bhutan',
+    'bo': 'Bolivia', 'ba': 'Bosnia and Herzegovina', 'bw': 'Botswana', 'br': 'Brazil',
+    'bn': 'Brunei', 'bg': 'Bulgaria', 'bf': 'Burkina Faso', 'bi': 'Burundi',
+    'kh': 'Cambodia', 'cm': 'Cameroon', 'ca': 'Canada', 'cv': 'Cape Verde',
+    'cf': 'Central African Republic', 'td': 'Chad', 'cl': 'Chile', 'cn': 'China',
+    'co': 'Colombia', 'km': 'Comoros', 'cg': 'Congo', 'cr': 'Costa Rica',
+    'hr': 'Croatia', 'cu': 'Cuba', 'cy': 'Cyprus', 'cz': 'Czech Republic',
+    'dk': 'Denmark', 'dj': 'Djibouti', 'dm': 'Dominica', 'do': 'Dominican Republic',
+    'ec': 'Ecuador', 'eg': 'Egypt', 'sv': 'El Salvador', 'gq': 'Equatorial Guinea',
+    'er': 'Eritrea', 'ee': 'Estonia', 'et': 'Ethiopia', 'fj': 'Fiji',
+    'fi': 'Finland', 'fr': 'France', 'ga': 'Gabon', 'gm': 'Gambia',
+    'ge': 'Georgia', 'de': 'Germany', 'gh': 'Ghana', 'gr': 'Greece',
+    'gd': 'Grenada', 'gt': 'Guatemala', 'gn': 'Guinea', 'gw': 'Guinea-Bissau',
+    'gy': 'Guyana', 'ht': 'Haiti', 'hn': 'Honduras', 'hu': 'Hungary',
+    'is': 'Iceland', 'in': 'India', 'id': 'Indonesia', 'ir': 'Iran',
+    'iq': 'Iraq', 'ie': 'Ireland', 'il': 'Israel', 'it': 'Italy',
+    'jm': 'Jamaica', 'jp': 'Japan', 'jo': 'Jordan', 'kz': 'Kazakhstan',
+    'ke': 'Kenya', 'ki': 'Kiribati', 'kp': 'North Korea', 'kr': 'South Korea',
+    'kw': 'Kuwait', 'kg': 'Kyrgyzstan', 'la': 'Laos', 'lv': 'Latvia',
+    'lb': 'Lebanon', 'ls': 'Lesotho', 'lr': 'Liberia', 'ly': 'Libya',
+    'li': 'Liechtenstein', 'lt': 'Lithuania', 'lu': 'Luxembourg', 'mk': 'Macedonia',
+    'mg': 'Madagascar', 'mw': 'Malawi', 'my': 'Malaysia', 'mv': 'Maldives',
+    'ml': 'Mali', 'mt': 'Malta', 'mh': 'Marshall Islands', 'mr': 'Mauritania',
+    'mu': 'Mauritius', 'mx': 'Mexico', 'fm': 'Micronesia', 'md': 'Moldova',
+    'mc': 'Monaco', 'mn': 'Mongolia', 'me': 'Montenegro', 'ma': 'Morocco',
+    'mz': 'Mozambique', 'mm': 'Myanmar', 'na': 'Namibia', 'nr': 'Nauru',
+    'np': 'Nepal', 'nl': 'Netherlands', 'nz': 'New Zealand', 'ni': 'Nicaragua',
+    'ne': 'Niger', 'ng': 'Nigeria', 'no': 'Norway', 'om': 'Oman',
+    'pk': 'Pakistan', 'pw': 'Palau', 'pa': 'Panama', 'pg': 'Papua New Guinea',
+    'py': 'Paraguay', 'pe': 'Peru', 'ph': 'Philippines', 'pl': 'Poland',
+    'pt': 'Portugal', 'qa': 'Qatar', 'ro': 'Romania', 'ru': 'Russia',
+    'rw': 'Rwanda', 'kn': 'Saint Kitts and Nevis', 'lc': 'Saint Lucia',
+    'vc': 'Saint Vincent and the Grenadines', 'ws': 'Samoa', 'sm': 'San Marino',
+    'st': 'Sao Tome and Principe', 'sa': 'Saudi Arabia', 'sn': 'Senegal',
+    'rs': 'Serbia', 'sc': 'Seychelles', 'sl': 'Sierra Leone', 'sg': 'Singapore',
+    'sk': 'Slovakia', 'si': 'Slovenia', 'sb': 'Solomon Islands', 'so': 'Somalia',
+    'za': 'South Africa', 'es': 'Spain', 'lk': 'Sri Lanka', 'sd': 'Sudan',
+    'sr': 'Suriname', 'sz': 'Swaziland', 'se': 'Sweden', 'ch': 'Switzerland',
+    'sy': 'Syria', 'tw': 'Taiwan', 'tj': 'Tajikistan', 'tz': 'Tanzania',
+    'th': 'Thailand', 'tl': 'Timor-Leste', 'tg': 'Togo', 'to': 'Tonga',
+    'tt': 'Trinidad and Tobago', 'tn': 'Tunisia', 'tr': 'Turkey', 'tm': 'Turkmenistan',
+    'tv': 'Tuvalu', 'ug': 'Uganda', 'ua': 'Ukraine', 'ae': 'United Arab Emirates',
+    'gb': 'United Kingdom', 'us': 'United States', 'uy': 'Uruguay', 'uz': 'Uzbekistan',
+    'vu': 'Vanuatu', 've': 'Venezuela', 'vn': 'Vietnam', 'ye': 'Yemen',
+    'zm': 'Zambia', 'zw': 'Zimbabwe'
+}
+
+REVERSED_COUNTRY_CODES = {
+    'Afghanistan': 'af', 'Albania': 'al', 'Algeria': 'dz', 'Andorra': 'ad',
+    'Angola': 'ao', 'Antigua and Barbuda': 'ag', 'Argentina': 'ar', 'Armenia': 'am',
+    'Australia': 'au', 'Austria': 'at', 'Azerbaijan': 'az', 'Bahamas': 'bs',
+    'Bahrain': 'bh', 'Bangladesh': 'bd', 'Barbados': 'bb', 'Belarus': 'by',
+    'Belgium': 'be', 'Belize': 'bz', 'Benin': 'bj', 'Bhutan': 'bt',
+    'Bolivia': 'bo', 'Bosnia and Herzegovina': 'ba', 'Botswana': 'bw', 'Brazil': 'br',
+    'Brunei': 'bn', 'Bulgaria': 'bg', 'Burkina Faso': 'bf', 'Burundi': 'bi',
+    'Cambodia': 'kh', 'Cameroon': 'cm', 'Canada': 'ca', 'Cape Verde': 'cv',
+    'Central African Republic': 'cf', 'Chad': 'td', 'Chile': 'cl', 'China': 'cn',
+    'Colombia': 'co', 'Comoros': 'km', 'Congo': 'cg', 'Costa Rica': 'cr',
+    'Croatia': 'hr', 'Cuba': 'cu', 'Cyprus': 'cy', 'Czech Republic': 'cz',
+    'Denmark': 'dk', 'Djibouti': 'dj', 'Dominica': 'dm', 'Dominican Republic': 'do',
+    'Ecuador': 'ec', 'Egypt': 'eg', 'El Salvador': 'sv', 'Equatorial Guinea': 'gq',
+    'Eritrea': 'er', 'Estonia': 'ee', 'Ethiopia': 'et', 'Fiji': 'fj',
+    'Finland': 'fi', 'France': 'fr', 'Gabon': 'ga', 'Gambia': 'gm',
+    'Georgia': 'ge', 'Germany': 'de', 'Ghana': 'gh', 'Greece': 'gr',
+    'Grenada': 'gd', 'Guatemala': 'gt', 'Guinea': 'gn', 'Guinea-Bissau': 'gw',
+    'Guyana': 'gy', 'Haiti': 'ht', 'Honduras': 'hn', 'Hungary': 'hu',
+    'Iceland': 'is', 'India': 'in', 'Indonesia': 'id', 'Iran': 'ir',
+    'Iraq': 'iq', 'Ireland': 'ie', 'Israel': 'il', 'Italy': 'it',
+    'Jamaica': 'jm', 'Japan': 'jp', 'Jordan': 'jo', 'Kazakhstan': 'kz',
+    'Kenya': 'ke', 'Kiribati': 'ki', 'North Korea': 'kp', 'South Korea': 'kr',
+    'Kuwait': 'kw', 'Kyrgyzstan': 'kg', 'Laos': 'la', 'Latvia': 'lv',
+    'Lebanon': 'lb', 'Lesotho': 'ls', 'Liberia': 'lr', 'Libya': 'ly',
+    'Liechtenstein': 'li', 'Lithuania': 'lt', 'Luxembourg': 'lu', 'Macedonia': 'mk',
+    'Madagascar': 'mg', 'Malawi': 'mw', 'Malaysia': 'my', 'Maldives': 'mv',
+    'Mali': 'ml', 'Malta': 'mt', 'Marshall Islands': 'mh', 'Mauritania': 'mr',
+    'Mauritius': 'mu', 'Mexico': 'mx', 'Micronesia': 'fm', 'Moldova': 'md',
+    'Monaco': 'mc', 'Mongolia': 'mn', 'Montenegro': 'me', 'Morocco': 'ma',
+    'Mozambique': 'mz', 'Myanmar': 'mm', 'Namibia': 'na', 'Nauru': 'nr',
+    'Nepal': 'np', 'Netherlands': 'nl', 'New Zealand': 'nz', 'Nicaragua': 'ni',
+    'Niger': 'ne', 'Nigeria': 'ng', 'Norway': 'no', 'Oman': 'om',
+    'Pakistan': 'pk', 'Palau': 'pw', 'Panama': 'pa', 'Papua New Guinea': 'pg',
+    'Paraguay': 'py', 'Peru': 'pe', 'Philippines': 'ph', 'Poland': 'pl',
+    'Portugal': 'pt', 'Qatar': 'qa', 'Romania': 'ro', 'Russia': 'ru',
+    'Rwanda': 'rw', 'Saint Kitts and Nevis': 'kn', 'Saint Lucia': 'lc',
+    'Saint Vincent and the Grenadines': 'vc', 'Samoa': 'ws', 'San Marino': 'sm',
+    'Sao Tome and Principe': 'st', 'Saudi Arabia': 'sa', 'Senegal': 'sn',
+    'Serbia': 'rs', 'Seychelles': 'sc', 'Sierra Leone': 'sl', 'Singapore': 'sg',
+    'Slovakia': 'sk', 'Slovenia': 'si', 'Solomon Islands': 'sb', 'Somalia': 'so',
+    'South Africa': 'za', 'Spain': 'es', 'Sri Lanka': 'lk', 'Sudan': 'sd',
+    'Suriname': 'sr', 'Swaziland': 'sz', 'Sweden': 'se', 'Switzerland': 'ch',
+    'Syria': 'sy', 'Taiwan': 'tw', 'Tajikistan': 'tj', 'Tanzania': 'tz',
+    'Thailand': 'th', 'Timor-Leste': 'tl', 'Togo': 'tg', 'Tonga': 'to',
+    'Trinidad and Tobago': 'tt', 'Tunisia': 'tn', 'Turkey': 'tr', 'Turkmenistan': 'tm',
+    'Tuvalu': 'tv', 'Uganda': 'ug', 'Ukraine': 'ua', 'United Arab Emirates': 'ae',
+    'United Kingdom': 'gb', 'United States': 'us', 'Uruguay': 'uy', 'Uzbekistan': 'uz',
+    'Vanuatu': 'vu', 'Venezuela': 've', 'Vietnam': 'vn', 'Yemen': 'ye',
+    'Zambia': 'zm', 'Zimbabwe': 'zw'
+}
+
+if __name__ == "__main__":
+    asyncio.run(main())

server.py

@@ -8,7 +8,7 @@ import json
 
 from app.util.gen_ai_base import GenAIBaseClient
 from app.util.browser_agent import BrowserAgent
-
+from app.util.visa_availability_scraper_playwright import PassportIndexVisaScraper
 import sys
 sys.stdout.reconfigure(line_buffering=True)
 
@@ -54,6 +54,27 @@ def create_app() -> Flask:
             import traceback
             traceback.print_exc()
             return jsonify({"error": str(e)}), 500
+    
+    @app.route('/visa-checker', methods=['POST'])
+    async def visa_checker():
+        raw = request.get_data(as_text=True)
+        body = json.loads(raw)
+        source = body.get('source')
+        destination = body.get('destination')
+        async with PassportIndexVisaScraper(debug=True) as scraper:
+            if not await scraper.initialize_session():
+                return jsonify({"error": "Failed to initialize session"}), 500
+            result = await scraper.check_visa_requirements(source, destination)
+
+            if result:
+                return jsonify(result), 200
+            else:
+                result = await scraper.check_visa_interactive(source, destination)
+                if result:
+                    return jsonify(result), 200
+                else:
+                    return jsonify({"error": "Failed to retrieve visa information"}), 500
+    
     @app.route('/', methods=['GET'])
     def hello_world():
         return "Flask server is running.", 200