Update app.py

app.py

@@ -58,95 +58,60 @@ app = Flask(__name__)
 
 @app.route("/bot", methods=["POST"])
 def bot():
-    """
-    Receives a message from the Node.js server.
-    The bot implements a conversation flow:
-      1. Ask for the 'From' email address.
-      2. Ask for the recipient ('To') email address.
-      3. Ask for the email subject.
-      4. Ask for the email body.
-    When all details are collected, the email is sent.
-    """
-    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}'")
+    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
 
-    # Retrieve or initialize conversation state
     state = conversations.get(sender_number, {})
 
     if not state:
-        logger.info(f"New conversation started for {sender_number}")
+        # Start a new conversation
         conversations[sender_number] = {"step": "from"}
-        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 = "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.")
-
+        response_msg = "Welcome to SMTP Assistant! Please provide your 'From' email address."
     else:
-        response_msg = "Unrecognized step. Restarting. Please provide your 'From' email address."
-        conversations[sender_number] = {"step": "from"}
+        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]
+        else:
+            response_msg = "Unrecognized step. Let's start over. Please provide your 'From' email address."
+            conversations[sender_number] = {"step": "from"}
 
-    logger.debug(f"Response message to {sender_number}: {response_msg}")
+    logger.info("Sending response to Node.js server: %s", response_msg)
     return jsonify({"status": "success", "message": response_msg})
-
-
+    
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=5000, debug=True)