Update app.py

app.py

@@ -10,7 +10,7 @@ import os
 load_dotenv()
 
 # Configure logging
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(level=logging.DEBUG, format="%(asctime)s - %(levelname)s - %(message)s")
 logger = logging.getLogger("bot_brain")
 
 # SMTP Server Configuration
@@ -33,22 +33,24 @@ def validate_email(email: str) -> bool:
 def send_email(from_addr: str, to_addr: str, subject: str, body: str) -> bool:
     """
     Sends an email using the SMTP client.
-    Note: Some SMTP servers restrict the 'From' address.
     """
     msg = MIMEText(body)
     msg['Subject'] = subject
     msg['From'] = from_addr
     msg['To'] = to_addr
 
+    logger.debug(f"Preparing to send email from {from_addr} to {to_addr} with subject '{subject}'")
+
     try:
         with smtplib.SMTP(CLIENT_SMTP_HOST, SMTP_PORT) as server:
             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 from {from_addr} to {to_addr}")
+
+        logger.info(f"Email sent successfully from {from_addr} to {to_addr}")
         return True
     except Exception as e:
-        logger.error(f"Error sending email from {from_addr} to {to_addr}: {e}")
+        logger.error(f"Error sending email: {e}")
         return False
 
 # Flask app acting as the Bot Brain API
@@ -65,57 +67,86 @@ def bot():
       4. Ask for the email body.
     When all details are collected, the email is sent.
     """
-    try:
-        data = request.json
-        sender_number = data.get("from", "unknown")
-        message_body = data.get("message", {}).get("text", {}).get("body", "").strip()
-    except Exception as e:
-        logger.error("Invalid JSON payload: %s", e)
-        return jsonify({"status": "error", "message": "Invalid payload"}), 400
+    logger.debug("Received a request at /bot")
+
+    # Ensure the request contains valid JSON
+    if not request.is_json:
+        logger.error("Invalid request format: Not JSON")
+        return jsonify({"status": "error", "message": "Request must be JSON"}), 400
+
+    data = request.json
+    logger.debug(f"Request payload: {data}")
+
+    # Extract sender number and message
+    sender_number = data.get("from")
+    message_body = data.get("message", {}).get("text", {}).get("body", "").strip()
 
+    if not sender_number or not message_body:
+        logger.error("Missing required fields in request")
+        return jsonify({"status": "error", "message": "Missing 'from' or 'message.text.body'"}), 400
+
+    logger.info(f"Processing message from {sender_number}: '{message_body}'")
+
+    # Retrieve or initialize conversation state
     state = conversations.get(sender_number, {})
 
     if not state:
-        # Start a new conversation
+        logger.info(f"New conversation started for {sender_number}")
         conversations[sender_number] = {"step": "from"}
-        response_msg = "Welcome to SMTP Assistant! Please provide your 'From' email address."
-    else:
-        step = state.get("step")
-        if step == "from":
-            if validate_email(message_body):
-                state["from"] = message_body
-                state["step"] = "to"
-                response_msg = "Thanks! Now, please provide the recipient's email address."
-            else:
-                response_msg = "Invalid email format. Please provide a valid 'From' email address."
-        elif step == "to":
-            if validate_email(message_body):
-                state["to"] = message_body
-                state["step"] = "subject"
-                response_msg = "Great! Please provide the email subject."
-            else:
-                response_msg = "Invalid email format. Please provide a valid recipient email address."
-        elif step == "subject":
-            state["subject"] = message_body
-            state["step"] = "body"
-            response_msg = "Almost done! Please provide the email body."
-        elif step == "body":
-            state["body"] = message_body
-            from_addr = state.get("from")
-            to_addr = state.get("to")
-            subject = state.get("subject")
-            body = state.get("body")
-            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
-            del conversations[sender_number]
+        return jsonify({"status": "success", "message": "Welcome! Please provide your 'From' email address."})
+
+    step = state.get("step")
+    logger.debug(f"Current step for {sender_number}: {step}")
+
+    response_msg = "Something went wrong."
+
+    if step == "from":
+        if validate_email(message_body):
+            state["from"] = message_body
+            state["step"] = "to"
+            response_msg = "Thanks! Now, please provide the recipient's email address."
         else:
-            response_msg = "Unrecognized step. Let's start over. Please provide your 'From' email address."
-            conversations[sender_number] = {"step": "from"}
+            response_msg = "Invalid email format. Please provide a valid 'From' email address."
 
+    elif step == "to":
+        if validate_email(message_body):
+            state["to"] = message_body
+            state["step"] = "subject"
+            response_msg = "Great! Please provide the email subject."
+        else:
+            response_msg = "Invalid email format. Please provide a valid recipient email address."
+
+    elif step == "subject":
+        state["subject"] = message_body
+        state["step"] = "body"
+        response_msg = "Almost done! Please provide the email body."
+
+    elif step == "body":
+        state["body"] = message_body
+
+        from_addr = state.get("from")
+        to_addr = state.get("to")
+        subject = state.get("subject")
+        body = state.get("body")
+
+        logger.info(f"Attempting to send email from {from_addr} to {to_addr} with subject '{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
+        del conversations[sender_number]
+        logger.info(f"Conversation with {sender_number} cleared.")
+
+    else:
+        response_msg = "Unrecognized step. Restarting. Please provide your 'From' email address."
+        conversations[sender_number] = {"step": "from"}
+
+    logger.debug(f"Response message to {sender_number}: {response_msg}")
     return jsonify({"status": "success", "message": response_msg})
 
+
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=5000)
+    app.run(host="0.0.0.0", port=5000, debug=True)