Update app.py

app.py

@@ -56,7 +56,14 @@ def sanitize_input(text):
     # Allow alphanumerics and some punctuation, then truncate to 500 characters.
     clean_text = re.sub(r"[^a-zA-Z0-9\s.,!?@:\-]", "", text)
     return clean_text.strip()[:500]
-
+    
+def sanitize_prompt(text):
+    """
+    Sanitizes input to prevent harmful content and limits its length.
+    """
+    # Allow alphanumerics and some punctuation, then truncate to 500 characters.
+    clean_text = re.sub(r"[^a-zA-Z0-9\s.,!?@:\-]", "", text)
+    return clean_text.strip()[:8000]
 def validate_email(email):
     """
     Validates that the provided email is in a valid format.
@@ -86,7 +93,7 @@ def submit_prompt(email, name, system_prompt):
     email = sanitize_input(email)
     name = sanitize_input(name)
     print("UNSAINITIZED SYSTEM PROMPT", system_prompt)
-    system_prompt = sanitize_input(system_prompt)
+    system_prompt = sanitize_prompt(system_prompt)
     print("CLEANED SYSTEM PROMPT", system_prompt)
 
     score = 0