Add Gradio changes and email data CSV

app.py

@@ -1,6 +1,8 @@
 # filepath: /workspaces/internship1/app.py
 import uvicorn
 import os
+import gradio as gr
+import json
 
 # Import the FastAPI app instance from api.py
 # Ensure the FastAPI instance in api.py is named 'app'
@@ -14,7 +16,36 @@ except Exception as e:
     print(f"An unexpected error occurred during import: {e}")
     app = None
 
+# Assumption: 'process_email_request' is the function that takes email text
+# and returns the dictionary with keys like 'input_email_body', 'list_of_masked_entities', etc.
+# Adjust the import based on where this function is defined (e.g., utils.py, api.py)
+try:
+    from utils import process_email_request
+    print("Successfully imported process_email_request from utils.")
+except ImportError:
+    print("Could not import process_email_request from utils. Trying from api...")
+    try:
+        # If the logic is maybe within api.py but callable
+        from api import process_email_request # Adjust if the function name is different
+        print("Successfully imported process_email_request from api.")
+    except ImportError:
+        print("ERROR: Could not find the core processing function 'process_email_request'.")
+        print("Please ensure it's defined and importable from utils.py or api.py")
+        # Define a dummy function so Gradio doesn't crash immediately
+        def process_email_request(email_body):
+            return {"error": "Processing function not found.", "input_email_body": email_body}
+
+# Define the Gradio interface
+iface = gr.Interface(
+    fn=process_email_request,
+    inputs=gr.Textbox(lines=15, label="Input Email Body", placeholder="Paste email content here..."),
+    outputs=gr.JSON(label="Processing Results"),
+    title="Email Classification and PII Masking API",
+    description="Enter the body of an email below. The API will process it to mask PII entities (like names, emails) and classify the email's category. The results will be shown in JSON format.",
+    allow_flagging="never" # Optional: Disables flagging/saving examples
+)
 
+# Launch the interface
 if __name__ == "__main__":
     if app:
         # Get port from environment variable PORT, default to 8000
@@ -24,4 +55,5 @@ if __name__ == "__main__":
         print(f"Starting Uvicorn server on host 0.0.0.0, port {port}")
         uvicorn.run(app, host="0.0.0.0", port=port)
     else:
-        print("Could not start server because the FastAPI app instance was not loaded.")
+        print("Could not start server because the FastAPI app instance was not loaded.")
+    iface.launch() # Share=False is default, suitable for HF Spaces

combined_emails_with_natural_pii.csv

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b747a5dc74aa834811e194f9779655cd4f87ec30207d8cf63555f407966d6086
+size 13968052

requirements.txt

@@ -7,7 +7,7 @@ scikit-learn==1.4.2
 joblib==1.4.2
 
 # --- PII Masking ---
-spacy
+spacy>=3.7,<3.9 # Example: Pin spacy version if needed
 # The following model is downloaded by spacy CLI, but good to list
 # en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.1/en_core_web_sm-3.7.1-py3-none-any.whl
 
@@ -17,3 +17,6 @@ spacy
 
 # --- Development/Testing (Optional) ---
 # requests==2.31.0 # For testing the API or running generate_output.py
+
+# --- Additional Dependencies ---
+gradio # Add gradio