Create helpers.py

utils/helpers.py

@@ -0,0 +1,42 @@
+import uuid
+import time
+import logging
+import socket
+import ipaddress
+from urllib.parse import urlparse
+import gradio as gr
+
+
+
+# Helper Functions
+def format_error(message):
+    """Formats error messages for consistent user feedback."""
+    return {"error": message}
+
+def toggle_visibility(show, *components):
+    """Toggles visibility for multiple Gradio components."""
+    return [gr.update(visible=show) for _ in components]
+
+def generate_session_id():
+    """Generates a unique session ID for tracking inputs."""
+    return str(uuid.uuid4())
+
+def log_runtime(start_time):
+    """Logs the runtime of a process."""
+    elapsed_time = time.time() - start_time
+    logger.info(f"Process completed in {elapsed_time:.2f} seconds.")
+    return elapsed_time
+
+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