Deploy Email Classifier API

Dockerfile

@@ -0,0 +1,30 @@
+FROM python:3.9-slim
+
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements first to leverage Docker cache
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY . .
+
+# Create directory for model files
+RUN mkdir -p /app/models
+
+# Expose port
+EXPOSE 7860
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Run the application
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

Synthetic_Email_Dataset.csv

@@ -0,0 +1,401 @@
+label,message,topic,intent,tone
+0,"Hello, You asked for it, so here is the notes. Warm wishes, David",notes,request,formal
+1,"Good morning, Just sharing the meeting agenda as requested. Cheers, Anna",document,send attachment,informal
+1,"Good morning, You asked for it, so here is the resume. Regards, Emily",resume,send attachment,formal
+0,"Dear team, As discussed, I'm sending the manual. Regards, Emily",manual,inform,informal
+1,"Hi all,Here are the meeting agenda, as promised. Warm wishes, David",meeting agenda,send attachment,informal
+1,"Dear team, You asked for it, so here is the technical log. Sincerely, Michael",technical log,send attachment,informal
+0,"Dear team, Attached is the error log you requested. Warm wishes, David",error log,inform,informal
+1,"Hi all, Kindly review the attached photos. Thanks, Sarah",photos,send attachment,informal
+0,"Dear team, Attached is the contract you requested. Cheers, Anna",contract,request,informal
+0,"Good afternoon, Please find attached the invoice. Sincerely, Michael",personal,request,informal
+1,"Dear team, Please find attached the report. Thanks, Sarah",report,send attachment,formal
+1,"Dear Paul, As discussed, I'm sending the presentation slides. Regards, Emily",presentation slides,send attachment,informal
+1,"Dear Hanna, You asked for it, so here is the itinerary. Regards, Emily",itinerary,send attachment,formal
+1,"Good morning, Just sharing the data analysis as requested. Best regards, John",data analysis,send attachment,informal
+0,"Hi, As discussed, I'm sending the presentation slides. Cheers, Anna",presentation slides,request,formal
+0,"Hi, Just sharing the error log as requested. Sincerely, Michael",error log,request,informal
+1,"Hi all, Kindly review the attached presentation slides. Regards, Emily",presentation slides,send attachment,informal
+0,"Good morning, As discussed, I'm sending the manual. Cheers, Anna",manual,request,formal
+0,"Hi, I've attached the notes for your review. Best regards, John",notes,request,formal
+1,"Good morning, Please find attached the technical log. Cheers, Anna",technical log,send attachment,informal
+0,"Hi all, Please find attached the presentation slides. Sincerely, Michael",presentation slides,inform,formal
+0,"Good afternoon,Here are the photos, as promised. Regards, Emily",photos,request,formal
+1,"Hi, As discussed, I'm sending the proposal. Warm wishes, David",proposal,send attachment,informal
+1,"Dear team, I've attached the project plan for your review. Best regards, John",project plan,send attachment,informal
+1,"Hello,Here are the data analysis, as promised. Warm wishes, David",data analysis,send attachment,informal
+1,"Hi, I've attached the notes for your review. Warm wishes, David",notes,send attachment,informal
+1,"Hello,Here are the presentation slides, as promised. Sincerely, Michael",presentation slides,send attachment,informal
+1,"Hi all, I've attached the brochure for your review. Regards, Emily",brochure,send attachment,informal
+1,"Hi all, Kindly review the attached photos. Best regards, John",photos,send attachment,formal
+1,"Good afternoon, Please find attached the itinerary. Thanks, Sarah",itinerary,send attachment,informal
+1,"Dear team,Here are the presentation slides, as promised. Thanks, Sarah",presentation slides,send attachment,informal
+0,"Dear Alexa, You asked for it, so here is the invoice. Best regards, John",invoice,request,formal
+1,"Hello, Just sharing the invoice as requested. Sincerely, Michael",invoice,send attachment,informal
+0,"Good afternoon, Attached is the meeting agenda you requested. Regards, Emily",meeting agenda,inform,formal
+1,"Hello,Here are the notes, as promised. Warm wishes, David",itinerary,send attachment,informal
+0,"Good morning, Just sharing the report as requested. Best regards, John",report,request,informal
+0,"Dear Haley, As discussed, I'm sending the assignment. Thanks, Sarah",assignment,request,informal
+1,"Hello, I've attached the notes for your review. Best regards, John",notes,send attachment,informal
+0,"Good afternoon, Please find attached the resume. Cheers, Anna",resume,request,formal
+1,"Good afternoon, Just sharing the error log as requested. Sincerely, Michael",project plan,send attachment,informal
+1,"Hi, Please find attached the manual. Warm wishes, David",manual,send attachment,formal
+1,"Hello, Kindly review the attached invoice. Cheers, Anna",invoice,send attachment,informal
+1,"Good morning, As discussed, I'm sending the invoice. Sincerely, Michael",invoice,send attachment,formal
+1,"Good morning, You asked for it, so here is the technical log. Sincerely, Michael",technical log,send attachment,formal
+1,"Good morning, I've attached the manual for your review. Warm wishes, David",manual,send attachment,formal
+0,"Dear Jessica, Attached is the resume you requested. Regards, Emily",resume,request,informal
+1,"Good morning, Just sharing the invoice as requested. Regards, Emily",invoice,send attachment,informal
+0,"Hi,Here are the invoice, as promised. Cheers, Anna",invoice,request,informal
+1,"Good morning, Kindly review the attached error log. Best regards, John",error log,send attachment,informal
+1,"Hi, Please find attached the meeting agenda. Regards, Emily",spreadsheet,send attachment,informal
+1,"Dear team, I've attached the error log for your review. Cheers, Anna",error log,send attachment,informal
+0,"Hi all, I've attached the presentation slides for your review. Regards, Emily",presentation slides,inform,informal
+1,"Good morning, You asked for it, so here is the report. Thanks, Sarah",report,send attachment,formal
+1,"Dear team, Please find attached the error log. Warm wishes, David",error log,send attachment,formal
+0,"Hi, Please find attached the invoice. Best regards, John",invoice,inform,informal
+0,"Hello, As discussed, I'm sending the invoice. Warm wishes, David",invoice,request,informal
+1,"Good afternoon, Kindly review the attached spreadsheet. Warm wishes, David",spreadsheet,send attachment,formal
+1,"Hello, Kindly review the attached manual. Sincerely, Michael",manual,send attachment,formal
+1,"Dear Mat, You asked for it, so here is the resume. Thanks, Sarah",resume,send attachment,informal
+0,"Hello, I've attached the invoice for your review. Regards, Emily",invoice,request,formal
+1,"Dear Anna, As discussed, I'm sending the data analysis. Regards, Emily",data analysis,send attachment,informal
+1,"Good morning, As discussed, I'm sending the manual. Best regards, John",manual,send attachment,formal
+0,"Hi, Please find attached the spreadsheet. Warm wishes, David",spreadsheet,inform,formal
+1,"Hello, Just sharing the notes as requested. Regards, Emily",notes,send attachment,informal
+1,"Hi all, Just sharing the meeting agenda as requested. Cheers, Anna",meeting agenda,send attachment,formal
+1,"Good afternoon, Attached is the brochure you requested. Best regards, John",brochure,send attachment,informal
+1,"Hi, You asked for it, so here is the data analysis. Thanks, Sarah",data analysis,send attachment,formal
+1,"Dear Kai, Please find attached the technical log. Regards, Emily",technical log,send attachment,informal
+1,"Hello, Kindly review the attached brochure. Best regards, John",brochure,send attachment,formal
+1,"Hi, Attached is the manual you requested. Warm wishes, David",manual,send attachment,formal
+1,"Hi all, Attached is the brochure you requested. Thanks, Sarah",brochure,send attachment,informal
+0,"Good afternoon, Please find attached the manual. Cheers, Anna",project plan,inform,informal
+1,"Hello, I've attached the technical log for your review. Warm wishes, David",technical log,send attachment,formal
+1,"Dear team, Attached is the resume you requested. Cheers, Anna",resume,send attachment,informal
+1,"Dear Chanyeol, Please find attached the technical log. Sincerely, Michael",update,send attachment,informal
+1,"Hi all, As discussed, I'm sending the proposal. Regards, Emily",proposal,send attachment,informal
+1,"Good morning, Kindly review the attached invoice. Sincerely, Michael",invoice,send attachment,formal
+1,"Hi all, Kindly review the attached presentation slides. Warm wishes, David",presentation slides,send attachment,formal
+1,"Hi, You asked for it, so here is the meeting agenda. Best regards, John",meeting agenda,send attachment,formal
+0,"Dear Michale, Just sharing the brochure as requested. Warm wishes, David",brochure,request,formal
+0,"Hello, As discussed, I'm sending the brochure. Sincerely, Michael",brochure,inform,formal
+0,"Hi, Just sharing the manual as requested. Thanks, Sarah",manual,inform,formal
+0,"Dear team,Here are the data analysis, as promised. Warm wishes, David",data analysis,request,formal
+0,"Dear ,Here are the meeting agenda, as promised. Warm wishes, David",meeting agenda,inform,formal
+1,"Good afternoon, Attached is the itinerary you requested. Regards, Emily",itinerary,send attachment,informal
+1,"Dear team, You asked for it, so here is the manual. Regards, Emily",manual,send attachment,formal
+1,"Hello, Attached is the resume you requested. Sincerely, Michael",resume,send attachment,informal
+1,"Dear Cur, You asked for it, so here is the itinerary. Best regards, John",itinerary,send attachment,informal
+1,"Hello, As discussed, I'm sending the technical log. Sincerely, Michael",technical log,send attachment,formal
+0,"Good morning, You asked for it, so here is the meeting agenda. Best regards, John",meeting agenda,request,informal
+0,"Good morning,Here are the data analysis, as promised. Best regards, John",brochure,inform,formal
+1,"Dear , You asked for it, so here is the assignment. Regards, Emily",assignment,send attachment,informal
+1,"Good afternoon, As discussed, I'm sending the invoice. Thanks, Sarah",contract,send attachment,informal
+1,"Hi, I've attached the proposal for your review. Regards, Emily",proposal,send attachment,formal
+0,"Good afternoon, Kindly review the attached assignment. Warm wishes, David",assignment,inform,formal
+1,"Dear team, I've attached the project plan for your review. Regards, Emily",project plan,send attachment,informal
+1,"Hi, You asked for it, so here is the spreadsheet. Best regards, John",spreadsheet,send attachment,formal
+1,"Dear Jes, You asked for it, so here is the technical log. Regards, Emily",technical log,send attachment,informal
+1,"Good morning, Attached is the photos you requested. Cheers, Anna",photos,send attachment,informal
+1,"Hi all, Just sharing the report as requested. Best regards, John",personal,send attachment,informal
+1,"Hi, You asked for it, so here is the proposal. Regards, Emily",manual,send attachment,formal
+0,"Good afternoon, You asked for it, so here is the manual. Thanks, Sarah",manual,inform,informal
+1,"Dear Josh, As discussed, I'm sending the invoice. Warm wishes, David",invoice,send attachment,formal
+1,"Good morning, Kindly review the attached proposal. Sincerely, Michael",assignment,send attachment,formal
+1,"Hello, Just sharing the brochure as requested. Regards, Emily",brochure,send attachment,formal
+1,"Good morning, Kindly review the attached itinerary. Sincerely, Michael",itinerary,send attachment,formal
+1,"Dear team, Kindly review the attached photos. Regards, Emily",photos,send attachment,formal
+1,"Dear Berlin,Here are the itinerary, as promised. Warm wishes, David",itinerary,send attachment,formal
+1,"Hi all, Kindly review the attached assignment. Warm wishes, David",assignment,send attachment,informal
+1,"Hi all, Please find attached the spreadsheet. Thanks, Sarah",spreadsheet,send attachment,formal
+0,"Good afternoon, As discussed, I'm sending the itinerary. Warm wishes, David",itinerary,request,formal
+1,"Dear Alex, Attached is the invoice you requested. Thanks, Sarah",invoice,send attachment,formal
+0,"Hello, You asked for it, so here is the presentation slides. Sincerely, Michael",presentation slides,inform,informal
+1,"Hello,Here are the project plan, as promised. Best regards, John",update,send attachment,formal
+1,"Dear Susan, Please find attached the notes. Sincerely, Michael",notes,send attachment,formal
+1,"Dear , As discussed, I'm sending the photos. Sincerely, Michael",photos,send attachment,formal
+1,"Hi,Here are the itinerary, as promised. Sincerely, Michael",itinerary,send attachment,formal
+0,"Dear , Kindly review the attached spreadsheet. Thanks, Sarah",spreadsheet,request,informal
+1,"Hello, I've attached the spreadsheet for your review. Sincerely, Michael",spreadsheet,send attachment,informal
+1,"Dear , Attached is the report you requested. Cheers, Anna",report,send attachment,formal
+1,"Hi, Attached is the brochure you requested. Sincerely, Michael",brochure,send attachment,formal
+1,"Hello, You asked for it, so here is the presentation slides. Sincerely, Michael",presentation slides,send attachment,informal
+1,"Good morning, Just sharing the brochure as requested. Best regards, John",brochure,send attachment,informal
+1,"Hi, Kindly review the attached resume. Thanks, Sarah",resume,send attachment,informal
+1,"Hi, Attached is the meeting agenda you requested. Warm wishes, David",meeting agenda,send attachment,formal
+1,"Hello, As discussed, I'm sending the assignment. Sincerely, Michael",assignment,send attachment,informal
+0,"Hi all, Please find attached the resume. Warm wishes, David",resume,request,formal
+0,"Hello,Here are the technical log, as promised. Best regards, John",technical log,inform,formal
+1,"Hi all, Just sharing the manual as requested. Thanks, Sarah",manual,send attachment,informal
+0,"Good afternoon,Here are the technical log, as promised. Sincerely, Michael",technical log,request,formal
+1,"Dear , Kindly review the attached contract. Warm wishes, David",contract,send attachment,informal
+1,"Good afternoon, I've attached the brochure for your review. Regards, Emily",brochure,send attachment,formal
+1,"Dear team, You asked for it, so here is the notes. Sincerely, Michael",notes,send attachment,formal
+1,"Good afternoon, I've attached the technical log for your review. Thanks, Sarah",technical log,send attachment,informal
+1,"Hi all, Kindly review the attached brochure. Best regards, John",brochure,send attachment,informal
+0,"Hello, Attached is the notes you requested. Regards, Emily",notes,inform,formal
+1,"Dear , Please find attached the notes. Best regards, John",notes,send attachment,formal
+1,"Dear team,Here are the notes, as promised. Sincerely, Michael",notes,send attachment,formal
+1,"Dear team, Just sharing the invoice as requested. Best regards, John",invoice,send attachment,formal
+1,"Dear team, You asked for it, so here is the photos. Thanks, Sarah",photos,send attachment,informal
+1,"Dear team, Kindly review the attached manual. Best regards, John",manual,send attachment,informal
+0,"Hi, Please find attached the resume. Warm wishes, David",resume,request,informal
+0,"Hello, Just sharing the project plan as requested. Cheers, Anna",project plan,request,formal
+1,"Hi all, Just sharing the brochure as requested. Warm wishes, David",brochure,send attachment,formal
+1,"Hello, I've attached the resume for your review. Thanks, Sarah",resume,send attachment,informal
+1,"Dear , You asked for it, so here is the contract. Warm wishes, David",contract,send attachment,formal
+1,"Hi all,Here are the meeting agenda, as promised. Warm wishes, David",meeting agenda,send attachment,formal
+0,"Good morning,Here are the notes, as promised. Regards, Emily",notes,inform,formal
+1,"Dear , I've attached the resume for your review. Best regards, John",resume,send attachment,informal
+0,"Hello, Kindly review the attached report. Sincerely, Michael",report,request,informal
+1,"Hi, I've attached the error log for your review. Cheers, Anna",error log,send attachment,informal
+1,"Hi all, Attached is the data analysis you requested. Sincerely, Michael",data analysis,send attachment,informal
+0,"Dear , As discussed, I'm sending the assignment. Warm wishes, David",assignment,inform,informal
+0,"Good afternoon, Please find attached the meeting agenda. Thanks, Sarah",meeting agenda,inform,informal
+1,"Good afternoon, You asked for it, so here is the assignment. Best regards, John",assignment,send attachment,informal
+1,"Good morning, You asked for it, so here is the presentation slides. Sincerely, Michael",presentation slides,send attachment,formal
+1,"Hi, You asked for it, so here is the meeting agenda. Cheers, Anna",meeting agenda,send attachment,informal
+1,"Good afternoon, Attached is the contract you requested. Warm wishes, David",contract,send attachment,informal
+1,"Dear , Attached is the spreadsheet you requested. Thanks, Sarah",spreadsheet,send attachment,formal
+1,"Hi all, You asked for it, so here is the itinerary. Sincerely, Michael",itinerary,send attachment,formal
+1,"Good afternoon, Please find attached the meeting agenda. Warm wishes, David",meeting agenda,send attachment,informal
+1,"Hi all,Here are the data analysis, as promised. Best regards, John",data analysis,send attachment,formal
+1,"Hello,Here are the technical log, as promised. Regards, Emily",technical log,send attachment,formal
+0,"Hi all, I've attached the notes for your review. Warm wishes, David",assignment,inform,informal
+1,"Good afternoon, As discussed, I'm sending the error log. Cheers, Anna",error log,send attachment,formal
+1,"Good morning,Here are the photos, as promised. Regards, Emily",photos,send attachment,informal
+1,"Dear team,Here are the spreadsheet, as promised. Best regards, John",spreadsheet,send attachment,informal
+1,"Good morning, Kindly review the attached itinerary. Thanks, Sarah",itinerary,send attachment,informal
+1,"Good afternoon, You asked for it, so here is the presentation slides. Thanks, Sarah",presentation slides,send attachment,formal
+1,"Hi all, Attached is the meeting agenda you requested. Thanks, Sarah",spreadsheet,send attachment,informal
+1,"Dear , You asked for it, so here is the proposal. Warm wishes, David",proposal,send attachment,informal
+1,"Hi all, You asked for it, so here is the technical log. Regards, Emily",technical log,send attachment,informal
+1,"Hello, I've attached the brochure for your review. Regards, Emily",brochure,send attachment,formal
+1,"Hi all, Kindly review the attached resume. Regards, Emily",resume,send attachment,informal
+1,"Dear , Just sharing the spreadsheet as requested. Regards, Emily",spreadsheet,send attachment,formal
+1,"Hi all, You asked for it, so here is the report. Regards, Emily",report,send attachment,formal
+1,"Dear , Attached is the photos you requested. Warm wishes, David",photos,send attachment,informal
+0,"Dear team, I've attached the presentation slides for your review. Thanks, Sarah",presentation slides,inform,informal
+1,"Dear , Just sharing the report as requested. Thanks, Sarah",report,send attachment,informal
+1,"Good afternoon, Just sharing the proposal as requested. Best regards, John",proposal,send attachment,informal
+1,"Hello, Please find attached the presentation slides. Thanks, Sarah",presentation slides,send attachment,informal
+1,"Dear , I've attached the notes for your review. Cheers, Anna",technical log,send attachment,formal
+1,"Good afternoon, Attached is the report you requested. Cheers, Anna",report,send attachment,informal
+1,"Hi all, Kindly review the attached proposal. Warm wishes, David",proposal,send attachment,formal
+1,"Hi all, Kindly review the attached contract. Warm wishes, David",project plan,send attachment,formal
+1,"Dear team, As discussed, I'm sending the notes. Thanks, Sarah",notes,send attachment,formal
+1,"Hello, I've attached the proposal for your review. Cheers, Anna",proposal,send attachment,formal
+1,"Hello, Attached is the spreadsheet you requested. Best regards, John",spreadsheet,send attachment,formal
+0,"Dear team, You asked for it, so here is the presentation slides. Cheers, Anna",presentation slides,request,formal
+1,"Good afternoon, As discussed, I'm sending the data analysis. Thanks, Sarah",data analysis,send attachment,informal
+1,"Dear team, Just sharing the manual as requested. Thanks, Sarah",manual,send attachment,informal
+1,"Hi all, Please find attached the assignment. Cheers, Anna",assignment,send attachment,formal
+1,"Hi, Please find attached the itinerary. Sincerely, Michael",itinerary,send attachment,informal
+1,"Hi all, Attached is the technical log you requested. Sincerely, Michael",technical log,send attachment,formal
+1,"Hello, You asked for it, so here is the report. Regards, Emily",report,send attachment,formal
+0,"Dear team, Just sharing the resume as requested. Thanks, Sarah",resume,inform,formal
+1,"Hi,Here are the manual, as promised. Cheers, Anna",manual,send attachment,formal
+1,"Good afternoon, Please find attached the presentation slides. Regards, Emily",spreadsheet,send attachment,informal
+0,"Good morning,Here are the assignment, as promised. Regards, Emily",assignment,request,informal
+1,"Good afternoon, I've attached the presentation slides for your review. Cheers, Anna",presentation slides,send attachment,informal
+0,"Hi all, Please find attached the technical log. Thanks, Sarah",technical log,request,formal
+0,"Good afternoon, Kindly review the attached contract. Cheers, Anna",contract,inform,formal
+1,"Dear , Just sharing the data analysis as requested. Warm wishes, David",data analysis,send attachment,formal
+1,"Hi, You asked for it, so here is the contract. Warm wishes, David",contract,send attachment,informal
+1,"Good afternoon, You asked for it, so here is the resume. Sincerely, Michael",resume,send attachment,formal
+1,"Good afternoon, I've attached the report for your review. Thanks, Sarah",report,send attachment,formal
+1,"Good morning, Please find attached the notes. Regards, Emily",notes,send attachment,formal
+1,"Good morning, Just sharing the technical log as requested. Regards, Emily",brochure,send attachment,formal
+0,"Hi, I've attached the technical log for your review. Warm wishes, David",technical log,request,informal
+1,"Good morning, Kindly review the attached itinerary. Cheers, Anna",itinerary,send attachment,informal
+1,"Dear , Attached is the resume you requested. Warm wishes, David",resume,send attachment,informal
+1,"Hello,Here are the manual, as promised. Best regards, John",manual,send attachment,informal
+0,"Dear team, Attached is the manual you requested. Sincerely, Michael",manual,request,informal
+1,"Good morning, Kindly review the attached meeting agenda. Cheers, Anna",meeting agenda,send attachment,formal
+0,"Dear team, Please find attached the report. Best regards, John",report,inform,informal
+0,"Dear team, I've attached the itinerary for your review. Best regards, John",itinerary,inform,formal
+0,"Hello, Attached is the contract you requested. Sincerely, Michael",contract,inform,formal
+1,"Hi all,Here are the meeting agenda, as promised. Warm wishes, David",meeting agenda,send attachment,informal
+1,"Dear , You asked for it, so here is the manual. Cheers, Anna",manual,send attachment,formal
+0,"Dear , Please find attached the assignment. Best regards, John",brochure,request,formal
+1,"Hi, Please find attached the manual. Warm wishes, David",manual,send attachment,informal
+1,"Hello, As discussed, I'm sending the data analysis. Regards, Emily",data analysis,send attachment,formal
+0,"Hi, As discussed, I'm sending the assignment. Thanks, Sarah",assignment,request,informal
+0,"Hi all, Attached is the itinerary you requested. Sincerely, Michael",itinerary,inform,formal
+1,"Good afternoon, Attached is the presentation slides you requested. Cheers, Anna",presentation slides,send attachment,formal
+1,"Dear team, As discussed, I'm sending the invoice. Warm wishes, David",invoice,send attachment,informal
+1,"Hi, Please find attached the itinerary. Cheers, Anna",itinerary,send attachment,formal
+0,"Hello, Please find attached the notes. Thanks, Sarah",notes,inform,formal
+1,"Hello, As discussed, I'm sending the photos. Warm wishes, David",photos,send attachment,informal
+1,"Good morning, Attached is the error log you requested. Sincerely, Michael",error log,send attachment,formal
+1,"Dear , Kindly review the attached manual. Warm wishes, David",manual,send attachment,formal
+1,"Good morning, Attached is the photos you requested. Thanks, Sarah",photos,send attachment,formal
+1,"Dear , Kindly review the attached error log. Thanks, Sarah",error log,send attachment,formal
+1,"Dear team, Attached is the itinerary you requested. Thanks, Sarah",itinerary,send attachment,formal
+0,"Hello, Attached is the technical log you requested. Cheers, Anna",technical log,inform,informal
+1,"Hello,Here are the resume, as promised. Best regards, John",resume,send attachment,formal
+1,"Good morning,Here are the spreadsheet, as promised. Thanks, Sarah",spreadsheet,send attachment,formal
+0,"Hi, I've attached the technical log for your review. Thanks, Sarah",technical log,inform,formal
+0,"Hi all, Kindly review the attached error log. Best regards, John",error log,request,formal
+1,"Hello, Please find attached the resume. Cheers, Anna",resume,send attachment,formal
+1,"Dear , Just sharing the assignment as requested. Cheers, Anna",assignment,send attachment,informal
+0,"Dear , As discussed, I'm sending the resume. Best regards, John",resume,request,formal
+0,"Dear team, I've attached the spreadsheet for your review. Best regards, John",spreadsheet,inform,informal
+1,"Dear , As discussed, I'm sending the brochure. Regards, Emily",brochure,send attachment,formal
+1,"Dear , Attached is the invoice you requested. Sincerely, Michael",invoice,send attachment,formal
+1,"Good morning, Kindly review the attached itinerary. Cheers, Anna",itinerary,send attachment,formal
+1,"Dear team, Please find attached the assignment. Sincerely, Michael",assignment,send attachment,informal
+0,"Hello, I've attached the report for your review. Sincerely, Michael",report,request,informal
+0,"Dear , Just sharing the assignment as requested. Best regards, John",assignment,request,informal
+1,"Dear team, Please find attached the report. Thanks, Sarah",report,send attachment,formal
+0,"Good morning,Here are the contract, as promised. Regards, Emily",contract,inform,informal
+1,"Dear team, Please find attached the spreadsheet. Warm wishes, David",spreadsheet,send attachment,informal
+1,"Hi all, As discussed, I'm sending the spreadsheet. Thanks, Sarah",spreadsheet,send attachment,informal
+1,"Dear team,Here are the data analysis, as promised. Best regards, John",data analysis,send attachment,informal
+1,"Good morning, Kindly review the attached photos. Sincerely, Michael",photos,send attachment,informal
+1,"Good morning, I've attached the contract for your review. Thanks, Sarah",contract,send attachment,formal
+1,"Hi, Just sharing the invoice as requested. Cheers, Anna",invoice,send attachment,informal
+0,"Hello, Just sharing the proposal as requested. Cheers, Anna",proposal,request,informal
+1,"Good afternoon,Here are the invoice, as promised. Best regards, John",invoice,send attachment,informal
+0,"Dear , Just sharing the resume as requested. Cheers, Anna",resume,inform,formal
+0,"Dear team, You asked for it, so here is the proposal. Cheers, Anna",proposal,request,informal
+0,"Hello, Attached is the assignment you requested. Thanks, Sarah",assignment,inform,formal
+0,"Hi all,Here are the invoice, as promised. Regards, Emily",invoice,request,formal
+0,"Good morning, You asked for it, so here is the notes. Warm wishes, David",notes,request,formal
+0,"Hi all, Please find attached the presentation slides. Warm wishes, David",presentation slides,request,informal
+0,"Good morning, You asked for it, so here is the itinerary. Best regards, John",itinerary,request,informal
+1,"Good afternoon, You asked for it, so here is the proposal. Sincerely, Michael",presentation slides,send attachment,informal
+1,"Good afternoon, Kindly review the attached photos. Cheers, Anna",photos,send attachment,informal
+0,"Dear team, Just sharing the meeting agenda as requested. Warm wishes, David",meeting agenda,inform,formal
+0,"Hello, Please find attached the data analysis. Thanks, Sarah",data analysis,inform,formal
+1,"Good morning, Please find attached the resume. Best regards, John",meeting agenda,send attachment,informal
+0,"Good afternoon,Here are the technical log, as promised. Sincerely, Michael",technical log,request,informal
+0,"Hello, You asked for it, so here is the meeting agenda. Sincerely, Michael",meeting agenda,request,formal
+1,"Hi all, Attached is the project plan you requested. Warm wishes, David",project plan,send attachment,formal
+1,"Hi all, Please find attached the resume. Warm wishes, David",resume,send attachment,formal
+1,"Hi, Kindly review the attached photos. Sincerely, Michael",photos,send attachment,informal
+1,"Good afternoon, Kindly review the attached presentation slides. Thanks, Sarah",presentation slides,send attachment,informal
+1,"Hi all, As discussed, I'm sending the contract. Cheers, Anna",contract,send attachment,informal
+0,"Good afternoon, Please find attached the technical log. Sincerely, Michael",technical log,request,formal
+0,"Hi all, Please find attached the itinerary. Best regards, John",itinerary,request,informal
+1,"Hello,Here are the report, as promised. Sincerely, Michael",report,send attachment,formal
+1,"Hello, As discussed, I'm sending the assignment. Warm wishes, David",assignment,send attachment,informal
+1,"Hi all, I've attached the technical log for your review. Regards, Emily",technical log,send attachment,formal
+0,"Good afternoon, Attached is the notes you requested. Regards, Emily",notes,inform,informal
+1,"Hello, You asked for it, so here is the invoice. Warm wishes, David",invoice,send attachment,formal
+0,"Dear team,Here are the itinerary, as promised. Warm wishes, David",itinerary,inform,informal
+1,"Hi all,Here are the brochure, as promised. Sincerely, Michael",brochure,send attachment,formal
+1,"Dear team,Here are the invoice, as promised. Thanks, Sarah",invoice,send attachment,informal
+1,"Hi, You asked for it, so here is the invoice. Thanks, Sarah",invoice,send attachment,informal
+1,"Hello, As discussed, I'm sending the manual. Thanks, Sarah",manual,send attachment,formal
+0,"Good afternoon,Here are the spreadsheet, as promised. Best regards, John",spreadsheet,inform,formal
+0,"Dear Rio, You asked for it, so here is the contract. Best regards, John",contract,inform,formal
+1,"Hi, Attached is the presentation slides you requested. Cheers, Anna",presentation slides,send attachment,informal
+1,"Good morning, Kindly review the attached notes. Best regards, John",notes,send attachment,formal
+1,"Dear team, You asked for it, so here is the notes. Cheers, Anna",proposal,send attachment,formal
+1,"Good afternoon, As discussed, I'm sending the project plan. Best regards, John",project plan,send attachment,formal
+0,"Good afternoon, You asked for it, so here is the resume. Best regards, John",resume,request,formal
+1,"Good morning,Here are the itinerary, as promised. Regards, Emily",itinerary,send attachment,formal
+1,"Dear Domic, Kindly review the attached notes. Regards, Emily",notes,send attachment,informal
+1,"Hello, Attached is the error log you requested. Sincerely, Michael",error log,send attachment,formal
+1,"Good afternoon, You asked for it, so here is the brochure. Sincerely, Michael",brochure,send attachment,informal
+0,"Hi all, Just sharing the spreadsheet as requested. Sincerely, Michael",spreadsheet,inform,informal
+0,"Good afternoon,Here are the proposal, as promised. Thanks, Sarah",proposal,inform,informal
+0,"Good morning, I've attached the data analysis for your review. Cheers, Anna",data analysis,inform,formal
+1,"Hi, Just sharing the presentation slides as requested. Sincerely, Michael",presentation slides,send attachment,formal
+1,"Hi, I've attached the manual for your review. Sincerely, Michael",manual,send attachment,formal
+0,"Hi, Just sharing the assignment as requested. Regards, Emily",assignment,inform,informal
+1,"Hi, Just sharing the project plan as requested. Thanks, Sarah",project plan,send attachment,formal
+1,"Hi, As discussed, I'm sending the proposal. Cheers, Anna",proposal,send attachment,informal
+0,"Hi all, Attached is the notes you requested. Thanks, Sarah",notes,request,informal
+0,"Dear ,Here are the invoice, as promised. Thanks, Sarah",invoice,inform,formal
+1,"Good morning,Here are the invoice, as promised. Warm wishes, David",invoice,send attachment,formal
+1,"Good afternoon, You asked for it, so here is the assignment. Warm wishes, David",itinerary,send attachment,formal
+0,"Hi all, Attached is the notes you requested. Thanks, Sarah",notes,inform,formal
+0,"Good morning, Kindly review the attached manual. Regards, Emily",manual,inform,formal
+1,"Hello, Kindly review the attached contract. Warm wishes, David",contract,send attachment,informal
+1,"Dear team, As discussed, I'm sending the contract. Cheers, Anna",contract,send attachment,formal
+1,"Hi all, I've attached the meeting agenda for your review. Warm wishes, David",document,send attachment,formal
+1,"Hi, Kindly review the attached photos. Best regards, John",photos,send attachment,formal
+1,"Hi, As discussed, I'm sending the error log. Best regards, John",error log,send attachment,formal
+0,"Dear team, I've attached the data analysis for your review. Best regards, John",data analysis,request,informal
+0,"Dear team, Just sharing the data analysis as requested. Thanks, Sarah",data analysis,inform,formal
+1,"Hi, Please find attached the report. Best regards, John",report,send attachment,informal
+0,"Hi all, As discussed, I'm sending the spreadsheet. Cheers, Anna",spreadsheet,request,formal
+0,"Dear , Please find attached the presentation slides. Cheers, Anna",presentation slides,request,informal
+1,"Hello, You asked for it, so here is the itinerary. Best regards, John",itinerary,send attachment,informal
+1,"Dear Mic, Just sharing the itinerary as requested. Thanks, Sarah",itinerary,send attachment,informal
+1,"Good afternoon,Here are the assignment, as promised. Cheers, Anna",assignment,send attachment,informal
+0,"Hello, Attached is the project plan you requested. Best regards, John",project plan,request,informal
+1,"Dear Cheerish, I've attached the spreadsheet for your review. Sincerely, Michael",spreadsheet,send attachment,informal
+0,"Dear Bill, As discussed, I'm sending the report. Regards, Emily",report,inform,informal
+0,"Dear team, I've attached the contract for your review. Thanks, Sarah",contract,request,informal
+1,"Hello, Just sharing the brochure as requested. Cheers, Anna",brochure,send attachment,informal
+1,"Dear Lana, You asked for it, so here is the error log. Warm wishes, David",error log,send attachment,formal
+0,"Hi all, Please find attached the manual. Best regards, John",manual,inform,informal
+0,"Dear Taylor, Attached is the itinerary you requested. Thanks, Sarah",itinerary,request,informal
+0,"Hello, Attached is the contract you requested. Best regards, John",contract,inform,informal
+0,"Good morning, You asked for it, so here is the data analysis. Warm wishes, David",data analysis,inform,informal
+1,"Dear team, Attached is the technical log you requested. Regards, Emily",manual,send attachment,informal
+1,"Good morning, Kindly review the attached project plan. Regards, Emily",project plan,send attachment,formal
+0,"Dear Justin, You asked for it, so here is the resume. Cheers, Anna",resume,inform,informal
+1,"Good afternoon,Here are the assignment, as promised. Regards, Emily",assignment,send attachment,formal
+1,"Hi, You asked for it, so here is the invoice. Best regards, John",invoice,send attachment,informal
+1,"Good afternoon,Here are the photos, as promised. Sincerely, Michael",itinerary,send attachment,informal
+0,"Hi, Just sharing the notes as requested. Warm wishes, David",notes,request,informal
+0,"Dear Weekend, Kindly review the attached proposal. Thanks, Sarah",proposal,request,informal
+0,"Good morning, You asked for it, so here is the assignment. Regards, Emily",assignment,inform,formal
+1,"Hi all, Just sharing the resume as requested. Regards, Emily",resume,send attachment,formal
+0,"Good morning, Kindly review the attached manual. Regards, Emily",manual,request,informal
+0,"Hi all, Attached is the technical log you requested. Warm wishes, David",technical log,inform,formal
+1,"Good morning, I've attached the spreadsheet for your review. Regards, Emily",spreadsheet,send attachment,formal
+0,"Good afternoon, As discussed, I'm sending the proposal. Thanks, Sarah",proposal,inform,formal
+1,"Dear Bernard, Kindly review the attached notes. Sincerely, Michael",notes,send attachment,informal
+1,"Good morning, Kindly review the attached invoice. Regards, Emily",invoice,send attachment,informal
+1,"Hi all, Kindly review the attached meeting agenda. Sincerely, Michael",meeting agenda,send attachment,informal
+1,"Hi, Just sharing the contract as requested. Best regards, John",contract,send attachment,formal
+0,"Hi all, As discussed, I'm sending the report. Warm wishes, David",report,inform,formal
+0,"Good morning, You asked for it, so here is the project plan. Sincerely, Michael",project plan,inform,formal
+0,"Dear team, Just sharing the assignment as requested. Thanks, Sarah",assignment,inform,informal
+1,"Good afternoon, Kindly review the attached proposal. Warm wishes, David",proposal,send attachment,informal
+0,"Dear Joe, Just sharing the manual as requested. Warm wishes, David",manual,request,informal
+0,"Dear Josh, Kindly review the attached data analysis. Sincerely, Michael",data analysis,inform,informal
+1,"Hi, I've attached the report for your review. Thanks, Sarah",report,send attachment,informal
+1,"Hi, I've attached the project plan for your review. Best regards, John",project plan,send attachment,formal
+0,"Hi, You asked for it, so here is the assignment. Cheers, Anna",assignment,inform,informal
+1,"Good afternoon,Here are the contract, as promised. Best regards, John",contract,send attachment,informal
+0,"Hi, Just sharing the notes as requested. Thanks, Sarah",notes,request,formal
+1,"Hello, You asked for it, so here is the photos. Thanks, Sarah",spreadsheet,send attachment,formal
+1,"Hi all, Kindly review the attached brochure. Thanks, Sarah",brochure,send attachment,formal
+0,"Dear Jolie, I've attached the assignment for your review. Sincerely, Michael",project plan,inform,informal
+1,"Dear William, Please find attached the data analysis. Thanks, Sarah",data analysis,send attachment,formal
+0,"Good morning, You asked for it, so here is the meeting agenda. Regards, Emily",meeting agenda,inform,informal
+1,"Hello, As discussed, I'm sending the invoice. Sincerely, Michael",invoice,send attachment,informal
+1,"Dear Will,Here are the itinerary, as promised. Regards, Emily",itinerary,send attachment,formal
+1,"Hello, As discussed, I'm sending the photos. Best regards, John",photos,send attachment,informal
+1,"Hi, Just sharing the meeting agenda as requested. Best regards, John",meeting agenda,send attachment,formal
+1,"Hi all,Here are the presentation slides, as promised. Regards, Emily",presentation slides,send attachment,informal
+1,"Hi, You asked for it, so here is the spreadsheet. Thanks, Sarah",spreadsheet,send attachment,informal
+1,"Hi all, Please find attached the resume. Sincerely, Michael",technical log,send attachment,formal
+1,"Hi, Please find attached the itinerary. Cheers, Anna",itinerary,send attachment,formal
+1,"Dear team,Here are the assignment, as promised. Best regards, John",assignment,send attachment,formal
+0,"Hi all, You asked for it, so here is the meeting agenda. Regards, Emily",meeting agenda,inform,informal
+1,"Dear team, Attached is the manual you requested. Sincerely, Michael",manual,send attachment,informal
+1,"Good morning, Please find attached the resume. Regards, Emily",resume,send attachment,formal
+1,"Good morning, You asked for it, so here is the notes. Warm wishes, David",notes,send attachment,informal
+0,"Dear Jas, Please find attached the technical log. Sincerely, Michael",technical log,request,formal
+0,"Dear team, You asked for it, so here is the report. Warm wishes, David",report,request,informal
+1,"Dear Leo, As discussed, I'm sending the data analysis. Sincerely, Michael",data analysis,send attachment,informal
+1,"Dear Lio, Attached is the spreadsheet you requested. Thanks, Sarah",spreadsheet,send attachment,formal
+1,"Hi all, As discussed, I'm sending the presentation slides. Cheers, Anna",presentation slides,send attachment,informal
+1,"Hi, Attached is the brochure you requested. Best regards, John",brochure,send attachment,informal
+1,"Hello, As discussed, I'm sending the presentation slides. Regards, Emily",presentation slides,send attachment,informal
+0,"Hi, Please find attached the project plan. Warm wishes, David",project plan,inform,formal
+0,"Good afternoon,Here are the photos, as promised. Best regards, John",photos,inform,formal
+1,"Good afternoon, As discussed, I'm sending the brochure. Regards, Emily",brochure,send attachment,formal
+1,"Dear Nah, Please find attached the manual. Best regards, John",manual,send attachment,informal
+0,"Hello, Kindly review the attached assignment. Thanks, Sarah",assignment,request,informal
+1,"Hello, Please find attached the photos. Warm wishes, David",photos,send attachment,formal
+1,"Dear Nick, Please find attached the project plan. Cheers, Anna",project plan,send attachment,formal
+1,"Hi, Attached is the contract you requested. Best regards, John",contract,send attachment,formal

