Update api.py

api.py

@@ -1,12 +1,13 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException, Form
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, EmailStr
 from typing import Optional, List, Dict, Any
 import pandas as pd
 import io
 import json
 from backend import RegexClassifier
+from email_service import send_welcome_email
 
 # Initialize FastAPI app
 app = FastAPI(
@@ -82,6 +83,10 @@ class ConfluenceRequest(BaseModel):
 class PDFPageRequest(BaseModel):
     page_number: int = 0
 
+class WelcomeEmailRequest(BaseModel):
+    name: str = Field(..., description="User's name")
+    email: EmailStr = Field(..., description="User's email address")
+
 # ==================== HELPER FUNCTIONS ====================
 
 def validate_file_size(file: UploadFile):
@@ -695,6 +700,35 @@ async def scan_confluence(request: ConfluenceRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Confluence scan failed: {str(e)}")
 
+# ==================== EMAIL FUNCTIONALITY ====================
+
+@app.post("/api/send-welcome")
+async def send_welcome(request: WelcomeEmailRequest):
+    """
+    Send a welcome email to a new user.
+    This endpoint is called by the frontend after a user submits the contact form.
+    """
+    try:
+        # Send the welcome email
+        success = send_welcome_email(request.name, request.email)
+        
+        if success:
+            return JSONResponse(content={
+                "success": True,
+                "message": f"Welcome email sent to {request.email}"
+            })
+        else:
+            raise HTTPException(
+                status_code=500,
+                detail="Failed to send welcome email. SMTP configuration may be missing."
+            )
+    
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Email sending failed: {str(e)}"
+        )
+
 # ==================== HEALTH CHECK ====================
 
 @app.get("/")
@@ -723,4 +757,4 @@ async def health_check():
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)  # HuggingFace Spaces default port
+    uvicorn.run(app, host="0.0.0.0", port=7860)  # HuggingFace Spaces default port