Upload folder using huggingface_hub

app/__init__.py

@@ -1,11 +1,19 @@
 import logging
 from flask import Flask, jsonify
+
 from . import config
 from .extensions import mongo, bcrypt, jwt, cors, session, oauth
 from .xero_client import api_client, xero  # <-- IMPORT XERO REMOTE APP
-
+from .ai_features import ai_bp
+from .page_features import page_bp
+# app.register_blueprint(api_bp, url_prefix='/api')
 def create_app():
+
     app = Flask(__name__)
+
+
+    app.register_blueprint(ai_bp, url_prefix='/api')
+    app.register_blueprint(page_bp, url_prefix='/api/pages')
     app.config.from_object(config)
     
     logging.basicConfig(level=logging.INFO)

app/ai_features.py

@@ -0,0 +1,617 @@
+# your_app/ai_features.py
+
+from flask import Blueprint, request, jsonify, current_app
+from bson.objectid import ObjectId
+from datetime import datetime, date, timedelta
+from flask_jwt_extended import jwt_required, get_jwt_identity
+from .extensions import bcrypt, mongo
+from .Company_Info import company_info
+
+from google import genai
+from google.genai import types
+import os
+import json
+
+# +++ START: WHATSAPP FEATURE IMPORTS +++
+import requests
+from twilio.twiml.messaging_response import MessagingResponse
+# +++ END: WHATSAPP FEATURE IMPORTS +++
+
+ai_bp = Blueprint('ai', __name__)
+
+today = date.today()
+weekday_number = today.weekday()
+days_of_week = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
+weekday_name = days_of_week[weekday_number]
+
+# --- NEW: Define functions for the AI model ---
+
+add_items_to_cart_function = {
+    "name": "add_items_to_cart",
+    "description": "Extracts order details from a user's message to add products to their shopping cart. Can also set a delivery date.",
+    "parameters": {
+        "type": "object", "properties": {
+            "items": { "type": "array", "description": "A list of items to add to the cart.", "items": { "type": "object", "properties": { "product_name": { "type": "string", "description": "The name of the product. This must match one of the available product names.", }, "quantity": { "type": "number", "description": "The quantity of the product.", }, "unit": { "type": "string", "description": "The unit of measure for the quantity (e.g., 'lb', 'pieces', 'box', 'bunch'). This must match one of the available units for the product.", }, }, "required": ["product_name", "quantity", "unit"], }, },
+            "delivery_date": {
+                "type": "string",
+                "description": "The desired delivery date for the order, in YYYY-MM-DD format. The user must explicitly state a date.",
+            }
+        },
+        "required": ["items"],
+    },
+}
+
+# +++ START: NEW FUNCTION DEFINITIONS FOR CART MANAGEMENT +++
+remove_items_from_cart_function = {
+    "name": "remove_items_from_cart",
+    "description": "Removes one or more specific items from the user's shopping cart based on their name.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "items": {
+                "type": "array",
+                "description": "A list of items to remove from the cart. Only product_name is required.",
+                "items": {
+                    "type": "object",
+                    "properties": {
+                        "product_name": {
+                            "type": "string",
+                            "description": "The name of the product to remove. This must match an available product name.",
+                        },
+                    },
+                    "required": ["product_name"],
+                },
+            }
+        },
+        "required": ["items"],
+    },
+}
+
+clear_cart_function = {
+    "name": "clear_cart",
+    "description": "Removes all items from the user's shopping cart. Use when the user wants to start over or empty their cart.",
+    "parameters": {"type": "object", "properties": {}}
+}
+# +++ END: NEW FUNCTION DEFINITIONS FOR CART MANAGEMENT +++
+
+navigate_to_page_function = {
+    "name": "navigate_to_page",
+    "description": "Provides a button in the chat to navigate the user to a specific page.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "page": {
+                "type": "string",
+                "description": "The destination page. Must be one of the allowed pages.",
+                "enum": ["orders", "cart", "catalog", "home", "about", "care", "login", "register"]
+            }
+        },
+        "required": ["page"]
+    }
+}
+
+get_my_orders_function = {
+    "name": "get_my_orders",
+    "description": "Retrieves a summary of the user's most recent orders to display in the chat.",
+    "parameters": {"type": "object", "properties": {}}
+}
+
+get_cart_items_function = {
+    "name": "get_cart_items",
+    "description": "Retrieves the items currently in the user's shopping cart and lists them.",
+    "parameters": {"type": "object", "properties": {}}
+}
+
+get_order_details_function = {
+    "name": "get_order_details",
+    "description": "Retrieves the specific items and details for a single order based on its ID.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "order_id": {
+                "type": "string",
+                "description": "The ID of the order to fetch. Can be the full ID or the last 6 characters."
+            }
+        },
+        "required": ["order_id"]
+    }
+}
+
+cancel_order_function = {
+    "name": "cancel_order",
+    "description": "Proposes to cancel an order based on its ID. This requires user confirmation.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "order_id": {
+                "type": "string",
+                "description": "The ID of the order to be cancelled. Can be the full ID or the last 6 characters."
+            }
+        },
+        "required": ["order_id"]
+    }
+}
+
+# +++ START: NEW FUNCTION DEFINITION FOR PLACING AN ORDER +++
+place_order_function = {
+    "name": "place_order",
+    "description": "Places a final order using the items in the user's cart. This action is final and will clear the cart.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "delivery_date": {
+                "type": "string",
+                "description": "The desired delivery date in YYYY-MM-DD format. Must be confirmed with the user."
+            },
+            "additional_info": {
+                "type": "string",
+                "description": "Any special instructions or notes from the user for the delivery."
+            }
+        },
+        "required": ["delivery_date"]
+    }
+}
+# +++ END: NEW FUNCTION DEFINITION FOR PLACING AN ORDER +++
+
+register_new_customer_function = {
+    "name": "register_new_customer",
+    "description": "Registers a new customer by collecting their essential details. Use this when a user expresses intent to sign up or is not recognized.",
+    "parameters": {
+        "type": "object",
+        "properties": {
+            "businessName": {
+                "type": "string",
+                "description": "The name of the customer's business or company."
+            },
+            "email": {
+                "type": "string",
+                "description": "The customer's primary email address. This will be their login username."
+            },
+            "password": {
+                "type": "string",
+                "description": "A password for the user's account. The user must provide this."
+            },
+            "phoneNumber": {
+                "type": "string",
+                "description": "The user's contact phone number."
+            },
+            "businessAddress": {
+                "type": "string",
+                "description": "The full delivery address for the business."
+            },
+            "contactPerson": {
+                "type": "string",
+                "description": "The customer's full name or primary contact person's name."
+            }
+        },
+        "required": ["businessName", "email", "password", "phoneNumber", "businessAddress"]
+    }
+}
+
+
+@ai_bp.route('/chat', methods=['POST'])
+@jwt_required()
+def handle_ai_chat():
+    # ... (Web chat endpoint remains unchanged)
+        user_email = get_jwt_identity()
+        MODES = ['case', 'bag', 'piece', 'tray', 'weight']
+        all_products = list(mongo.db.products.find({}, {'name': 1, 'unit': 1, '_id': 0}))
+        product_context_list = []
+        for p in all_products:
+            if 'name' not in p: continue
+            unit_string = p.get('unit', '').lower()
+            available_modes = [mode for mode in MODES if mode in unit_string]
+            if available_modes:
+                product_context_list.append(f"{p['name']} (available units: {', '.join(available_modes)})")
+            else:
+                product_context_list.append(p['name'])
+        client = genai.Client(api_key="AIzaSyB7yKIdfW7Umv62G47BCdJjoHTJ9TeiAko")
+
+        current_date = datetime.now().strftime('%Y-%m-%d')
+        min_delivery_date = (datetime.now() + timedelta(days=2)).strftime('%Y-%m-%d')
+
+        tools = types.Tool(function_declarations=[
+            add_items_to_cart_function,
+            navigate_to_page_function,
+            get_my_orders_function,
+            cancel_order_function,
+            register_new_customer_function
+        ])
+        config = types.GenerateContentConfig(tools=[tools])
+        
+        instruction_prompt = f"""
+        You are an expert assistant for a food wholesale company. Your capabilities include:
+        1.  Ordering & Delivery Date: Use `add_items_to_cart` to add items to the user's cart. You can also set a delivery date if the user specifies one. The current date is {current_date,weekday_name}. The earliest possible delivery date is {min_delivery_date}. If the user asks for a date earlier than this, inform them of the constraint and ask them to provide a new valid date.
+        2.  Product List: {'; '.join(product_context_list)}.
+        3.  Page Navigation: Use `navigate_to_page` if the user wants to GO to a page like 'my orders', 'about us', 'customer care', 'login', etc.
+        4.  Order Summary: Use `get_my_orders` to LIST a summary of recent orders in the chat.
+        5.  Order Cancellation: Use `cancel_order` to start the cancellation process for a SPECIFIC order. The user must provide an order ID.
+        6.  New Customer Registration: Use `register_new_customer` if a user wants to sign up.
+        7.  The Capability You must use AT ALL TIMES: Luring the customer to place an order and persuading them to buy more products.
+        *Always Brag about the quality of our products and our customer service and try to persuade the customer to place an order.*
+
+
+        Company Info for general questions: {company_info}
+        *** IMPORTANT: When providing a text-based answer, YOU MUST USE MARKDOWN. Use headings, bold text, bullet points (`* item`), and other formatting to make your answers clear, structured, and easy to read. ***
+        Always refer to the chat history for context. **If information is missing, ask for it.**
+        """
+
+        user_input_part, history_data = None, []
+        if 'multipart/form-data' in request.content_type:
+            audio_file = request.files.get('audio')
+            history_json = request.form.get('history', '[]')
+            history_data = json.loads(history_json)
+            user_input_part = types.Part.from_bytes(data=audio_file.read(), mime_type=audio_file.mimetype.split(';')[0])
+        elif 'application/json' in request.content_type:
+            json_data = request.get_json()
+            history_data = json_data.get('history', [])
+            user_input_part = json_data.get('message')
+        else: return jsonify({"msg": "Unsupported format"}), 415
+
+        history_contents = [msg.get('data') for msg in history_data if msg.get('type') == 'text']
+        final_contents = [instruction_prompt] + history_contents + [user_input_part]
+        response = client.models.generate_content(model="gemini-2.5-flash", contents=final_contents, config=config)
+        response_part = response.candidates[0].content.parts[0]
+
+        if response_part.function_call:
+            function_call = response_part.function_call
+
+            if function_call.name == "add_items_to_cart":
+                proposal_data = {
+                    "items": function_call.args.get('items', []),
+                    "delivery_date": function_call.args.get('delivery_date')
+                }
+                return jsonify({"type": "order_proposal", "data": proposal_data}), 200
+
+            elif function_call.name == "register_new_customer":
+                return jsonify({
+                    "type": "navigation_proposal",
+                    "data": {
+                        "path": "/register",
+                        "button_text": "Go to Sign Up Page",
+                        "prompt_text": "Excellent! I can help with that. Please click the button below to go to our registration page."
+                    }
+                }), 200
+
+            elif function_call.name == "navigate_to_page":
+                page = function_call.args.get('page')
+                path_map = {'orders': '/order', 'cart': '/cart', 'catalog': '/catalog', 'home': '/', 'about': '/about', 'care': '/care', 'login': '/login', 'register': '/register'}
+                text_map = {'orders': 'Go to My Orders', 'cart': 'Go to Cart', 'catalog': 'Go to Catalog', 'home': 'Go to Homepage', 'about': 'Go to About Page', 'care': 'Go to Customer Care', 'login': 'Go to Login', 'register': 'Go to Sign Up'}
+                if page in path_map:
+                    return jsonify({"type": "navigation_proposal", "data": {"path": path_map[page], "button_text": text_map[page], "prompt_text": f"Sure, let's go to the {page} page."}}), 200
+
+            elif function_call.name == "get_my_orders":
+                recent_orders = list(mongo.db.orders.find({'user_email': user_email}).sort('created_at', -1).limit(5))
+                if not recent_orders: return jsonify({"type": "text", "data": "You have no recent orders."}), 200
+                details_text=""
+                for order in recent_orders:
+                    product_ids = [ObjectId(item['productId']) for item in order.get('items', [])]
+                    products_map = {str(p['_id']): p for p in mongo.db.products.find({'_id': {'$in': product_ids}})}
+                    item_lines = [f"- {item['quantity']} {item.get('mode', 'pcs')} of {products_map.get(item['productId'], {}).get('name', 'Unknown Product')}" for item in order.get('items', [])]
+                    details_text += f"\n\n **Details for Order #{str(order['_id'])[-4:]}** _(Delivery Status: {order.get('status')})_:\n" + "\n".join(item_lines)
+                return jsonify({"type": "text", "data": "**Here are your recent orders**:\n" + details_text}), 200
+
+            elif function_call.name == "cancel_order":
+                order_id_frag = function_call.args.get("order_id", "").strip()
+                all_user_orders = mongo.db.orders.find({'user_email': user_email})
+                order = None
+                for o in all_user_orders:
+                    if str(o['_id']).endswith(order_id_frag):
+                        order = o
+                        break
+                if not order: return jsonify({"type": "text", "data": f"Sorry, I couldn't find an order with ID matching '{order_id_frag}'."}), 200
+                if order.get('status') not in ['pending', 'confirmed']: return jsonify({"type": "text", "data": f"Order #{str(order['_id'])[-4:]} cannot be cancelled as its status is '{order.get('status')}'."}), 200
+                return jsonify({
+                    "type": "cancellation_proposal",
+                    "data": {"order_id": str(order['_id']), "prompt_text": f"Are you sure you want to cancel order #{str(order['_id'])[-4:]}?"}
+                }), 200
+
+        return jsonify({"type": "text", "data": response.text}), 200
+
+@ai_bp.route('/whatsapp', methods=['POST'])
+def whatsapp_reply():
+    """ Responds to incoming WhatsApp messages (text or audio) via Twilio. """
+    final_response_text = "I'm sorry, I encountered an error. Please try again."
+    twilio_resp = MessagingResponse()
+    
+    try:
+        # --- 1. Get User and Message from Twilio Request ---
+        whatsapp_number = request.values.get('From')
+        user_message_text = request.values.get('Body', '').strip()
+        num_media = int(request.values.get('NumMedia', 0))
+
+        if not whatsapp_number:
+            current_app.logger.error("Request received without a 'From' number.")
+            twilio_resp.message("Could not identify your number. Please try again.")
+            return str(twilio_resp)
+
+        # --- 2. Identify User or Prepare for Registration ---
+        user = mongo.db.users.find_one({'whatsapp_number': whatsapp_number})
+        
+        # --- 3. Setup AI Client and Product Context ---
+        client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
+        MODES = ['case', 'bag', 'piece', 'tray', 'weight']
+        all_products = list(mongo.db.products.find({}, {'name': 1, 'unit': 1, '_id': 0}))
+        product_context_list = []
+        for p in all_products:
+            if 'name' not in p: continue
+            unit_string = p.get('unit', '').lower()
+            available_modes = [mode for mode in MODES if mode in unit_string]
+            if available_modes:
+                product_context_list.append(f"{p['name']} (available units: {', '.join(available_modes)})")
+            else:
+                product_context_list.append(p['name'])        
+        
+        # --- 4. Process User Input (Text or Audio) ---
+        user_input_part = None
+        if num_media > 0:
+            media_url = request.values.get('MediaUrl0')
+            mime_type = request.values.get('MediaContentType0')
+            if 'audio' in mime_type:
+                audio_response = requests.get(media_url)
+                if audio_response.status_code == 200:
+                    user_input_part = types.Part.from_bytes(data=audio_response.content, mime_type=mime_type.split(';')[0])
+        
+        if not user_input_part and user_message_text:
+            user_input_part = user_message_text
+
+        if not user_input_part:
+            twilio_resp.message("Please send a message or a voice note.")
+            return str(twilio_resp)
+
+        # --- 5. Configure AI based on User Status and Conversation History ---
+        # +++ START: RETRIEVE AND MANAGE CONVERSATION HISTORY +++
+        history_doc = mongo.db.whatsapp_history.find_one({'whatsapp_number': whatsapp_number})
+        chat_history = history_doc.get('history', []) if history_doc else []
+        # +++ END: RETRIEVE AND MANAGE CONVERSATION HISTORY +++
+
+        if user:
+            # User is known, set up for ordering, viewing orders, and placing orders
+            tools = types.Tool(function_declarations=[
+                add_items_to_cart_function,
+                remove_items_from_cart_function, # ADDED
+                clear_cart_function,             # ADDED
+                get_cart_items_function,
+                place_order_function,
+                get_my_orders_function,
+                get_order_details_function,
+            ])
+            instruction_prompt = f"""
+            You are a helpful WhatsApp ordering assistant for A WholeSale Grocery Company Named Matax Express. The user is {user.get('contactPerson', 'a valued customer')}.
+            Your goal is to help them manage their order.
+            - To add items, use `add_items_to_cart`.
+            - To remove specific items from the cart, use `remove_items_from_cart`.
+            - To clear the entire cart, use `clear_cart`.
+            - To list items in the cart, use `get_cart_items`.
+            - To get a summary of recent orders, use `get_my_orders`.
+            - To get details for a specific order, use `get_order_details`.
+            - To finalize the purchase, use `place_order`. You MUST confirm the delivery date and ask for any special instructions before calling this function.
+            Available products: {', '.join(product_context_list)}.
+            Always be friendly and confirm actions clearly. Refer to the chat history for context.
+            Company Info for general questions: {company_info}.
+            **Whenever the user adds items to the cart,places an order , or registers as new customer inform them that they can also view the cart, view their orders and also order on https://matax-express.vercel.app/ logging in with the account they used to register on whatsapp.**
+            Finally when the user places the order, inform them that they can view the placed order details on https://matax-express.vercel.app/ by logging in with the account they used to register on whatsapp.
+            Always respond in whatsapp formatted markdown manner.
+            """
+        else:
+            # User is new, set up for registration
+            tools = types.Tool(function_declarations=[register_new_customer_function])
+            instruction_prompt = f"""
+            You are a helpful WhatsApp registration assistant forA WholeSale Grocery Company named matax express. The user is new.
+            Your primary goal is to register them using the `register_new_customer` function.
+            You MUST collect their business name, email, a password, phone number, and business address.
+            Ask for the information clearly. Advise the user to choose a secure password.
+            Company Info for general questions: {company_info}.
+            **Whenever the user adds items to the cart,places an order , or registers as new customer inform them that they can also view the cart, view their orders and also order  on https://matax-express.vercel.app/  logging in with the account they used to register on whatsapp.**
+            Always respond in whatsapp formatted markdown manner.
+            """
+        
+        config = types.GenerateContentConfig(tools=[tools])
+        
+        # +++ START: ADD HISTORY TO THE API CALL +++
+        final_contents = [instruction_prompt] + chat_history + [user_input_part]
+        response = client.models.generate_content(
+            model="gemini-2.5-flash",
+            contents=final_contents,
+            config=config
+        )
+        # +++ END: ADD HISTORY TO THE API CALL +++
+        
+        response_part = response.candidates[0].content.parts[0]
+
+        # --- 6. Process AI Response (Function Call or Text) ---
+        if response_part.function_call:
+            function_call = response_part.function_call
+            
+            if function_call.name == 'register_new_customer':
+                args = function_call.args
+                email = args.get('email').lower()
+                password = args.get('password')
+
+                if mongo.db.users.find_one({'email': email}):
+                    final_response_text = f"An account with the email '{email}' already exists. Please try a different email or contact support."
+                elif not password:
+                    final_response_text = "A password is required to create an account. Please provide a password."
+                else:
+                    hashed_password = bcrypt.generate_password_hash(password).decode('utf-8')
+                    new_user_doc = {
+                        "businessName": args.get('businessName'),
+                        "companyName": args.get('businessName'),
+                        "email": email,
+                        "password": hashed_password,
+                        "phoneNumber": args.get('phoneNumber'),
+                        "businessAddress": args.get('businessAddress'),
+                        "contactPerson": args.get('contactPerson'),
+                        "whatsapp_number": whatsapp_number,
+                        "is_approved": True,
+                        "is_admin": False,
+                        "created_at": datetime.utcnow()
+                    }
+                    mongo.db.users.insert_one(new_user_doc)
+                    final_response_text = f"Welcome, {args.get('businessName')}! You are now registered. You can start placing orders by telling me what you need."
+
+            elif function_call.name == 'add_items_to_cart' and user:
+                items_to_add = function_call.args.get('items', [])
+                added_items_messages = []
+                db_items = []
+
+                for item in items_to_add:
+                    p_name = item.get('product_name')
+                    product_doc = mongo.db.products.find_one({'name': {'$regex': f'^{p_name}$', '$options': 'i'}})
+                    
+                    if product_doc:
+                        db_items.append({
+                            "productId": str(product_doc['_id']),
+                            "quantity": item.get('quantity'),
+                            "mode": item.get('unit')
+                        })
+                        added_items_messages.append(f"{item.get('quantity')} {item.get('unit')} of {product_doc['name']}")
+                    else:
+                        added_items_messages.append(f"could not find '{p_name}'")
+                
+                if db_items:
+                    mongo.db.carts.update_one(
+                        {'user_email': user['email']},
+                        {'$push': {'items': {'$each': db_items}}, '$set': {'user_email': user['email'], 'updated_at': datetime.utcnow()}},
+                        upsert=True
+                    )
+                final_response_text = f"OK, I've updated your cart: I added {', '.join(added_items_messages)}. You can view your cart or place the order. What's next?"
+            
+            # +++ START: NEW HANDLER FOR REMOVING ITEMS FROM CART +++
+            elif function_call.name == 'remove_items_from_cart' and user:
+                items_to_remove = function_call.args.get('items', [])
+                removed_product_names = []
+                not_found_product_names = []
+                product_ids_to_pull = []
+
+                for item in items_to_remove:
+                    p_name = item.get('product_name')
+                    product_doc = mongo.db.products.find_one({'name': {'$regex': f'^{p_name}$', '$options': 'i'}})
+                    if product_doc:
+                        product_ids_to_pull.append(str(product_doc['_id']))
+                        removed_product_names.append(product_doc['name'])
+                    else:
+                        not_found_product_names.append(p_name)
+                
+                if product_ids_to_pull:
+                    mongo.db.carts.update_one(
+                        {'user_email': user['email']},
+                        {'$pull': {'items': {'productId': {'$in': product_ids_to_pull}}}}
+                    )
+
+                response_parts = []
+                if removed_product_names:
+                    response_parts.append(f"I have removed {', '.join(removed_product_names)} from your cart.")
+                if not_found_product_names:
+                    response_parts.append(f"I couldn't find {', '.join(not_found_product_names)} in your cart to remove.")
+                
+                if not response_parts:
+                    final_response_text = "I didn't find any items to remove based on your request."
+                else:
+                    final_response_text = " ".join(response_parts)
+            # +++ END: NEW HANDLER FOR REMOVING ITEMS FROM CART +++
+
+            # +++ START: NEW HANDLER FOR CLEARING THE CART +++
+            elif function_call.name == 'clear_cart' and user:
+                result = mongo.db.carts.update_one(
+                    {'user_email': user['email']},
+                    {'$set': {'items': []}}
+                )
+                if result.matched_count > 0 and result.modified_count > 0:
+                    final_response_text = "OK, I've cleared all items from your shopping cart."
+                else:
+                    final_response_text = "Your cart is already empty."
+            # +++ END: NEW HANDLER FOR CLEARING THE CART +++
+
+            elif function_call.name == 'get_cart_items' and user:
+                cart = mongo.db.carts.find_one({'user_email': user['email']})
+                if not cart or not cart.get('items'):
+                    final_response_text = "Your shopping cart is currently empty."
+                else:
+                    cart_items = cart.get('items', [])
+                    product_ids = [ObjectId(item['productId']) for item in cart_items if 'productId' in item]
+                    products_map = {str(p['_id']): p for p in mongo.db.products.find({'_id': {'$in': product_ids}})}
+                    
+                    item_lines = []
+                    for item in cart_items:
+                        product_details = products_map.get(item.get('productId'), {})
+                        product_name = product_details.get('name', 'Unknown Product')
+                        item_lines.append(f"- {item.get('quantity')} {item.get('mode', 'pcs')} of {product_name}")
+                    
+                    if item_lines:
+                        final_response_text = "*Here are the items in your cart:*\n" + "\n".join(item_lines)
+                    else:
+                        final_response_text = "Your shopping cart is currently empty."
+            
+            elif function_call.name == 'place_order' and user:
+                cart = mongo.db.carts.find_one({'user_email': user['email']})
+                if not cart or not cart.get('items'):
+                    final_response_text = "Your cart is empty. Please add some items before placing an order."
+                else:
+                    args = function_call.args
+                    order_doc = {
+                        'user_email': user['email'],
+                        'items': cart['items'],
+                        'delivery_date': args.get('delivery_date'),
+                        'delivery_address': user.get('businessAddress'),
+                        'mobile_number': user.get('phoneNumber'),
+                        'additional_info': args.get('additional_info', ''),
+                        'status': 'pending',
+                        'created_at': datetime.utcnow()
+                    }
+                    order_id = mongo.db.orders.insert_one(order_doc).inserted_id
+                    mongo.db.carts.delete_one({'user_email': user['email']}) # Clear cart
+                    final_response_text = f"Thank you! Your order #{str(order_id)[-6:]} has been placed for delivery on {args.get('delivery_date')}. You will receive a confirmation shortly."
+
+            elif function_call.name == 'get_my_orders' and user:
+                recent_orders = list(mongo.db.orders.find({'user_email': user['email']}).sort('created_at', -1).limit(3))
+                if not recent_orders:
+                    final_response_text = "You don't have any recent orders."
+                else:
+                    details_text = "Here are your last few orders:\n"
+                    for order in recent_orders:
+                         details_text += f"\n- Order #{str(order['_id'])[-6:]} (Status: {order.get('status')}) on {order.get('delivery_date')}"
+                    final_response_text = details_text
+
+            elif function_call.name == 'get_order_details' and user:
+                order_id_frag = function_call.args.get("order_id", "").strip()
+                all_user_orders = mongo.db.orders.find({'user_email': user['email']})
+                order_doc = None
+                for o in all_user_orders:
+                    if str(o['_id']).endswith(order_id_frag):
+                        order_doc = o
+                        break
+
+                if not order_doc:
+                    final_response_text = f"Sorry, I couldn't find an order with ID matching '{order_id_frag}'."
+                else:
+                    product_ids = [ObjectId(item['productId']) for item in order_doc.get('items', [])]
+                    products_map = {str(p['_id']): p for p in mongo.db.products.find({'_id': {'$in': product_ids}})}
+                    item_lines = [f"- {item['quantity']} {item.get('mode', 'pcs')} of {products_map.get(item['productId'], {}).get('name', 'Unknown Product')}" for item in order_doc.get('items', [])]
+                    details_text = f"*Details for Order #{str(order_doc['_id'])[-6:]}*\n*Status:* {order_doc.get('status', 'N/A')}\n*Delivery Date:* {order_doc.get('delivery_date', 'N/A')}\n\n*Items:*\n" + "\n".join(item_lines)
+                    final_response_text = details_text
+        else:
+            final_response_text = response.text
+
+        # +++ START: SAVE CONVERSATION TO DATABASE +++
+        user_message_to_save = user_message_text if user_message_text else "Audio message"
+        chat_history.append({"role": "user", "parts": [{"text": user_message_to_save}]})
+        chat_history.append({"role": "model", "parts": [{"text": final_response_text}]})
+        
+        # Keep the history from getting too long
+        if len(chat_history) > 10:
+            chat_history = chat_history[-10:]
+
+        mongo.db.whatsapp_history.update_one(
+            {'whatsapp_number': whatsapp_number},
+            {'$set': {'history': chat_history, 'updated_at': datetime.utcnow()}},
+            upsert=True
+        )
+        # +++ END: SAVE CONVERSATION TO DATABASE +++
+
+    except Exception as e:
+        current_app.logger.error(f"WhatsApp endpoint error: {e}")
+        final_response_text = "I'm having a little trouble right now. Please try again in a moment."
+
+    twilio_resp.message(final_response_text)
+    return str(twilio_resp)