app.py

@@ -0,0 +1,339 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import HTMLResponse
+from pydantic import BaseModel
+import joblib
+import pandas as pd
+import numpy as np
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.naive_bayes import MultinomialNB
+from sklearn.pipeline import Pipeline
+from sklearn.model_selection import train_test_split
+from sklearn.metrics import classification_report, accuracy_score
+import re
+import os
+from typing import List, Dict, Any
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# Initialize FastAPI app
+app = FastAPI(
+    title="Email Attachment Classifier API",
+    description="API to classify whether an email has attachments or not using Naive Bayes",
+    version="1.0.0",
+    docs_url="/docs",
+    redoc_url="/redoc"
+)
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Pydantic models
+class EmailInput(BaseModel):
+    message: str
+
+class EmailBatchInput(BaseModel):
+    messages: List[str]
+
+class PredictionResponse(BaseModel):
+    message: str
+    prediction: int
+    prediction_label: str
+    confidence: float
+    probabilities: Dict[str, float]
+
+class BatchPredictionResponse(BaseModel):
+    predictions: List[PredictionResponse]
+
+class ModelInfo(BaseModel):
+    model_type: str
+    accuracy: float
+    feature_count: int
+    training_samples: int
+
+# Global variables
+model_pipeline = None
+model_info = None
+
+def preprocess_text(text: str) -> str:
+    """Preprocess email text"""
+    # Convert to lowercase
+    text = text.lower()
+    # Remove extra whitespace
+    text = re.sub(r'\s+', ' ', text)
+    # Remove special characters but keep basic punctuation
+    text = re.sub(r'[^\w\s,.\-!?]', ' ', text)
+    return text.strip()
+
+def load_and_train_model():
+    """Load data and train the Naive Bayes model"""
+    global model_pipeline, model_info
+    
+    try:
+        # Load the dataset (assuming it's in the same directory)
+        if os.path.exists('Synthetic_Email_Dataset.csv'):
+            df = pd.read_csv('Synthetic_Email_Dataset.csv')
+        else:
+            logger.warning("Dataset file not found, creating sample data")
+            # Create sample data for demonstration
+            sample_data = {
+                'label': [0, 1, 0, 1] * 100,
+                'message': [
+                    "Hello, You asked for it, so here is the notes. Warm wishes, David",
+                    "Good morning, Just sharing the meeting agenda as requested. Cheers, Anna",
+                    "Dear team, As discussed, I'm sending the manual. Regards, Emily",
+                    "Hi all, Please find attached the project plan. Thanks, Michael"
+                ] * 100
+            }
+            df = pd.DataFrame(sample_data)
+        
+        # Preprocess messages
+        df['processed_message'] = df['message'].apply(preprocess_text)
+        
+        # Split data
+        X = df['processed_message']
+        y = df['label']
+        
+        X_train, X_test, y_train, y_test = train_test_split(
+            X, y, test_size=0.2, random_state=42, stratify=y
+        )
+        
+        # Create pipeline
+        model_pipeline = Pipeline([
+            ('tfidf', TfidfVectorizer(
+                max_features=1000,
+                ngram_range=(1, 2),
+                stop_words='english',
+                lowercase=True,
+                min_df=1,
+                max_df=0.95
+            )),
+            ('classifier', MultinomialNB(alpha=1.0))
+        ])
+        
+        # Train model
+        logger.info("Training Naive Bayes model...")
+        model_pipeline.fit(X_train, y_train)
+        
+        # Evaluate model
+        y_pred = model_pipeline.predict(X_test)
+        accuracy = accuracy_score(y_test, y_pred)
+        
+        # Store model info
+        model_info = ModelInfo(
+            model_type="Multinomial Naive Bayes",
+            accuracy=round(accuracy, 4),
+            feature_count=model_pipeline.named_steps['tfidf'].vocabulary_.__len__(),
+            training_samples=len(X_train)
+        )
+        
+        logger.info(f"Model trained successfully with accuracy: {accuracy:.4f}")
+        logger.info(f"Feature count: {model_info.feature_count}")
+        
+        # Save model
+        joblib.dump(model_pipeline, 'email_classifier_model.pkl')
+        logger.info("Model saved successfully")
+        
+        return True
+        
+    except Exception as e:
+        logger.error(f"Error in training model: {str(e)}")
+        return False
+
+def load_pretrained_model():
+    """Load pretrained model if available"""
+    global model_pipeline, model_info
+    
+    try:
+        if os.path.exists('email_classifier_model.pkl'):
+            model_pipeline = joblib.load('email_classifier_model.pkl')
+            logger.info("Pretrained model loaded successfully")
+            
+            # Set default model info if not available
+            if model_info is None:
+                model_info = ModelInfo(
+                    model_type="Multinomial Naive Bayes",
+                    accuracy=0.92,  # Default value
+                    feature_count=len(model_pipeline.named_steps['tfidf'].vocabulary_),
+                    training_samples=320  # Default value
+                )
+            return True
+    except Exception as e:
+        logger.error(f"Error loading pretrained model: {str(e)}")
+    
+    return False
+
+@app.on_event("startup")
+async def startup_event():
+    """Initialize model on startup"""
+    logger.info("Starting Email Classifier API...")
+    
+    # Try to load pretrained model first
+    if not load_pretrained_model():
+        # If no pretrained model, train new one
+        if not load_and_train_model():
+            logger.error("Failed to initialize model")
+
+@app.get("/", response_class=HTMLResponse)
+async def root():
+    """Root endpoint with API documentation"""
+    html_content = """
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <title>Email Attachment Classifier API</title>
+        <style>
+            body { font-family: Arial, sans-serif; margin: 40px; }
+            .header { color: #2c3e50; }
+            .endpoint { background-color: #f8f9fa; padding: 15px; margin: 10px 0; border-radius: 5px; }
+            .method { color: #27ae60; font-weight: bold; }
+            code { background-color: #e9ecef; padding: 2px 4px; border-radius: 3px; }
+        </style>
+    </head>
+    <body>
+        <h1 class="header">📧 Email Attachment Classifier API</h1>
+        <p>This API classifies whether an email message indicates an attachment or not using Naive Bayes classifier.</p>
+        
+        <h2>Available Endpoints:</h2>
+        
+        <div class="endpoint">
+            <h3><span class="method">GET</span> /info</h3>
+            <p>Get model information and statistics</p>
+        </div>
+        
+        <div class="endpoint">
+            <h3><span class="method">POST</span> /predict</h3>
+            <p>Predict single email message</p>
+            <p><strong>Body:</strong> <code>{"message": "Your email content here"}</code></p>
+        </div>
+        
+        <div class="endpoint">
+            <h3><span class="method">POST</span> /predict-batch</h3>
+            <p>Predict multiple email messages</p>
+            <p><strong>Body:</strong> <code>{"messages": ["Email 1", "Email 2", ...]}</code></p>
+        </div>
+        
+        <div class="endpoint">
+            <h3><span class="method">GET</span> /health</h3>
+            <p>Check API health status</p>
+        </div>
+        
+        <h2>Interactive Documentation:</h2>
+        <p>Visit <a href="/docs">/docs</a> for Swagger UI or <a href="/redoc">/redoc</a> for ReDoc</p>
+        
+        <h2>Labels:</h2>
+        <ul>
+            <li><strong>0:</strong> No attachment mentioned</li>
+            <li><strong>1:</strong> Attachment mentioned</li>
+        </ul>
+    </body>
+    </html>
+    """
+    return HTMLResponse(content=html_content, status_code=200)
+
+@app.get("/health")
+async def health_check():
+    """Health check endpoint"""
+    if model_pipeline is None:
+        return {"status": "unhealthy", "message": "Model not loaded"}
+    return {"status": "healthy", "message": "API is running"}
+
+@app.get("/info", response_model=ModelInfo)
+async def get_model_info():
+    """Get model information"""
+    if model_info is None:
+        raise HTTPException(status_code=503, detail="Model not initialized")
+    return model_info
+
+@app.post("/predict", response_model=PredictionResponse)
+async def predict_single(email: EmailInput):
+    """Predict single email message"""
+    if model_pipeline is None:
+        raise HTTPException(status_code=503, detail="Model not loaded")
+    
+    try:
+        # Preprocess input
+        processed_message = preprocess_text(email.message)
+        
+        # Make prediction
+        prediction = model_pipeline.predict([processed_message])[0]
+        probabilities = model_pipeline.predict_proba([processed_message])[0]
+        
+        # Prepare response
+        prediction_label = "Has attachment" if prediction == 1 else "No attachment"
+        confidence = float(max(probabilities))
+        
+        prob_dict = {
+            "no_attachment": float(probabilities[0]),
+            "has_attachment": float(probabilities[1])
+        }
+        
+        return PredictionResponse(
+            message=email.message,
+            prediction=int(prediction),
+            prediction_label=prediction_label,
+            confidence=confidence,
+            probabilities=prob_dict
+        )
+        
+    except Exception as e:
+        logger.error(f"Prediction error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Prediction failed: {str(e)}")
+
+@app.post("/predict-batch", response_model=BatchPredictionResponse)
+async def predict_batch(emails: EmailBatchInput):
+    """Predict multiple email messages"""
+    if model_pipeline is None:
+        raise HTTPException(status_code=503, detail="Model not loaded")
+    
+    if len(emails.messages) > 100:
+        raise HTTPException(status_code=400, detail="Maximum 100 messages per batch")
+    
+    try:
+        predictions = []
+        
+        # Preprocess all messages
+        processed_messages = [preprocess_text(msg) for msg in emails.messages]
+        
+        # Make batch predictions
+        batch_predictions = model_pipeline.predict(processed_messages)
+        batch_probabilities = model_pipeline.predict_proba(processed_messages)
+        
+        # Prepare responses
+        for i, (message, prediction, probabilities) in enumerate(
+            zip(emails.messages, batch_predictions, batch_probabilities)
+        ):
+            prediction_label = "Has attachment" if prediction == 1 else "No attachment"
+            confidence = float(max(probabilities))
+            
+            prob_dict = {
+                "no_attachment": float(probabilities[0]),
+                "has_attachment": float(probabilities[1])
+            }
+            
+            predictions.append(PredictionResponse(
+                message=message,
+                prediction=int(prediction),
+                prediction_label=prediction_label,
+                confidence=confidence,
+                probabilities=prob_dict
+            ))
+        
+        return BatchPredictionResponse(predictions=predictions)
+        
+    except Exception as e:
+        logger.error(f"Batch prediction error: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Batch prediction failed: {str(e)}")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)

deploy.py

@@ -0,0 +1,310 @@
+#!/usr/bin/env python3
+"""
+Automated deployment script for Hugging Face Spaces
+"""
+
+import os
+import subprocess
+import sys
+import shutil
+from pathlib import Path
+import json
+import requests
+from datetime import datetime
+
+class HuggingFaceDeployer:
+    def __init__(self, username: str, space_name: str, token: str = None):
+        self.username = username
+        self.space_name = space_name
+        self.token = token
+        self.space_url = f"https://huggingface.co/spaces/{username}/{space_name}"
+        self.git_url = f"https://huggingface.co/spaces/{username}/{space_name}.git"
+        
+    def check_requirements(self):
+        """Check if all required files exist"""
+        required_files = [
+            'app.py',
+            'requirements.txt',
+            'Dockerfile',
+            'README.md',
+            'Synthetic_Email_Dataset.csv'
+        ]
+        
+        missing_files = []
+        for file in required_files:
+            if not os.path.exists(file):
+                missing_files.append(file)
+        
+        if missing_files:
+            print(f"❌ Missing required files: {', '.join(missing_files)}")
+            return False
+        
+        print("✅ All required files found")
+        return True
+    
+    def setup_huggingface_space(self):
+        """Create Hugging Face Space via API"""
+        if not self.token:
+            print("⚠️  No Hugging Face token provided. You'll need to create the space manually.")
+            print(f"Go to: https://huggingface.co/new-space")
+            input("Press Enter after creating the space...")
+            return True
+        
+        headers = {"Authorization": f"Bearer {self.token}"}
+        data = {
+            "type": "space",
+            "name": self.space_name,
+            "sdk": "docker",
+            "private": False
+        }
+        
+        response = requests.post(
+            f"https://huggingface.co/api/repos/create",
+            headers=headers,
+            json=data
+        )
+        
+        if response.status_code == 201:
+            print(f"✅ Space created successfully: {self.space_url}")
+            return True
+        elif response.status_code == 409:
+            print(f"ℹ️  Space already exists: {self.space_url}")
+            return True
+        else:
+            print(f"❌ Failed to create space: {response.text}")
+            return False
+    
+    def prepare_files(self):
+        """Prepare files for deployment"""
+        print("📁 Preparing files...")
+        
+        # Create README.md with proper header
+        readme_header = f'''---
+title: {self.space_name.replace('-', ' ').title()}
+emoji: 📧
+colorFrom: blue
+colorTo: green
+sdk: docker
+app_port: 7860
+pinned: false
+license: mit
+tags:
+- email-classification
+- naive-bayes
+- text-classification
+- nlp
+- fastapi
+---
+
+'''
+        
+        # Read existing README and add header
+        if os.path.exists('README.md'):
+            with open('README.md', 'r', encoding='utf-8') as f:
+                existing_content = f.read()
+            
+            # Remove existing header if present
+            if existing_content.startswith('---'):
+                parts = existing_content.split('---', 2)
+                if len(parts) > 2:
+                    existing_content = parts[2].strip()
+            
+            with open('README.md', 'w', encoding='utf-8') as f:
+                f.write(readme_header + existing_content)
+        
+        print("✅ Files prepared")
+        return True
+    
+    def clone_and_deploy(self):
+        """Clone the space repository and deploy"""
+        print("🔄 Cloning repository...")
+        
+        # Create temporary directory
+        temp_dir = f"temp_{self.space_name}"
+        if os.path.exists(temp_dir):
+            shutil.rmtree(temp_dir)
+        
+        try:
+            # Clone the repository
+            result = subprocess.run([
+                'git', 'clone', self.git_url, temp_dir
+            ], capture_output=True, text=True)
+            
+            if result.returncode != 0:
+                print(f"❌ Failed to clone repository: {result.stderr}")
+                return False
+            
+            # Copy files to the cloned repository
+            files_to_copy = [
+                'app.py',
+                'requirements.txt', 
+                'Dockerfile',
+                'README.md',
+                'train_model.py',
+                'Synthetic_Email_Dataset.csv'
+            ]
+            
+            for file in files_to_copy:
+                if os.path.exists(file):
+                    shutil.copy2(file, temp_dir)
+                    print(f"📄 Copied {file}")
+            
+            # Change to the repository directory
+            os.chdir(temp_dir)
+            
+            # Configure git
+            subprocess.run(['git', 'config', 'user.email', '[email protected]'])
+            subprocess.run(['git', 'config', 'user.name', 'Auto Deployer'])
+            
+            # Add all files
+            subprocess.run(['git', 'add', '.'])
+            
+            # Commit changes
+            commit_message = f"Deploy Email Classifier API - {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}"
+            result = subprocess.run(['git', 'commit', '-m', commit_message], 
+                                 capture_output=True, text=True)
+            
+            if result.returncode != 0:
+                print(f"ℹ️  Nothing to commit or commit failed: {result.stderr}")
+            
+            # Push changes
+            if self.token:
+                # Use token for authentication
+                auth_url = self.git_url.replace('https://', f'https://oauth2:{self.token}@')
+                result = subprocess.run(['git', 'push', auth_url, 'main'], 
+                                     capture_output=True, text=True)
+            else:
+                result = subprocess.run(['git', 'push', 'origin', 'main'], 
+                                     capture_output=True, text=True)
+            
+            if result.returncode == 0:
+                print("✅ Successfully pushed to Hugging Face!")
+                print(f"🚀 Your API will be available at: {self.space_url}")
+                return True
+            else:
+                print(f"❌ Failed to push: {result.stderr}")
+                return False
+                
+        except Exception as e:
+            print(f"❌ Deployment failed: {str(e)}")
+            return False
+        finally:
+            # Clean up
+            os.chdir('..')
+            if os.path.exists(temp_dir):
+                shutil.rmtree(temp_dir)
+    
+    def test_deployment(self):
+        """Test the deployed API"""
+        print("🧪 Testing deployed API...")
+        
+        api_url = f"{self.space_url.replace('/spaces/', '/spaces/')}"
+        
+        # Wait a bit for deployment
+        import time
+        print("⏳ Waiting for deployment to complete...")
+        time.sleep(30)
+        
+        try:
+            # Test health endpoint
+            response = requests.get(f"{api_url}/health", timeout=30)
+            if response.status_code == 200:
+                print("✅ Health check passed")
+                
+                # Test prediction endpoint
+                test_response = requests.post(
+                    f"{api_url}/predict",
+                    json={"message": "Please find attached the document."},
+                    timeout=30
+                )
+                
+                if test_response.status_code == 200:
+                    result = test_response.json()
+                    print(f"✅ Prediction test passed: {result['prediction_label']}")
+                    return True
+                else:
+                    print(f"❌ Prediction test failed: {test_response.status_code}")
+            else:
+                print(f"❌ Health check failed: {response.status_code}")
+                
+        except requests.exceptions.RequestException as e:
+            print(f"⚠️  API not yet accessible (this is normal for new deployments): {str(e)}")
+            print("🕐 The API may take a few minutes to become available")
+        
+        return False
+    
+    def deploy(self):
+        """Full deployment process"""
+        print("🚀 Starting deployment to Hugging Face Spaces...")
+        print(f"📦 Space: {self.space_url}")
+        print("="*60)
+        
+        # Check requirements
+        if not self.check_requirements():
+            return False
+        
+        # Setup Hugging Face Space
+        if not self.setup_huggingface_space():
+            return False
+        
+        # Prepare files
+        if not self.prepare_files():
+            return False
+        
+        # Deploy
+        if not self.clone_and_deploy():
+            return False
+        
+        print("\n" + "="*60)
+        print("🎉 DEPLOYMENT COMPLETED!")
+        print("="*60)
+        print(f"📧 API URL: {self.space_url}")
+        print(f"📚 Documentation: {self.space_url}/docs")
+        print(f"🔍 Interactive API: {self.space_url}/redoc")
+        print("\n⏳ Note: It may take a few minutes for the API to become fully available.")
+        
+        # Test deployment
+        self.test_deployment()
+        
+        return True
+
+def main():
+    """Main deployment function"""
+    print("🔧 Hugging Face Spaces Deployment Tool")
+    print("="*40)
+    
+    # Get user input
+    username = input("Enter your Hugging Face username: ").strip()
+    if not username:
+        print("❌ Username is required")
+        sys.exit(1)
+    
+    space_name = input("Enter space name (e.g., email-classifier-api): ").strip()
+    if not space_name:
+        print("❌ Space name is required")
+        sys.exit(1)
+    
+    token = input("Enter your Hugging Face token (optional, press Enter to skip): ").strip()
+    if not token:
+        print("ℹ️  No token provided. You'll need to create the space manually.")
+    
+    # Create deployer and deploy
+    deployer = HuggingFaceDeployer(username, space_name, token)
+    
+    if deployer.deploy():
+        print("\n✅ Deployment successful!")
+        print("\n📋 Next steps:")
+        print("1. Wait 2-3 minutes for the build to complete")
+        print("2. Check the build logs in your Hugging Face Space")
+        print("3. Test your API endpoints")
+        print("4. Share your API with others!")
+    else:
+        print("\n❌ Deployment failed!")
+        print("\n🔧 Troubleshooting:")
+        print("1. Check that all required files are present")
+        print("2. Verify your Hugging Face credentials")
+        print("3. Ensure git is installed and configured")
+        print("4. Check the build logs in your Hugging Face Space")
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,8 @@
+fastapi==0.104.1
+uvicorn[standard]==0.24.0
+pandas==2.1.3
+numpy==1.25.2
+scikit-learn==1.3.2
+joblib==1.3.2
+pydantic==2.5.0
+python-multipart==0.0.6

train_model.py

@@ -0,0 +1,186 @@
+#!/usr/bin/env python3
+"""
+Standalone script to train the email classifier model
+"""
+
+import pandas as pd
+import numpy as np
+from sklearn.feature_extraction.text import TfidfVectorizer
+from sklearn.naive_bayes import MultinomialNB
+from sklearn.pipeline import Pipeline
+from sklearn.model_selection import train_test_split, cross_val_score
+from sklearn.metrics import classification_report, accuracy_score, confusion_matrix
+import joblib
+import re
+import os
+from datetime import datetime
+
+def preprocess_text(text: str) -> str:
+    """Preprocess email text"""
+    text = text.lower()
+    text = re.sub(r'\s+', ' ', text)
+    text = re.sub(r'[^\w\s,.\-!?]', ' ', text)
+    return text.strip()
+
+def load_data(file_path: str):
+    """Load and preprocess the dataset"""
+    print(f"Loading dataset from {file_path}...")
+    
+    if not os.path.exists(file_path):
+        print(f"Error: Dataset file {file_path} not found!")
+        return None, None
+    
+    df = pd.read_csv(file_path)
+    print(f"Dataset loaded: {len(df)} samples")
+    print(f"Columns: {list(df.columns)}")
+    
+    # Basic data info
+    print(f"\nLabel distribution:")
+    print(df['label'].value_counts())
+    
+    # Preprocess messages
+    df['processed_message'] = df['message'].apply(preprocess_text)
+    
+    return df['processed_message'], df['label']
+
+def train_model(X, y):
+    """Train the Naive Bayes model"""
+    print("\nSplitting data...")
+    X_train, X_test, y_train, y_test = train_test_split(
+        X, y, test_size=0.2, random_state=42, stratify=y
+    )
+    
+    print(f"Training set: {len(X_train)} samples")
+    print(f"Test set: {len(X_test)} samples")
+    
+    # Create pipeline
+    print("\nCreating model pipeline...")
+    pipeline = Pipeline([
+        ('tfidf', TfidfVectorizer(
+            max_features=1000,
+            ngram_range=(1, 2),
+            stop_words='english',
+            lowercase=True,
+            min_df=1,
+            max_df=0.95
+        )),
+        ('classifier', MultinomialNB(alpha=1.0))
+    ])
+    
+    # Train model
+    print("Training model...")
+    pipeline.fit(X_train, y_train)
+    
+    # Cross-validation
+    print("Performing cross-validation...")
+    cv_scores = cross_val_score(pipeline, X_train, y_train, cv=5, scoring='accuracy')
+    print(f"Cross-validation scores: {cv_scores}")
+    print(f"Mean CV accuracy: {cv_scores.mean():.4f} (+/- {cv_scores.std() * 2:.4f})")
+    
+    # Test set evaluation
+    print("\nEvaluating on test set...")
+    y_pred = pipeline.predict(X_test)
+    test_accuracy = accuracy_score(y_test, y_pred)
+    
+    print(f"Test accuracy: {test_accuracy:.4f}")
+    print("\nClassification Report:")
+    print(classification_report(y_test, y_pred, target_names=['No Attachment', 'Has Attachment']))
+    
+    print("\nConfusion Matrix:")
+    print(confusion_matrix(y_test, y_pred))
+    
+    # Feature analysis
+    print("\nAnalyzing most important features...")
+    feature_names = pipeline.named_steps['tfidf'].get_feature_names_out()
+    feature_scores = pipeline.named_steps['classifier'].feature_log_prob_
+    
+    # Top features for each class
+    for class_idx, class_name in enumerate(['No Attachment', 'Has Attachment']):
+        top_features_idx = np.argsort(feature_scores[class_idx])[-20:]
+        top_features = [feature_names[i] for i in top_features_idx]
+        print(f"\nTop 20 features for {class_name}:")
+        print(", ".join(reversed(top_features)))
+    
+    return pipeline, test_accuracy
+
+def save_model(pipeline, accuracy, output_path='email_classifier_model.pkl'):
+    """Save the trained model"""
+    print(f"\nSaving model to {output_path}...")
+    
+    # Add metadata
+    model_info = {
+        'pipeline': pipeline,
+        'accuracy': accuracy,
+        'feature_count': len(pipeline.named_steps['tfidf'].vocabulary_),
+        'training_date': datetime.now().isoformat(),
+        'model_type': 'Multinomial Naive Bayes',
+        'preprocessing': 'TF-IDF with 1-2 grams'
+    }
+    
+    joblib.dump(model_info, output_path)
+    print(f"Model saved successfully!")
+    print(f"Model info:")
+    print(f"  - Accuracy: {accuracy:.4f}")
+    print(f"  - Features: {model_info['feature_count']}")
+    print(f"  - Training date: {model_info['training_date']}")
+
+def test_model_predictions(pipeline):
+    """Test model with sample predictions"""
+    print("\n" + "="*50)
+    print("TESTING MODEL WITH SAMPLE PREDICTIONS")
+    print("="*50)
+    
+    test_messages = [
+        "Hello, please find attached the document you requested.",
+        "Good morning, I'm sharing the report as discussed.",
+        "Hi team, attached is the presentation for tomorrow's meeting.",
+        "Dear all, kindly review the attached files.",
+        "Hello, how are you doing today?",
+        "I will send you the information later.",
+        "Please let me know if you need any clarification.",
+        "The meeting is scheduled for 3 PM tomorrow."
+    ]
+    
+    for msg in test_messages:
+        processed_msg = preprocess_text(msg)
+        prediction = pipeline.predict([processed_msg])[0]
+        probabilities = pipeline.predict_proba([processed_msg])[0]
+        confidence = max(probabilities)
+        
+        label = "Has Attachment" if prediction == 1 else "No Attachment"
+        print(f"\nMessage: '{msg}'")
+        print(f"Prediction: {label} (confidence: {confidence:.3f})")
+        print(f"Probabilities: No={probabilities[0]:.3f}, Yes={probabilities[1]:.3f}")
+
+def main():
+    """Main training function"""
+    print("="*60)
+    print("EMAIL ATTACHMENT CLASSIFIER TRAINING")
+    print("="*60)
+    
+    # Load data
+    dataset_path = 'Synthetic_Email_Dataset.csv'
+    X, y = load_data(dataset_path)
+    
+    if X is None:
+        print("Failed to load dataset. Exiting...")
+        return
+    
+    # Train model
+    pipeline, accuracy = train_model(X, y)
+    
+    # Save model
+    save_model(pipeline, accuracy)
+    
+    # Test predictions
+    test_model_predictions(pipeline)
+    
+    print("\n" + "="*60)
+    print("TRAINING COMPLETED SUCCESSFULLY!")
+    print("="*60)
+    print(f"Final model accuracy: {accuracy:.4f}")
+    print("Model saved as 'email_classifier_model.pkl'")
+    print("You can now deploy the API using 'python app.py'")
+
+if __name__ == "__main__":
+    main()