Update app.py

app.py

@@ -75,6 +75,77 @@ if HF_TOKEN:
             logger.error(f"Attempt {attempt + 1} failed to initialize Hugging Face API: {str(e)}")
             time.sleep(2 ** attempt)
 
+# ========== UTILITY FUNCTIONS ==========
+class DataEncryptor:
+    def __init__(self, key: str):
+        self.cipher = Fernet(key.encode())
+    
+    def encrypt(self, data: str) -> str:
+        return self.cipher.encrypt(data.encode()).decode()
+    
+    def decrypt(self, encrypted_data: str) -> str:
+        return self.cipher.decrypt(encrypted_data.encode()).decode()
+
+encryptor = DataEncryptor(ENCRYPTION_KEY)
+
+def generate_session_token() -> str:
+    alphabet = string.ascii_letters + string.digits
+    return ''.join(secrets.choice(alphabet) for _ in range(SESSION_TOKEN_LENGTH))
+
+def sanitize_input(text: str) -> str:
+    if not text:
+        return ""
+    text = html.escape(text.strip())
+    text = re.sub(r'<[^>]*>', '', text)
+    text = re.sub(r'[^\w\s\-.,!?@#\$%^&*()+=]', '', text)
+    return text
+
+def validate_name(name: str) -> str:
+    name = name.strip()
+    if not name:
+        raise ValueError("Name cannot be empty.")
+    if len(name) > 100:
+        raise ValueError("Name is too long (maximum 100 characters).")
+    if any(c.isdigit() for c in name):
+        raise ValueError("Name cannot contain numbers.")
+    return name
+
+def validate_age(age: Union[int, float, str]) -> int:
+    try:
+        age_int = int(age)
+        if not MIN_AGE <= age_int <= MAX_AGE:
+            raise ValueError(f"Age must be between {MIN_AGE} and {MAX_AGE}.")
+        return age_int
+    except (ValueError, TypeError):
+        raise ValueError("Please enter a valid age number.")
+
+def validate_file(file_obj) -> None:
+    if not file_obj:
+        raise ValueError("Please upload a file first")
+    
+    file_ext = os.path.splitext(file_obj.name)[1].lower()
+    if file_ext not in ALLOWED_FILE_TYPES:
+        raise ValueError(f"Invalid file type. Allowed types: {', '.join(ALLOWED_FILE_TYPES)}")
+    
+    file_size = os.path.getsize(file_obj.name) / (1024 * 1024)
+    if file_size > MAX_FILE_SIZE_MB:
+        raise ValueError(f"File too large. Maximum size is {MAX_FILE_SIZE_MB}MB.")
+
+def remove_sensitive_info(text: str) -> str:
+    patterns = [
+        (r'\b\d{3}-\d{2}-\d{4}\b', '[REDACTED-SSN]'),
+        (r'\b\d{6,9}\b', '[ID]'),
+        (r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b', '[EMAIL]'),
+        (r'\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b', '[IP]'),
+        (r'\b[A-Z][a-z]+ [A-Z][a-z]+\b', '[NAME]'),
+        (r'\b\d{3}\) \d{3}-\d{4}\b', '[PHONE]'),
+        (r'\b\d{1,5} [A-Z][a-z]+ [A-Z][a-z]+, [A-Z]{2} \d{5}\b', '[ADDRESS]')
+    ]
+    
+    for pattern, replacement in patterns:
+        text = re.sub(pattern, replacement, text)
+    return text
+
 # ========== LEARNING STYLE QUIZ ==========
 class LearningStyleQuiz:
     def __init__(self):
@@ -1582,77 +1653,6 @@ class EnhancedTeachingAssistant:
 # Initialize teaching assistant
 teaching_assistant = EnhancedTeachingAssistant()
 
-# ========== UTILITY FUNCTIONS ==========
-class DataEncryptor:
-    def __init__(self, key: str):
-        self.cipher = Fernet(key.encode())
-    
-    def encrypt(self, data: str) -> str:
-        return self.cipher.encrypt(data.encode()).decode()
-    
-    def decrypt(self, encrypted_data: str) -> str:
-        return self.cipher.decrypt(encrypted_data.encode()).decode()
-
-encryptor = DataEncryptor(ENCRYPTION_KEY)
-
-def generate_session_token() -> str:
-    alphabet = string.ascii_letters + string.digits
-    return ''.join(secrets.choice(alphabet) for _ in range(SESSION_TOKEN_LENGTH))
-
-def sanitize_input(text: str) -> str:
-    if not text:
-        return ""
-    text = html.escape(text.strip())
-    text = re.sub(r'<[^>]*>', '', text)
-    text = re.sub(r'[^\w\s\-.,!?@#\$%^&*()+=]', '', text)
-    return text
-
-def validate_name(name: str) -> str:
-    name = name.strip()
-    if not name:
-        raise ValueError("Name cannot be empty.")
-    if len(name) > 100:
-        raise ValueError("Name is too long (maximum 100 characters).")
-    if any(c.isdigit() for c in name):
-        raise ValueError("Name cannot contain numbers.")
-    return name
-
-def validate_age(age: Union[int, float, str]) -> int:
-    try:
-        age_int = int(age)
-        if not MIN_AGE <= age_int <= MAX_AGE:
-            raise ValueError(f"Age must be between {MIN_AGE} and {MAX_AGE}.")
-        return age_int
-    except (ValueError, TypeError):
-        raise ValueError("Please enter a valid age number.")
-
-def validate_file(file_obj) -> None:
-    if not file_obj:
-        raise ValueError("Please upload a file first")
-    
-    file_ext = os.path.splitext(file_obj.name)[1].lower()
-    if file_ext not in ALLOWED_FILE_TYPES:
-        raise ValueError(f"Invalid file type. Allowed types: {', '.join(ALLOWED_FILE_TYPES)}")
-    
-    file_size = os.path.getsize(file_obj.name) / (1024 * 1024)
-    if file_size > MAX_FILE_SIZE_MB:
-        raise ValueError(f"File too large. Maximum size is {MAX_FILE_SIZE_MB}MB.")
-
-def remove_sensitive_info(text: str) -> str:
-    patterns = [
-        (r'\b\d{3}-\d{2}-\d{4}\b', '[REDACTED-SSN]'),
-        (r'\b\d{6,9}\b', '[ID]'),
-        (r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b', '[EMAIL]'),
-        (r'\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b', '[IP]'),
-        (r'\b[A-Z][a-z]+ [A-Z][a-z]+\b', '[NAME]'),
-        (r'\b\d{3}\) \d{3}-\d{4}\b', '[PHONE]'),
-        (r'\b\d{1,5} [A-Z][a-z]+ [A-Z][a-z]+, [A-Z]{2} \d{5}\b', '[ADDRESS]')
-    ]
-    
-    for pattern, replacement in patterns:
-        text = re.sub(pattern, replacement, text)
-    return text
-
 # ========== GRADIO INTERFACE ==========
 def create_enhanced_interface():
     with gr.Blocks(theme=gr.themes.Soft(), title="Student Learning Assistant") as app: