Update app.py

app.py

@@ -2,10 +2,13 @@ import logging
 import smtplib
 import re
 from email.mime.text import MIMEText
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI
 from pydantic import BaseModel, Field
 from dotenv import load_dotenv
 import os
+from aiosmtpd.controller import Controller
+from aiosmtpd.handlers import Sink
+import uvicorn
 
 # Load environment variables
 load_dotenv()
@@ -14,10 +17,13 @@ load_dotenv()
 logging.basicConfig(level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger("bot_brain")
 
-# SMTP Server Configuration
-SMTP_HOST = os.getenv("SMTP_HOST", "0.0.0.0")
-SMTP_PORT = int(os.getenv("SMTP_PORT", "587"))
-CLIENT_SMTP_HOST = os.getenv("CLIENT_SMTP_HOST", "localhost")
+# SMTP server configuration for our own SMTP server
+SMTP_SERVER_HOST = os.getenv("SMTP_SERVER_HOST", "localhost")
+SMTP_SERVER_PORT = int(os.getenv("SMTP_SERVER_PORT", "1025"))
+
+# Configuration for the SMTP client (sending emails via our own SMTP server)
+CLIENT_SMTP_HOST = os.getenv("CLIENT_SMTP_HOST", SMTP_SERVER_HOST)
+SMTP_PORT = int(os.getenv("SMTP_PORT", SMTP_SERVER_PORT))
 SMTP_USER = os.getenv("SMTP_USER", "noreply@yourdomain.com")
 SMTP_PASSWORD = os.getenv("SMTP_PASSWORD", "password")
 
@@ -64,15 +70,14 @@ class BotRequest(BaseModel):
     from_: str = Field(..., alias="from")
     message: Message
 
-# FastAPI app instance
+# Create FastAPI app
 app = FastAPI()
 
 @app.post("/bot")
 async def bot_endpoint(request_data: BotRequest):
-    # Log the received request payload
     payload = request_data.dict(by_alias=True)
     logger.debug("Received request payload: %s", payload)
-    
+
     sender_number = request_data.from_
     message_body = request_data.message.text.body.strip()
     logger.info("Processing message from %s: '%s'", sender_number, message_body)
@@ -80,7 +85,6 @@ async def bot_endpoint(request_data: BotRequest):
     state = conversations.get(sender_number, {})
 
     if not state:
-        # Start a new conversation
         conversations[sender_number] = {"step": "from"}
         response_msg = "Welcome to SMTP Assistant! Please provide your 'From' email address."
     else:
@@ -125,6 +129,20 @@ async def bot_endpoint(request_data: BotRequest):
     logger.debug("Response message to %s: %s", sender_number, response_msg)
     return {"status": "success", "message": response_msg}
 
+def start_smtp_server():
+    """
+    Start our own SMTP server using aiosmtpd.
+    For simplicity, we use the Sink handler which accepts and prints all emails.
+    """
+    handler = Sink()
+    controller = Controller(handler, hostname=SMTP_SERVER_HOST, port=SMTP_SERVER_PORT)
+    controller.start()
+    logger.info(f"SMTP server started at {SMTP_SERVER_HOST}:{SMTP_SERVER_PORT}")
+    return controller
+
 if __name__ == "__main__":
-    import uvicorn
+    # Start the SMTP server (our own SMTP server)
+    smtp_controller = start_smtp_server()
+
+    # Run the FastAPI app using Uvicorn
     uvicorn.run("main:app", host="0.0.0.0", port=5000, reload=True)