app/api.py

@@ -1,234 +1,22 @@
 # your_app/api.py
 
-from flask import Blueprint, request, jsonify, current_app,redirect
+from flask import Blueprint, request, jsonify, current_app, redirect
 from bson.objectid import ObjectId
-from datetime import datetime,date
+from datetime import datetime
 from flask_jwt_extended import create_access_token, jwt_required, get_jwt_identity
 from .extensions import bcrypt, mongo
 from .xero_utils import trigger_po_creation, trigger_contact_creation
 from .email_utils import send_order_confirmation_email, send_registration_email, send_login_notification_email, send_cart_reminder_email
-from google import genai
-from google.genai import types
-from datetime import datetime, timedelta # <-- Import timedelta
-import os
-import json
-from .Company_Info import company_info
-api_bp = Blueprint('api', __name__)
 
 
-today = date.today()
-weekday_number = today.weekday()
-days_of_week = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"]
-weekday_name = days_of_week[weekday_number]
-
-# --- NEW: Define functions for the AI model ---
-
-add_items_to_cart_function = {
-    "name": "add_items_to_cart",
-    "description": "Extracts order details from a user's message to add products to their shopping cart. Can also set a delivery date.",
-    "parameters": {
-        "type": "object", "properties": { 
-            "items": { "type": "array", "description": "A list of items to add to the cart.", "items": { "type": "object", "properties": { "product_name": { "type": "string", "description": "The name of the product. This must match one of the available product names.", }, "quantity": { "type": "number", "description": "The quantity of the product.", }, "unit": { "type": "string", "description": "The unit of measure for the quantity (e.g., 'lb', 'pieces', 'box', 'bunch'). This must match one of the available units for the product.", }, }, "required": ["product_name", "quantity", "unit"], }, },
-            # --- START NEW PARAMETER ---
-            "delivery_date": {
-                "type": "string",
-                "description": "The desired delivery date for the order, in YYYY-MM-DD format. The user must explicitly state a date.",
-            }
-            # --- END NEW PARAMETER ---
-        },
-        "required": ["items"],
-    },
-}
-
-navigate_to_page_function = {
-    "name": "navigate_to_page",
-    "description": "Provides a button in the chat to navigate the user to a specific page.",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "page": {
-                "type": "string",
-                "description": "The destination page. Must be one of the allowed pages.",
-                "enum": ["orders", "cart", "catalog", "home", "about", "care", "login", "register"]
-            }
-        },
-        "required": ["page"]
-    }
-}
-
-get_my_orders_function = {
-    "name": "get_my_orders",
-    "description": "Retrieves a summary of the user's most recent orders to display in the chat.",
-    "parameters": {"type": "object", "properties": {}}
-}
-
-# --- NEW: Function to get details of a specific order ---
-get_order_details_function = {
-    "name": "get_order_details",
-    "description": "Retrieves the specific items and details for a single order based on its ID.",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "order_id": {
-                "type": "string",
-                "description": "The ID of the order to fetch. Can be the full ID or the last 6 characters."
-            }
-        },
-        "required": ["order_id"]
-    }
-}
-
-# --- NEW: Function to initiate an order cancellation ---
-cancel_order_function = {
-    "name": "cancel_order",
-    "description": "Proposes to cancel an order based on its ID. This requires user confirmation.",
-    "parameters": {
-        "type": "object",
-        "properties": {
-            "order_id": {
-                "type": "string",
-                "description": "The ID of the order to be cancelled. Can be the full ID or the last 6 characters."
-            }
-        },
-        "required": ["order_id"]
-    }
-}
 
