Update app.py

app.py

@@ -2,7 +2,8 @@ import logging
 import smtplib
 import re
 from email.mime.text import MIMEText
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel, Field
 from dotenv import load_dotenv
 import os
 
@@ -46,27 +47,35 @@ def send_email(from_addr: str, to_addr: str, subject: str, body: str) -> bool:
             server.starttls()  # Upgrade to TLS
             server.login(SMTP_USER, SMTP_PASSWORD)
             server.sendmail(from_addr, [to_addr], msg.as_string())
-
         logger.info(f"Email sent successfully from {from_addr} to {to_addr}")
         return True
     except Exception as e:
         logger.error(f"Error sending email: {e}")
         return False
 
-# Flask app acting as the Bot Brain API
-app = Flask(__name__)
+# Pydantic models for request validation
+class Text(BaseModel):
+    body: str
 
-@app.route("/bot", methods=["POST"])
-def bot():
-    try:
-        data = request.json
-        logger.info("Received request from Node.js server: %s", data)
-        sender_number = data.get("from", "unknown")
-        message_body = data.get("message", {}).get("text", {}).get("body", "").strip()
-        logger.info("Processing message from %s: %s", sender_number, message_body)
-    except Exception as e:
-        logger.error("Invalid JSON payload: %s", e)
-        return jsonify({"status": "error", "message": "Invalid payload"}), 400
+class Message(BaseModel):
+    text: Text
+
+class BotRequest(BaseModel):
+    from_: str = Field(..., alias="from")
+    message: Message
+
+# FastAPI app instance
+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)
 
     state = conversations.get(sender_number, {})
 
@@ -76,6 +85,7 @@ def bot():
         response_msg = "Welcome to SMTP Assistant! Please provide your 'From' email address."
     else:
         step = state.get("step")
+        logger.debug("Current step for %s: %s", sender_number, step)
         if step == "from":
             if validate_email(message_body):
                 state["from"] = message_body
@@ -100,18 +110,21 @@ def bot():
             to_addr = state.get("to")
             subject = state.get("subject")
             body = state.get("body")
+            logger.info("Attempting to send email from %s to %s with subject '%s'", from_addr, to_addr, subject)
             if send_email(from_addr, to_addr, subject, body):
                 response_msg = f"Email sent successfully from {from_addr} to {to_addr}."
             else:
                 response_msg = "Failed to send email. Please try again later."
-            # Clear the conversation state after sending
+            # Clear conversation state after sending
             del conversations[sender_number]
+            logger.info("Cleared conversation state for %s", sender_number)
         else:
             response_msg = "Unrecognized step. Let's start over. Please provide your 'From' email address."
             conversations[sender_number] = {"step": "from"}
 
-    logger.info("Sending response to Node.js server: %s", response_msg)
-    return jsonify({"status": "success", "message": response_msg})
-    
+    logger.debug("Response message to %s: %s", sender_number, response_msg)
+    return {"status": "success", "message": response_msg}
+
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000, debug=True)
+    import uvicorn
+    uvicorn.run("main:app", host="0.0.0.0", port=5000, reload=True)