Update app.py

app.py

@@ -73,19 +73,35 @@ class URLProcessor:
             if not validators.url(url):
                 return {'is_valid': False, 'message': 'Invalid URL format'}
 
-            # Some sites block HEAD requests but allow GET
-            try:
-                response = self.session.head(url, timeout=self.timeout)
-                response.raise_for_status()
-            except (requests.exceptions.RequestException, Exception) as e:
-                logger.warning(f"HEAD request failed for {url}, trying GET: {e}")
-                # Try with GET request if HEAD fails
-                response = self.session.get(url, timeout=self.timeout, stream=True)
-                response.raise_for_status()
-                # Close the connection to avoid downloading the entire content
-                response.close()
+            # Try with DNS resolution retry
+            for attempt in range(3):  # Try up to 3 times
+                try:
+                    # Some sites block HEAD requests but allow GET
+                    try:
+                        response = self.session.head(url, timeout=self.timeout)
+                        response.raise_for_status()
+                    except (requests.exceptions.RequestException, Exception) as e:
+                        logger.warning(f"HEAD request failed for {url}, trying GET: {e}")
+                        # Try with GET request if HEAD fails
+                        response = self.session.get(url, timeout=self.timeout, stream=True)
+                        response.raise_for_status()
+                        # Close the connection to avoid downloading the entire content
+                        response.close()
+                    
+                    return {'is_valid': True, 'message': 'URL is valid and accessible'}
+                except requests.exceptions.ConnectionError as e:
+                    if "NameResolutionError" in str(e) or "Failed to resolve" in str(e):
+                        logger.warning(f"DNS resolution failed for {url}, attempt {attempt+1}/3")
+                        time.sleep(1)  # Wait a bit before retrying
+                        continue
+                    else:
+                        raise
+                except Exception as e:
+                    raise
+            
+            # If we get here, all attempts failed
+            return {'is_valid': False, 'message': f'URL validation failed: DNS resolution failed after multiple attempts'}
                 
-            return {'is_valid': True, 'message': 'URL is valid and accessible'}
         except Exception as e:
             logger.error(f"URL validation failed for {url}: {str(e)}")
             return {'is_valid': False, 'message': f'URL validation failed: {str(e)}'}
@@ -496,13 +512,48 @@ def create_interface():
     """)
     return interface
 
+def check_network_connectivity():
+    """Check if the network is working properly by testing connection to common sites"""
+    test_sites = ["https://www.google.com", "https://www.cloudflare.com", "https://www.amazon.com"]
+    results = []
+    
+    for site in test_sites:
+        try:
+            response = requests.get(site, timeout=5)
+            results.append({
+                "site": site,
+                "status": "OK" if response.status_code == 200 else f"Error: {response.status_code}",
+                "response_time": response.elapsed.total_seconds()
+            })
+        except Exception as e:
+            results.append({
+                "site": site,
+                "status": f"Error: {str(e)}",
+                "response_time": None
+            })
+    
+    # If all sites failed, there might be a network issue
+    if all(result["status"].startswith("Error") for result in results):
+        logger.error("Network connectivity issue detected. All test sites failed.")
+        return False, results
+    
+    return True, results
+
+# Add this to the main function
 def main():
     # Configure system settings
     mimetypes.init()
-
+    
+    # Check network connectivity
+    network_ok, network_results = check_network_connectivity()
+    if not network_ok:
+        logger.warning("Network connectivity issues detected. Some features may not work properly.")
+        for result in network_results:
+            logger.warning(f"Test site {result['site']}: {result['status']}")
+    
     # Create and launch interface
     interface = create_interface()
-
+    
     # Launch with proper configuration
     interface.launch(
         server_name="0.0.0.0",