+api_bp = Blueprint('api', __name__)
 
 @api_bp.route('/clear')
 def clear_all():
-    # mongo.db.users.delete_many({})
     mongo.db.orders.delete_many({})
     return "✅"
 
-
-@api_bp.route('/chat', methods=['POST'])
-@jwt_required()
-def handle_ai_chat():
-    # try:
-        # --- 1. Universal Setup ---
-        user_email = get_jwt_identity()
-        MODES = ['case', 'bag', 'piece', 'tray', 'weight']
-        all_products = list(mongo.db.products.find({}, {'name': 1, 'unit': 1, '_id': 0}))
-        product_context_list = []
-        for p in all_products:
-            if 'name' not in p: continue
-            unit_string = p.get('unit', '').lower()
-            available_modes = [mode for mode in MODES if mode in unit_string]
-            if available_modes:
-                product_context_list.append(f"{p['name']} (available units: {', '.join(available_modes)})")
-            else:
-                product_context_list.append(p['name'])        
-        client = genai.Client(api_key="AIzaSyB7yKIdfW7Umv62G47BCdJjoHTJ9TeiAko")
-        
-        # --- START MODIFICATION: Add all new functions to the tools list and calculate dates ---
-        current_date = datetime.now().strftime('%Y-%m-%d')
-        min_delivery_date = (datetime.now() + timedelta(days=2)).strftime('%Y-%m-%d')
-        
-        tools = types.Tool(function_declarations=[
-            add_items_to_cart_function,
-            navigate_to_page_function,
-            get_my_orders_function,
-            cancel_order_function
-        ])
-        config = types.GenerateContentConfig(tools=[tools])
-
-        # --- 2. UPDATED: Prepare the main instruction prompt for the AI ---
-        instruction_prompt = f"""
-        You are an expert assistant for a food wholesale company. Your capabilities include:
-        1.  Ordering & Delivery Date: Use `add_items_to_cart` to add items to the user's cart. You can also set a delivery date if the user specifies one. The current date is {current_date,weekday_name}. The earliest possible delivery date is {min_delivery_date}. If the user asks for a date earlier than this, inform them of the constraint and ask them to provide a new valid date.
-        2.  Product List: {'; '.join(product_context_list)}.
-        3.  Page Navigation: Use `navigate_to_page` if the user wants to GO to a page like 'my orders', 'about us', 'customer care', 'login', etc.
-        4.  Order Summary: Use `get_my_orders` to LIST a summary of recent orders in the chat.
-        5.  Order Cancellation: Use `cancel_order` to start the cancellation process for a SPECIFIC order. The user must provide an order ID.
-        6.  The Capability You must use AT ALL TIMES: Luring the customer to place an order and persuading them to buy more products.
-        *Always Brag about the quality of our products and our customer service and try to persuade the customer to place an order.*
-
-        
-        Company Info for general questions: {company_info}
-        *** IMPORTANT: When providing a text-based answer, YOU MUST USE MARKDOWN. Use headings, bold text, bullet points (`* item`), and other formatting to make your answers clear, structured, and easy to read. ***
-        Always refer to the chat history for context. **If information is missing, ask for it.**
-        """
-        # --- END MODIFICATION ---
-
-        # --- 3. Process incoming user input (This section remains the same) ---
-        user_input_part, history_data = None, []
-        if 'multipart/form-data' in request.content_type:
-            audio_file = request.files.get('audio')
-            history_json = request.form.get('history', '[]')
-            history_data = json.loads(history_json)
-            user_input_part = types.Part.from_bytes(data=audio_file.read(), mime_type=audio_file.mimetype.split(';')[0])
-        elif 'application/json' in request.content_type:
-            json_data = request.get_json()
-            history_data = json_data.get('history', [])
-            user_input_part = json_data.get('message')
-        else: return jsonify({"msg": "Unsupported format"}), 415
-        
-        # --- 4. Make API call (This section remains the same) ---
-        history_contents = [msg.get('data') for msg in history_data if msg.get('type') == 'text']
-        final_contents = [instruction_prompt] + history_contents + [user_input_part]
-        response = client.models.generate_content(model="gemini-2.5-flash", contents=final_contents, config=config)
-        response_part = response.candidates[0].content.parts[0]
-        
-        # --- 5. UPDATED: Process the response with new function handlers ---
-        if response_part.function_call:
-            function_call = response_part.function_call
-            
-            if function_call.name == "add_items_to_cart":
-                # --- START MODIFICATION ---
-                # Create a payload with both items and the potential delivery date
-                proposal_data = {
-                    "items": function_call.args.get('items', []),
-                    "delivery_date": function_call.args.get('delivery_date') # This will be None if not provided by the AI
-                }
-                return jsonify({"type": "order_proposal", "data": proposal_data}), 200
-                # --- END MODIFICATION ---
-
-            elif function_call.name == "navigate_to_page":
-                page = function_call.args.get('page')
-                path_map = {'orders': '/order', 'cart': '/cart', 'catalog': '/catalog', 'home': '/', 'about': '/about', 'care': '/care', 'login': '/login', 'register': '/register'}
-                text_map = {'orders': 'Go to My Orders', 'cart': 'Go to Cart', 'catalog': 'Go to Catalog', 'home': 'Go to Homepage', 'about': 'Go to About Page', 'care': 'Go to Customer Care', 'login': 'Go to Login', 'register': 'Go to Sign Up'}
-                if page in path_map:
-                    return jsonify({"type": "navigation_proposal", "data": {"path": path_map[page], "button_text": text_map[page], "prompt_text": f"Sure, let's go to the {page} page."}}), 200
-
-            elif function_call.name == "get_my_orders":
-                recent_orders = list(mongo.db.orders.find({'user_email': user_email}).sort('created_at', -1).limit(5))
-                if not recent_orders: return jsonify({"type": "text", "data": "You have no recent orders."}), 200
-                summary = "\n".join([f"- Order Details for #{str(o['_id'])[-6:]}\n{str(o)} \n\n" for o in recent_orders])
-                details_text=""
-                for order in recent_orders:
-                    product_ids = [ObjectId(item['productId']) for item in order.get('items', [])]
-                    products_map = {str(p['_id']): p for p in mongo.db.products.find({'_id': {'$in': product_ids}})}
-                    item_lines = [f"- {item['quantity']} {item.get('mode', 'pcs')} of {products_map.get(item['productId'], {}).get('name', 'Unknown Product')}" for item in order.get('items', [])]
-                    details_text += f"\n\n **Details for Order #{str(order['_id'])[-4:]}** _(Delivery Status: {order.get('status')})_:\n" + "\n".join(item_lines)
-                return jsonify({"type": "text", "data": "**Here are your recent orders**:\n" + details_text}), 200
-
-            elif function_call.name == "cancel_order":
-                order_id_frag = function_call.args.get("order_id", "").strip()
-                
-                all_user_orders = mongo.db.orders.find({'user_email': user_email})
-                order = None
-                for o in all_user_orders:
-                    if str(o['_id']).endswith(order_id_frag):
-                        order = o
-                        break
-                if not order: return jsonify({"type": "text", "data": f"Sorry, I couldn't find an order with ID matching '{order_id_frag}'."}), 200
-                if order.get('status') not in ['pending', 'confirmed']: return jsonify({"type": "text", "data": f"Order #{str(order['_id'])[-4:]} cannot be cancelled as its status is '{order.get('status')}'."}), 200
-                
-                return jsonify({
-                    "type": "cancellation_proposal",
-                    "data": {"order_id": str(order['_id']), "prompt_text": f"Are you sure you want to cancel order #{str(order['_id'])[-4:]}?"}
-                }), 200
-
-        # Fallback to text response
-        return jsonify({"type": "text", "data": response.text}), 200
-
-    # except Exception as e:
-    #     current_app.logger.error(f"Gemini API Error: {e}")
-    #     return jsonify({"msg": "Sorry, I'm having trouble. Please try again."}), 500
-
-# --- The rest of your api.py file remains unchanged ---
-
 @api_bp.route('/register', methods=['POST'])
 def register():
     data = request.get_json()
@@ -246,7 +34,7 @@ def register():
     user_document = data.copy()
     user_document['password'] = hashed_password
     user_document['company_name'] = company_name
-    user_document['is_approved'] = False
+    user_document['is_approved'] = True
     user_document['is_admin'] = False
     
     mongo.db.users.insert_one(user_document)
@@ -259,13 +47,14 @@ def register():
 
     return jsonify({"msg": "Registration successful! Your application is being processed."}), 201
 
+# ... (the rest of your api.py file remains unchanged)
 @api_bp.route('/login', methods=['POST'])
 def login():
     data = request.get_json()
     email, password = data.get('email'), data.get('password')
     user = mongo.db.users.find_one({'email': email})
 
-    if user and bcrypt.check_password_hash(user['password'], password):
+    if user and user.get('password') and bcrypt.check_password_hash(user['password'], password):
         if not user.get('is_approved', False): return jsonify({"msg": "Account pending approval"}), 403
         
         try:
@@ -274,7 +63,7 @@ def login():
             current_app.logger.error(f"Failed to send login notification email to {email}: {e}")
 
         access_token = create_access_token(identity=email)
-        return jsonify(access_token=access_token, email=user['email'], companyName=user['company_name'],contactPerson=user.get('contactPerson', '')) , 200
+        return jsonify(access_token=access_token, email=user['email'], companyName=user['businessName'],contactPerson=user.get('contactPerson', '')) , 200
     
     return jsonify({"msg": "Bad email or password"}), 401
 
@@ -333,7 +122,6 @@ def handle_cart():
         })
 
     if request.method == 'POST':
-        # --- START MODIFICATION: Handle partial updates to prevent data loss ---
         data = request.get_json()
         
         update_doc = {
@@ -341,7 +129,6 @@ def handle_cart():
             'updated_at': datetime.utcnow()
         }
         
-        # Only update fields that are present in the request
         if 'items' in data:
             update_doc['items'] = data['items']
 
@@ -353,7 +140,6 @@ def handle_cart():
             {'$set': update_doc},
             upsert=True
         )
-        # --- END MODIFICATION ---
         return jsonify({"msg": "Cart updated successfully"})
 
 @api_bp.route('/orders', methods=['GET', 'POST'])
@@ -496,21 +282,14 @@ def cancel_order(order_id):
     
     return jsonify({"msg": "Order has been cancelled."}), 200
 
-# +++ START: NEW /sendmail ENDPOINT +++
 @api_bp.route('/sendmail', methods=['GET'])
 def send_cart_reminders():
-    """
-    Triggers reminder emails to all users with items in their cart.
-    This can be triggered manually or by a scheduled job (e.g., cron).
-    """
     try:
-        # 1. Find all carts that have items
         carts_with_items = list(mongo.db.carts.find({'items': {'$exists': True, '$ne': []}}))
         
         if not carts_with_items:
             return jsonify({"msg": "No users with pending items in cart."}), 200
 
-        # 2. Collect all user emails and product IDs for efficient bulk fetching
         user_emails = [cart['user_email'] for cart in carts_with_items]
         all_product_ids = {
             ObjectId(item['productId']) 
@@ -518,24 +297,20 @@ def send_cart_reminders():
             for item in cart.get('items', [])
         }
 
-        # 3. Fetch all required users and products from the DB
         users_cursor = mongo.db.users.find({'email': {'$in': user_emails}})
         products_cursor = mongo.db.products.find({'_id': {'$in': list(all_product_ids)}})
 
-        # 4. Create lookup maps for fast access
         users_map = {user['email']: user for user in users_cursor}
         products_map = {str(prod['_id']): prod for prod in products_cursor}
 
         emails_sent_count = 0
         
-        # 5. Iterate through carts, prepare data, and send emails
         for cart in carts_with_items:
             user = users_map.get(cart['user_email'])
             if not user:
                 current_app.logger.warning(f"Cart found for non-existent user: {cart['user_email']}")
                 continue
 
-            # Populate cart items with full product details
             populated_items = []
             for item in cart.get('items', []):
                 product_details = products_map.get(item['productId'])
@@ -560,7 +335,6 @@ def send_cart_reminders():
     except Exception as e:
         current_app.logger.error(f"Error in /sendmail endpoint: {e}")
         return jsonify({"msg": "An internal error occurred while sending reminders."}), 500
-# +++ END: NEW /sendmail ENDPOINT +++
 
 @api_bp.route('/admin/users/approve/<user_id>', methods=['POST'])
 @jwt_required()
@@ -568,168 +342,6 @@ def approve_user(user_id):
     mongo.db.users.update_one({'_id': ObjectId(user_id)}, {'$set': {'is_approved': True}})
     return jsonify({"msg": f"User {user_id} approved"})
 
-
-# Default data for initialization if pages don't exist in the database
-DEFAULT_ABOUT_DATA = {
-    "_id": "about",
-    "title": "Our Commitment to You",
-    "slogan": "It's more than produce — it's about being your daily, trusted partner.",
-    "paragraphs": [
-        "Welcome to Matax Express Ltd, your trusted wholesale produce provider serving the Greater Toronto Area for over 30 years!",
-        "At Matax Express Ltd, our commitment to service has always been at the core of what we do. From day one, we’ve focused on understanding the needs of our customers and working tirelessly to meet them. Whether delivering to bustling restaurants, local markets, or retail stores, we strive to ensure your success and satisfaction every step of the way.",
-        "We’re proud of the journey we’ve taken over the past three decades, and we also understand the importance of continuous improvement. Your feedback has been crucial in helping us grow and adapt, and we are working hard to ensure every interaction reflects the high standard of service you deserve.",
-        "While freshness remains an important priority, we are equally dedicated to creating a service experience that exceeds expectations. From reliable deliveries to personalized support, our team is here to make partnering with Matax Express Ltd seamless and efficient.",
-        "For us, it’s not just about providing produce—it’s about being a dependable partner that you can count on daily. We look forward to building stronger relationships and delivering better service for years to come."
-    ]
-}
-
-DEFAULT_CONTACT_DATA = {
-    "_id": "contact",
-    "title": "Contact Us",
-    "intro": "We're here to help! Reach out to us through any of the channels below. We aim to respond to all inquiries within 24 business hours.",
-    "details": [
-        {"type": "Phone Support", "value": "+1 (800) 123-4567"},
-        {"type": "Email Support", "value": "support@mataxexpress.com"},
-        {"type": "Business Hours", "value": "Monday - Friday, 9:00 AM - 5:00 PM (EST)"},
-        {"type": "Mailing Address", "value": "123 Fresh Produce Lane, Farmville, ST 54321"}
-    ]
-}
-
-@api_bp.route('/pages/<page_name>', methods=['GET'])
-def get_page_content(page_name):
-    """API endpoint for the frontend to fetch page content."""
-    content = mongo.db.pages.find_one({'_id': page_name})
-    if not content:
-        # If content doesn't exist, create it from default and return it
-        if page_name == 'about':
-            mongo.db.pages.insert_one(DEFAULT_ABOUT_DATA)
-            content = DEFAULT_ABOUT_DATA
-        elif page_name == 'contact':
-            mongo.db.pages.insert_one(DEFAULT_CONTACT_DATA)
-            content = DEFAULT_CONTACT_DATA
-        else:
-            return jsonify({"msg": "Page not found"}), 404
-    # Ensure _id is a string if it's an ObjectId
-    if '_id' in content and not isinstance(content['_id'], str):
-        content['_id'] = str(content['_id'])
-    return jsonify(content)
-
-@api_bp.route('/pages/update', methods=['POST'])
-def update_page_content():
-    """Handles form submission from the /update UI to save changes."""
-    page_name = request.form.get('page_name')
-    if page_name == 'about':
-        paragraphs = request.form.get('paragraphs', '').strip().split('\n')
-        paragraphs = [p.strip() for p in paragraphs if p.strip()]
-        
-        update_data = {
-            "title": request.form.get('title'),
-            "slogan": request.form.get('slogan'),
-            "paragraphs": paragraphs
-        }
-    elif page_name == 'contact':
-        update_data = {
-            "title": request.form.get('title'),
-            "intro": request.form.get('intro'),
-            "details": [
-                {"type": "Phone Support", "value": request.form.get('phone_value')},
-                {"type": "Email Support", "value": request.form.get('email_value')},
-                {"type": "Business Hours", "value": request.form.get('hours_value')},
-                {"type": "Mailing Address", "value": request.form.get('address_value')}
-            ]
-        }
-    else:
-        return redirect('/api/update')
-
-    mongo.db.pages.update_one(
-        {'_id': page_name},
-        {'$set': update_data},
-        upsert=True
-    )
-    return redirect('/api/update')
-
-@api_bp.route('/update', methods=['GET'])
-def update_ui():
-    """Serves the simple HTML UI for editing page content."""
-    about_data = mongo.db.pages.find_one({'_id': 'about'}) or DEFAULT_ABOUT_DATA
-    contact_data = mongo.db.pages.find_one({'_id': 'contact'}) or DEFAULT_CONTACT_DATA
-
-    about_paragraphs_text = "\n".join(about_data.get('paragraphs', []))
-    contact_details = {item['type']: item['value'] for item in contact_data.get('details', [])}
-
-    html = f"""
-    <!DOCTYPE html>
-    <html lang="en">
-    <head>
-        <meta charset="UTF-8">
-        <meta name="viewport" content="width=device-width, initial-scale=1.0">
-        <title>Update Page Content</title>
-        <style>
-            body {{ font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif; margin: 2em; background-color: #f8f9fa; color: #212529; }}
-            .container {{ max-width: 800px; margin: auto; }}
-            h1, h2 {{ color: #343a40; border-bottom: 2px solid #dee2e6; padding-bottom: 0.5em; }}
-            form {{ background: white; padding: 2em; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.05); margin-bottom: 2em; }}
-            label {{ display: block; margin-top: 1em; margin-bottom: 0.5em; font-weight: bold; color: #495057; }}
-            input[type="text"], textarea {{ width: 100%; padding: 0.8em; border: 1px solid #ced4da; border-radius: 4px; box-sizing: border-box; font-size: 1rem; }}
-            textarea {{ height: 250px; resize: vertical; }}
-            button {{ background-color: #007bff; color: white; padding: 0.8em 1.5em; border: none; border-radius: 4px; cursor: pointer; font-size: 1em; font-weight: bold; }}
-            button:hover {{ background-color: #0056b3; }}
-        </style>
-    </head>
-    <body>
-        <div class="container">
-            <h1>Update Website Content</h1>
-
-            <!-- About Us Page Form -->
-            <h2>About Us Page</h2>
-            <form action="/api/pages/update" method="post">
-                <input type="hidden" name="page_name" value="about">
-                <label for="about_title">Title:</label>
-                <input type="text" id="about_title" name="title" value="{about_data.get('title', '')}">
-
-                <label for="about_slogan">Slogan:</label>
-                <input type="text" id="about_slogan" name="slogan" value="{about_data.get('slogan', '')}">
-                
-                <label for="about_paragraphs">Paragraphs (one paragraph per line):</label>
-                <textarea id="about_paragraphs" name="paragraphs">{about_paragraphs_text}</textarea>
-                
-                <br><br>
-                <button type="submit">Update About Page</button>
-            </form>
-
-            <!-- Contact Page Form -->
-            <h2>Customer Care Page</h2>
-            <form action="/api/pages/update" method="post">
-                <input type="hidden" name="page_name" value="contact">
-
-                <label for="contact_title">Title:</label>
-                <input type="text" id="contact_title" name="title" value="{contact_data.get('title', '')}">
-
-                <label for="contact_intro">Intro Text:</label>
-                <input type="text" id="contact_intro" name="intro" value="{contact_data.get('intro', '')}">
-
-                <label for="phone_value">Phone Support:</label>
-                <input type="text" id="phone_value" name="phone_value" value="{contact_details.get('Phone Support', '')}">
-
-                <label for="email_value">Email Support:</label>
-                <input type="text" id="email_value" name="email_value" value="{contact_details.get('Email Support', '')}">
-
-                <label for="hours_value">Business Hours:</label>
-                <input type="text" id="hours_value" name="hours_value" value="{contact_details.get('Business Hours', '')}">
-
-                <label for="address_value">Mailing Address:</label>
-                <input type="text" id="address_value" name="address_value" value="{contact_details.get('Mailing Address', '')}">
-                
-                <br><br>
-                <button type="submit">Update Contact Page</button>
-            </form>
-        </div>
-    </body>
-    </html>
-    """
-    return html
-
-
 # +++ START: NEW ENDPOINT FOR ITEM REQUESTS +++
 @api_bp.route('/request-item', methods=['POST'])
 @jwt_required()

app/page_features.py

@@ -0,0 +1,167 @@
+# your_app/page_features.py
+
+from flask import Blueprint, request, jsonify, redirect
+from .extensions import mongo
+
+page_bp = Blueprint('pages', __name__)
+
+# Default data for initialization if pages don't exist in the database
+DEFAULT_ABOUT_DATA = {
+    "_id": "about",
+    "title": "Our Commitment to You",
+    "slogan": "It's more than produce — it's about being your daily, trusted partner.",
+    "paragraphs": [
+        "Welcome to Matax Express Ltd, your trusted wholesale produce provider serving the Greater Toronto Area for over 30 years!",
+        "At Matax Express Ltd, our commitment to service has always been at the core of what we do. From day one, we’ve focused on understanding the needs of our customers and working tirelessly to meet them. Whether delivering to bustling restaurants, local markets, or retail stores, we strive to ensure your success and satisfaction every step of the way.",
+        "We’re proud of the journey we’ve taken over the past three decades, and we also understand the importance of continuous improvement. Your feedback has been crucial in helping us grow and adapt, and we are working hard to ensure every interaction reflects the high standard of service you deserve.",
+        "While freshness remains an important priority, we are equally dedicated to creating a service experience that exceeds expectations. From reliable deliveries to personalized support, our team is here to make partnering with Matax Express Ltd seamless and efficient.",
+        "For us, it’s not just about providing produce—it’s about being a dependable partner that you can count on daily. We look forward to building stronger relationships and delivering better service for years to come."
+    ]
+}
+
+DEFAULT_CONTACT_DATA = {
+    "_id": "contact",
+    "title": "Contact Us",
+    "intro": "We're here to help! Reach out to us through any of the channels below. We aim to respond to all inquiries within 24 business hours.",
+    "details": [
+        {"type": "Phone Support", "value": "+1 (800) 123-4567"},
+        {"type": "Email Support", "value": "support@mataxexpress.com"},
+        {"type": "Business Hours", "value": "Monday - Friday, 9:00 AM - 5:00 PM (EST)"},
+        {"type": "Mailing Address", "value": "123 Fresh Produce Lane, Farmville, ST 54321"}
+    ]
+}
+
+@page_bp.route('/<page_name>', methods=['GET'])
+def get_page_content(page_name):
+    """API endpoint for the frontend to fetch page content."""
+    content = mongo.db.pages.find_one({'_id': page_name})
+    if not content:
+        # If content doesn't exist, create it from default and return it
+        if page_name == 'about':
+            mongo.db.pages.insert_one(DEFAULT_ABOUT_DATA)
+            content = DEFAULT_ABOUT_DATA
+        elif page_name == 'contact':
+            mongo.db.pages.insert_one(DEFAULT_CONTACT_DATA)
+            content = DEFAULT_CONTACT_DATA
+        else:
+            return jsonify({"msg": "Page not found"}), 404
+    # Ensure _id is a string if it's an ObjectId
+    if '_id' in content and not isinstance(content['_id'], str):
+        content['_id'] = str(content['_id'])
+    return jsonify(content)
+
+@page_bp.route('/update', methods=['POST'])
+def update_page_content():
+    """Handles form submission from the /update UI to save changes."""
+    page_name = request.form.get('page_name')
+    if page_name == 'about':
+        paragraphs = request.form.get('paragraphs', '').strip().split('\n')
+        paragraphs = [p.strip() for p in paragraphs if p.strip()]
+        
+        update_data = {
+            "title": request.form.get('title'),
+            "slogan": request.form.get('slogan'),
+            "paragraphs": paragraphs
+        }
+    elif page_name == 'contact':
+        update_data = {
+            "title": request.form.get('title'),
+            "intro": request.form.get('intro'),
+            "details": [
+                {"type": "Phone Support", "value": request.form.get('phone_value')},
+                {"type": "Email Support", "value": request.form.get('email_value')},
+                {"type": "Business Hours", "value": request.form.get('hours_value')},
+                {"type": "Mailing Address", "value": request.form.get('address_value')}
+            ]
+        }
+    else:
+        return redirect('/api/update')
+
+    mongo.db.pages.update_one(
+        {'_id': page_name},
+        {'$set': update_data},
+        upsert=True
+    )
+    return redirect('/api/update')
+
+@page_bp.route('/update_ui', methods=['GET'])
+def update_ui():
+    """Serves the simple HTML UI for editing page content."""
+    # Note: Route changed to /update_ui to avoid conflict with /pages/update
+    about_data = mongo.db.pages.find_one({'_id': 'about'}) or DEFAULT_ABOUT_DATA
+    contact_data = mongo.db.pages.find_one({'_id': 'contact'}) or DEFAULT_CONTACT_DATA
+
+    about_paragraphs_text = "\n".join(about_data.get('paragraphs', []))
+    contact_details = {item['type']: item['value'] for item in contact_data.get('details', [])}
+
+    html = f"""
+    <!DOCTYPE html>
+    <html lang="en">
+    <head>
+        <meta charset="UTF-8">
+        <meta name="viewport" content="width=device-width, initial-scale=1.0">
+        <title>Update Page Content</title>
+        <style>
+            body {{ font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Helvetica Neue", Arial, sans-serif; margin: 2em; background-color: #f8f9fa; color: #212529; }}
+            .container {{ max-width: 800px; margin: auto; }}
+            h1, h2 {{ color: #343a40; border-bottom: 2px solid #dee2e6; padding-bottom: 0.5em; }}
+            form {{ background: white; padding: 2em; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.05); margin-bottom: 2em; }}
+            label {{ display: block; margin-top: 1em; margin-bottom: 0.5em; font-weight: bold; color: #495057; }}
+            input[type="text"], textarea {{ width: 100%; padding: 0.8em; border: 1px solid #ced4da; border-radius: 4px; box-sizing: border-box; font-size: 1rem; }}
+            textarea {{ height: 250px; resize: vertical; }}
+            button {{ background-color: #007bff; color: white; padding: 0.8em 1.5em; border: none; border-radius: 4px; cursor: pointer; font-size: 1em; font-weight: bold; }}
+            button:hover {{ background-color: #0056b3; }}
+        </style>
+    </head>
+    <body>
+        <div class="container">
+            <h1>Update Website Content</h1>
+
+            <!-- About Us Page Form -->
+            <h2>About Us Page</h2>
+            <form action="/api/pages/update" method="post">
+                <input type="hidden" name="page_name" value="about">
+                <label for="about_title">Title:</label>
+                <input type="text" id="about_title" name="title" value="{about_data.get('title', '')}">
+
+                <label for="about_slogan">Slogan:</label>
+                <input type="text" id="about_slogan" name="slogan" value="{about_data.get('slogan', '')}">
+                
+                <label for="about_paragraphs">Paragraphs (one paragraph per line):</label>
+                <textarea id="about_paragraphs" name="paragraphs">{about_paragraphs_text}</textarea>
+                
+                <br><br>
+                <button type="submit">Update About Page</button>
+            </form>
+
+            <!-- Contact Page Form -->
+            <h2>Customer Care Page</h2>
+            <form action="/api/pages/update" method="post">
+                <input type="hidden" name="page_name" value="contact">
+
+                <label for="contact_title">Title:</label>
+                <input type="text" id="contact_title" name="title" value="{contact_data.get('title', '')}">
+
+                <label for="contact_intro">Intro Text:</label>
+                <input type="text" id="contact_intro" name="intro" value="{contact_data.get('intro', '')}">
+
+                <label for="phone_value">Phone Support:</label>
+                <input type="text" id="phone_value" name="phone_value" value="{contact_details.get('Phone Support', '')}">
+
+                <label for="email_value">Email Support:</label>
+                <input type="text" id="email_value" name="email_value" value="{contact_details.get('Email Support', '')}">
+
+                <label for="hours_value">Business Hours:</label>
+                <input type="text" id="hours_value" name="hours_value" value="{contact_details.get('Business Hours', '')}">
+
+                <label for="address_value">Mailing Address:</label>
+                <input type="text" id="address_value" name="address_value" value="{contact_details.get('Mailing Address', '')}">
+                
+                <br><br>
+                <button type="submit">Update Contact Page</button>
+            </form>
+        </div>
+    </body>
+    </html>
+    """
+    return html

