Update app.py

app.py

@@ -6,6 +6,9 @@ import os
 import sys
 import psutil  # Added missing import
 from urllib.parse import urlparse
+import threading
+import re
+
 
 # Configure logging
 logging.basicConfig(
@@ -31,7 +34,13 @@ def is_valid_url(url):
 
 def sanitize_query(query):
     """Sanitize user input to prevent injection attacks"""
-    return re.sub(r'[^\w\s-]', '', query)[:256]
+    try:
+        # Remove non-alphanumeric characters except spaces and hyphens
+        clean_query = re.sub(r'[^\w\s-]', '', query)
+        return clean_query[:256]  # Truncate to prevent long queries
+    except Exception as e:
+        logger.error(f"Query sanitization failed: {str(e)}")
+        return query[:256]  # Fallback truncation
 
 # Diagnostics integration
 try: