Update app.py

app.py

@@ -9,9 +9,6 @@ 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
@@ -411,36 +408,6 @@ 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_order_email(admin_emails, subject, body):
-    smtp_server = os.getenv("SMTP_SERVER", "smtp.gmail.com")
-    smtp_port = int(os.getenv("SMTP_PORT", 587))
-    smtp_username = os.getenv("SMTP_USERNAME", "yungdml31@gmail.com")
-    smtp_password = os.getenv("SMTP_PASSWORD", "uddvxabxotlvfewk")
-    sender_email = os.getenv("SENDER_EMAIL", "yungdml31@gmail.com")
-
-    msg = MIMEMultipart()
-    msg["From"] = sender_email
-    msg["To"] = ", ".join(admin_emails)
-    msg["Subject"] = subject
-    msg.attach(MIMEText(body, "plain"))
-    
-    try:
-        server = smtplib.SMTP(smtp_server, smtp_port)
-        server.starttls()
-        server.login(smtp_username, smtp_password)
-        server.sendmail(sender_email, admin_emails, msg.as_string())
-        server.quit()
-        return True
-    except Exception as e:
-        print(f"Error sending email: {e}")
-        return False
-
 
 async def process_order_flow(user_id: str, message: str) -> str:
     """
@@ -699,7 +666,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 = get_dish_price(state.data["dish"])  # Use your helper to get the correct price
+            price_per_serving = 1500
             quantity = state.data.get("quantity", 1)
             shipping_cost = state.data.get("shipping_cost", 0)
             total_price = (quantity * price_per_serving) + shipping_cost
@@ -721,29 +688,33 @@ async def process_order_flow(user_id: str, message: str) -> str:
             asyncio.create_task(save_order())
             asyncio.create_task(log_order_tracking(order_id, "Order Placed", "Order placed and awaiting payment."))
     
-            # Build the email body using order details
-            email_subject = f"New Order Received: {order_id}"
-            email_body = (
-                f"New Order Received:\n"
-                f"Order ID: {order_id}\n"
-                f"Dish: {state.data['dish']}\n"
-                f"Quantity: {state.data['quantity']}\n"
-                f"Total Price: N{state.data['price']}\n"
-                f"Phone: {state.data.get('phone_number', '')}\n"
-                f"Delivery Address: {state.data.get('address', 'Not Provided')}\n"
-                f"Extras: {state.data.get('extras', 'None')}\n"
-                f"Status: Pending 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", "")
+            }
+            asyncio.create_task(notify_management_order(order_details))
     
-            # List of admin emails - set in your environment variables or hardcode as needed
-            admin_emails = os.getenv("ADMIN_EMAILS", "samyung05@gmail.com").split(",")
-            # Send email notification to admins
-            email_sent = send_order_email(admin_emails, email_subject, email_body)
-            
-            # 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)
+            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)
             dish_name = state.data.get("dish", "")
+            # Clean up state
             state.reset()
             if user_id in user_state:
                 del user_state[user_id]
@@ -765,12 +736,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]
-            return "Order canceled. Let me know if you'd like to try again."
-    
+
+    else:
+        state.reset()
+        if user_id in user_state:
+            del user_state[user_id]
+
     # Final fallback
     return ""