Upload app.py

app.py

@@ -30,11 +30,72 @@ sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
 
 # Import configuration and security systems
 from config.settings import get_config, is_production, is_development
-from security.error_handler import (
-    InputValidator, RateLimiter, SessionManager, ErrorHandler,
-    handle_errors, validate_input, rate_limit, require_session,
-    ValidationError, RateLimitError, AuthenticationError
-)
+
+# Try to import security systems, with fallback for Hugging Face deployment
+try:
+    from security.error_handler import (
+        InputValidator, RateLimiter, SessionManager, ErrorHandler,
+        handle_errors, validate_input, rate_limit, require_session,
+        ValidationError, RateLimitError, AuthenticationError
+    )
+    SECURITY_AVAILABLE = True
+except ImportError as e:
+    logger.warning(f"Security modules not available: {e}")
+    logger.warning("Running in Hugging Face compatibility mode")
+    SECURITY_AVAILABLE = False
+    
+    # Create fallback classes for Hugging Face deployment
+    class InputValidator:
+        def __init__(self, max_length=1000):
+            self.max_length = max_length
+        def validate(self, text):
+            return text[:self.max_length] if text else ""
+    
+    class RateLimiter:
+        def __init__(self, requests_per_minute=60, requests_per_hour=1000):
+            pass
+        def check_rate_limit(self, user_id):
+            return True
+    
+    class SessionManager:
+        def __init__(self, session_timeout_minutes=30):
+            pass
+        def create_session(self, user_id):
+            return f"session_{user_id}_{datetime.now().timestamp()}"
+        def validate_session(self, session_id):
+            return True
+    
+    class ErrorHandler:
+        def handle_error(self, error, context=""):
+            logger.error(f"Error: {error} - Context: {context}")
+            return "An error occurred. Please try again."
+    
+    # Fallback decorators
+    def handle_errors(func):
+        def wrapper(*args, **kwargs):
+            try:
+                return func(*args, **kwargs)
+            except Exception as e:
+                logger.error(f"Error in {func.__name__}: {e}")
+                return "An error occurred. Please try again."
+        return wrapper
+    
+    def validate_input(func):
+        return func
+    
+    def rate_limit(func):
+        return func
+    
+    def require_session(func):
+        return func
+    
+    # Fallback exceptions
+    class ValidationError(Exception):
+        pass
+    class RateLimitError(Exception):
+        pass
+    class AuthenticationError(Exception):
+        pass
 
 # Import the real API chat system for Hugging Face deployment
 from conversational_travel_chat_real_apis import RealAPIConversationalTravelChat
@@ -60,16 +121,23 @@ class ProductionChatApp:
         # Load configuration
         self.config = get_config()
         
-        # Initialize security systems
-        self.input_validator = InputValidator(max_length=self.config.security.max_input_length)
-        self.rate_limiter = RateLimiter(
-            requests_per_minute=self.config.api.rate_limit_per_minute,
-            requests_per_hour=self.config.api.rate_limit_per_hour
-        )
-        self.session_manager = SessionManager(
-            session_timeout_minutes=self.config.security.session_timeout_minutes
-        )
-        self.error_handler = ErrorHandler()
+        # Initialize security systems (with fallback for Hugging Face)
+        if SECURITY_AVAILABLE:
+            self.input_validator = InputValidator(max_length=self.config.security.max_input_length)
+            self.rate_limiter = RateLimiter(
+                requests_per_minute=self.config.api.rate_limit_per_minute,
+                requests_per_hour=self.config.api.rate_limit_per_hour
+            )
+            self.session_manager = SessionManager(
+                session_timeout_minutes=self.config.security.session_timeout_minutes
+            )
+            self.error_handler = ErrorHandler()
+        else:
+            # Use fallback security systems for Hugging Face
+            self.input_validator = InputValidator(max_length=1000)
+            self.rate_limiter = RateLimiter(requests_per_minute=60, requests_per_hour=1000)
+            self.session_manager = SessionManager(session_timeout_minutes=30)
+            self.error_handler = ErrorHandler()
         
         # Initialize real API chat system for Hugging Face
         self.real_api_chat = RealAPIConversationalTravelChat()