Update core/input_handler.py

core/input_handler.py

@@ -1,16 +1,17 @@
-def is_public_ip(url):
-    """
-    Checks whether the resolved IP address of a URL is public (non-local).
-    Prevents SSRF by blocking internal addresses like 127.0.0.1 or 192.168.x.x.
-    """
-    try:
-        hostname = urlparse(url).hostname
-        ip = socket.gethostbyname(hostname)
-        ip_obj = ipaddress.ip_address(ip)
-        return ip_obj.is_global  # Only allow globally routable IPs
-    except Exception as e:
-        logger.warning(f"URL IP validation failed: {e}")
-        return False
+import logging
+import os
+import io
+import tempfile
+import requests
+from PIL import Image
+import cv2
+from urllib.parse import urlparse
+import socket
+import ipaddress
+
+# Setup logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
 
 
 def fetch_media_from_url(url):