Update app.py

app.py

@@ -9,6 +9,9 @@ from bs4 import BeautifulSoup
 from sqlalchemy import select
 from fuzzywuzzy import fuzz
 from pydantic import BaseModel
+import smtplib
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
 
 from fastapi import FastAPI, Request, HTTPException, BackgroundTasks, UploadFile, File, Form
 from fastapi.responses import JSONResponse, StreamingResponse, RedirectResponse
@@ -408,6 +411,47 @@ def match_dishes(user_input: str, threshold: int = 70) -> list:
                 matched_dishes.append(dish_name)
     return list(set(matched_dishes))
 
+def get_dish_price(dish: str) -> int:
+    for item in menu_items:
+        if item["name"].lower() == dish.lower():
+            return item["price"]
+    return 0  # default if not found
+
+def send_email_notification(order_details):
+    # Construct the email payload. Adjust values as needed.
+    payload = {
+        "from": "yungdml31@gmail.com",
+        "to": "samyung05@gmail.com",  # You can also use a comma-separated list for multiple recipients.
+        "subject": f"New Order Received: {order_details['order_id']}",
+        "body": (
+            f"New Order Received:\n"
+            f"Order ID: {order_details['order_id']}\n"
+            f"Dish: {order_details['dish']}\n"
+            f"Quantity: {order_details['quantity']}\n"
+            f"Total Price: N{order_details['price']}\n"
+            f"Phone: {order_details.get('phone_number', 'Not Provided')}\n"
+            f"Delivery Address: {order_details.get('address', 'Not Provided')}\n"
+            f"Extras: {order_details.get('extras', 'None')}\n"
+            f"Status: Pending Payment"
+        ),
+        # Even though your endpoint is set up to receive these fields, if they're fixed you can omit them or keep them here:
+        "smtpHost": "smtp.gmail.com",   # or your SMTP host
+        "smtpPort": 587,                # or your SMTP port
+        "smtpSecure": "false",          # boolean as string if needed
+        "smtpUser": "yungdml31@gmail.com",      
+        "smtpPassword": "uddvxabxotlvfewk",
+    }
+    
+    # Replace the URL with the URL of your SMTP API endpoint.
+    url = "https://smtp-server-ten.vercel.app/smtp"
+    try:
+        response = requests.post(url, json=payload, timeout=10)
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        print(f"Error sending email: {e}")
+        return None
+
 
 async def process_order_flow(user_id: str, message: str) -> str:
     """
@@ -666,7 +710,7 @@ async def process_order_flow(user_id: str, message: str) -> str:
         if message.lower() in ["yes", "y"]:
             order_id = f"ORD-{int(time.time())}"
             state.data["order_id"] = order_id
-            price_per_serving = 1500
+            price_per_serving = get_dish_price(state.data["dish"])  # Use your helper to get the correct price
             quantity = state.data.get("quantity", 1)
             shipping_cost = state.data.get("shipping_cost", 0)
             total_price = (quantity * price_per_serving) + shipping_cost
@@ -687,34 +731,27 @@ async def process_order_flow(user_id: str, message: str) -> str:
                     await session.commit()
             asyncio.create_task(save_order())
             asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
-    
-            async def notify_management_order(order_details: dict):
-                message_body = (
-                    f"New Order Received:\n"
-                    f"Order ID: {order_details['order_id']}\n"
-                    f"Dish: {order_details['dish']}\n"
-                    f"Quantity: {order_details['quantity']}\n"
-                    f"Total Price: {order_details['price']}\n"
-                    f"Phone: {state.data.get('phone_number', '')}\n"
-                    f"Delivery Address: {order_details.get('address', 'Not Provided')}\n"
-                    f"Extras: {state.data.get('extras', 'None')}\n"
-                    f"Status: Pending Payment"
-                )
-                await asyncio.to_thread(send_whatsapp_message, MANAGEMENT_WHATSAPP_NUMBER, message_body)
+            
             order_details = {
                 "order_id": order_id,
                 "dish": state.data["dish"],
                 "quantity": state.data["quantity"],
                 "price": state.data["price"],
-                "address": state.data.get("address", "")
+                "address": state.data.get("address", ""),
+                "phone_number": state.data.get("phone_number", ""),
+                "extras": state.data.get("extras", "None")
             }
-            asyncio.create_task(notify_management_order(order_details))
-    
-            email = "customer@example.com"  # or extract the email from the user's profile if available
-            # create_paystack_payment_link returns a dict with a key "status" and the link under "data.authorization_url"
-            payment_data = create_paystack_payment_link(email, total_price * 100, order_id)
+            # Notify admin via email
+            email_response = send_email_notification(order_details)
+            if email_response:
+                print("Email notification sent successfully:", email_response)
+            else:
+                print("Failed to send email notification.")
+            
+            # Create Paystack payment link
+            email_for_paystack = "customer@example.com"  # Replace with actual email if available
+            payment_data = create_paystack_payment_link(email_for_paystack, total_price * 100, order_id)
             dish_name = state.data.get("dish", "")
-            # Clean up state
             state.reset()
             if user_id in user_state:
                 del user_state[user_id]
@@ -736,12 +773,12 @@ async def process_order_flow(user_id: str, message: str) -> str:
                         "but we could not initialize online payment. Please try again later, or "
                         "you may opt to pay via bank transfer to Account Number 1433042821, Access Bank, Angelo Food Court 2 "
                         "and send your payment screenshot to this chatbot.")
-
-    else:
-        state.reset()
-        if user_id in user_state:
-            del user_state[user_id]
-
+        else:
+            state.reset()
+            if user_id in user_state:
+                del user_state[user_id]
+            return "Order canceled. Let me know if you'd like to try again."
+    
     # Final fallback
     return ""