Update app.py

app.py

@@ -38,9 +38,9 @@ classifier = EmailClassifier(model_path=model_path)
 def process_email(email_body):
     """
     Process email by masking PII and classifying it.
-        Args:
+ Args:
         email_body: Raw email text
-    Returns:
+   Returns:
         tuple: (masked_email, entities_text, category)
     """
     # Mask PII
@@ -52,11 +52,10 @@ def process_email(email_body):
     # Classify email
     category = classifier.classify(processed_email)
 
-    # Format entities for display
-    entities_text = "\n".join([
-        f"- {entity['classification']}: {entity['entity']}"
-        for entity in entities
-    ])
+    # Format entities for display - fixed formatting
+    entities_text = ""
+    for entity in entities:
+        entities_text += f"- {entity['classification']}: {entity['entity']}\n"
 
     return masked_email, entities_text, category
 
@@ -98,7 +97,7 @@ demo = gr.Interface(
     ],
     article="""
     ## How It Works
-   1. **PII Masking**: The system identifies and masks personal information
+  1. **PII Masking**: The system identifies and masks personal information
     2. **Email Classification**: The masked email is classified into categories
     3. **Results**: View the masked version, detected PII, and email category
     """