app/xero_utils.py

@@ -269,5 +269,4 @@ def trigger_po_creation(order_details):
         thread.daemon = True
         thread.start()
     except Exception as e:
-        logger.error("Failed to start Xero PO creation thread for order %s. Error: %s", order_details.get('order_id'), e)
-
+        logger.error("Failed to start Xero PO creation thread for order %s. Error: %s", order_details.get('order_id'), e)

twillio_gemini_api.py

@@ -1,56 +1,155 @@
 # app.py
 
 import os
+import requests  # To download the audio file from Twilio's URL
 from flask import Flask, request
 from dotenv import load_dotenv
-import google.generativeai as genai
+# Use the new top-level import and types
+from google import genai
+from google.genai import types
 from twilio.twiml.messaging_response import MessagingResponse
 
-# Load environment variables from .env file
-load_dotenv()
+# --- Define Calculator Functions for the Model to Use ---
+# The SDK uses the function signature (name, parameters) and docstring
+# to tell the model how and when to use these tools.
+
+def add(a: float, b: float):
+    """
+    Adds two numbers together.
+
+    Args:
+        a: The first number.
+        b: The second number.
+    """
+    print(f"Tool Call: add(a={a}, b={b})")
+    return a + b
+
+def subtract(a: float, b: float):
+    """
+    Subtracts the second number from the first.
+
+    Args:
+        a: The number to subtract from.
+        b: The number to subtract.
+    """
+    print(f"Tool Call: subtract(a={a}, b={b})")
+    return a - b
+
+def multiply(a: float, b: float):
+    """
+    Multiplies two numbers.
+
+    Args:
+        a: The first number.
+        b: The second number.
+    """
+    print(f"Tool Call: multiply(a={a}, b={b})")
+    return a * b
+
+def divide(a: float, b: float):
+    """
+    Divides the first number by the second.
+
+    Args:
+        a: The numerator.
+        b: The denominator.
+    """
+    print(f"Tool Call: divide(a={a}, b={b})")
+    if b == 0:
+        return "Error: Cannot divide by zero."
+    return a / b
+
+# A list of all the functions the model can call
+calculator_tools = [add, subtract, multiply, divide]
+
 
 # --- Initialize Flask App and APIs ---
+
+# Load environment variables from .env file
+load_dotenv(r"C:\Users\Vaibhav Arora\Documents\MyExperimentsandCodes\APPS_WEBSITES\CANADA_WHOLESALE_PROJECT\GITHUB_REPOS\mvp-vue\wholesale-grocery-app\AIAPPS\.env")
+
 app = Flask(__name__)
 
-# Configure the Gemini API
+# Configure the Gemini API and initialize the client
 try:
-    genai.configure(api_key=os.environ.get("GEMINI_API_KEY"))
-    # Initialize the Gemini-1.5-Pro model
-    # Use 'gemini-1.5-pro-latest' for the most recent version
-    model = genai.GenerativeModel('gemini-1.5-pro-latest')
-    print("Gemini model initialized successfully.")
+    client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
+    print("Gemini client and calculator tools initialized successfully.")
 except Exception as e:
-    print(f"Error initializing Gemini model: {e}")
-    model = None
+    print(f"Error initializing Gemini client: {e}")
+    client = None
+
 
 # --- Define the Webhook Endpoint ---
 @app.route("/sms", methods=['POST'])
 def sms_reply():
-    """Respond to incoming messages with a Gemini-generated response."""
+    """Respond to incoming text or audio messages, using calculator functions if needed."""
     
-    # Get the message body from the incoming Twilio request
-    incoming_msg = request.values.get('Body', '').strip()
-    print(f"Received message: '{incoming_msg}'")
-
-    # Start the TwiML response
     twilio_resp = MessagingResponse()
 
-    if not model:
-        twilio_resp.message("The AI model is not configured correctly. Please check the server logs.")
+    if not client:
+        twilio_resp.message("The AI client is not configured correctly. Please check the server logs.")
         return str(twilio_resp)
 
-    if not incoming_msg:
-        twilio_resp.message("Please send a message to get a response.")
+    # Prepare the contents list for the Gemini API call
+    contents = []
+    
+    # Check if the incoming message contains media
+    num_media = int(request.values.get('NumMedia', 0))
+
+    if num_media > 0:
+        media_url = request.values.get('MediaUrl0')
+        mime_type = request.values.get('MediaContentType0')
+
+        # Process only if the media is audio
+        if 'audio' in mime_type:
+            print(f"Received audio message. URL: {media_url}, MIME Type: {mime_type}")
+            
+            # Download the audio file from the Twilio URL
+            audio_response = requests.get(media_url)
+            
+            if audio_response.status_code == 200:
+                audio_bytes = audio_response.content
+                audio_part = types.Part.from_bytes(data=audio_bytes, mime_type=mime_type)
+                prompt = "Please transcribe this audio. If it contains a calculation or a question, please answer it."
+                contents = [prompt, audio_part]
+            else:
+                error_message = "Sorry, I couldn't download the audio file to process it. Please try again."
+                twilio_resp.message(error_message)
+                return str(twilio_resp)
+        else:
+            # Handle non-audio media like images or videos
+            twilio_resp.message("Sorry, I can only process text and audio messages.")
+            return str(twilio_resp)
+
+    else:
+        # Fallback to text message processing
+        incoming_msg = request.values.get('Body', '').strip()
+        print(f"Received text message: '{incoming_msg}'")
+        if not incoming_msg:
+            twilio_resp.message("Please send a text or audio message to get a response.")
+            return str(twilio_resp)
+        contents = [incoming_msg]
+
+    if not contents:
+        twilio_resp.message("Could not determine content to process. Please send a message.")
         return str(twilio_resp)
         
     try:
-        # Send the message to Gemini API and get the response
-        print("Sending message to Gemini...")
-        gemini_response = model.generate_content(incoming_msg)
+        print("Sending content to Gemini with calculator tools...")
+
+        # Configure the request to use our calculator functions
+        config = types.GenerateContentConfig(tools=calculator_tools)
+
+        # The SDK handles the multi-turn process for function calling automatically
+        gemini_response = client.models.generate_content(
+            model="gemini-2.5-flash",
+            contents=contents, # This will contain either text or [prompt, audio_part]
+            config=config,
+        )
         
-        # Extract the text part of the response
+        # This is the final text answer after any function calls have been resolved.
         ai_answer = gemini_response.text
-        print(f"Gemini response: '{ai_answer}'")
+        print(f"Gemini final response: '{ai_answer}'")
         
         # Add the Gemini response to the TwiML message
         twilio_resp.message(ai_answer)
@@ -61,12 +160,9 @@ def sms_reply():
         error_message = "Sorry, I'm having trouble connecting to my brain right now. Please try again later."
         twilio_resp.message(error_message)
 
-    # Return the TwiML response as a string
     return str(twilio_resp)
 
 
 # --- Run the Flask App ---
 if __name__ == "__main__":
-    # The debug=True option is for development only.
-    # In production, use a proper WSGI server like Gunicorn or Waitress.
     app.run(debug=True, port=5000)