Update app.py

app.py

@@ -231,10 +231,101 @@ def set_item_availability(item_name: str, available: bool) -> str:
     finally:
         release_db_connection(conn)
 
-# --- System Instructions (unchanged) ---
+# --- NEW: Full CRUD tools for admin chat ---
+def get_menu_items() -> str:
+    """
+    Returns the complete menu with IDs, names, descriptions, prices, and availability statuses.
+    """
+    menu = load_menu()
+    return json.dumps(menu, ensure_ascii=False)
+
+def add_menu_item(name: str, description: str = "", price: str = "", available: bool = True) -> str:
+    """
+    Add a new item to the cafe menu. Returns confirmation message.
+    """
+    conn = get_db_connection()
+    try:
+        with conn.cursor() as cur:
+            cur.execute(
+                "INSERT INTO menu (name, description, price, available) VALUES (%s, %s, %s, %s) RETURNING id;",
+                (name.strip(), description.strip(), price.strip(), available)
+            )
+            new_id = cur.fetchone()[0]
+        conn.commit()
+        return f"آیتم '{name}' با شناسه {new_id} با موفقیت اضافه شد."
+    except Exception as e:
+        conn.rollback()
+        return f"خطا در افزودن آیتم: {str(e)}"
+    finally:
+        release_db_connection(conn)
+
+def update_menu_item(item_id: int, name: str = None, description: str = None, price: str = None, available: bool = None) -> str:
+    """
+    Update an existing menu item by ID. Only provided fields are updated.
+    """
+    conn = get_db_connection()
+    try:
+        updates = {}
+        if name is not None:
+            updates["name"] = name.strip()
+        if description is not None:
+            updates["description"] = description.strip()
+        if price is not None:
+            updates["price"] = price.strip()
+        if available is not None:
+            updates["available"] = available
+        if not updates:
+            return "هیچ فیلدی برای بروزرسانی ارسال نشده است."
+        with conn.cursor() as cur:
+            set_clause = ", ".join([f"{field} = %s" for field in updates])
+            values = list(updates.values()) + [item_id]
+            cur.execute(f"UPDATE menu SET {set_clause} WHERE id = %s;", values)
+            conn.commit()
+            if cur.rowcount == 0:
+                return f"آیتم با شناسه {item_id} یافت نشد."
+            return f"آیتم {item_id} با موفقیت بروزرسانی شد."
+    except Exception as e:
+        conn.rollback()
+        return f"خطا در بروزرسانی آیتم: {str(e)}"
+    finally:
+        release_db_connection(conn)
+
+def delete_menu_item(item_id: int) -> str:
+    """
+    Delete a menu item by ID.
+    """
+    conn = get_db_connection()
+    try:
+        with conn.cursor() as cur:
+            cur.execute("DELETE FROM menu WHERE id = %s;", (item_id,))
+            conn.commit()
+            if cur.rowcount == 0:
+                return f"آیتم با شناسه {item_id} یافت نشد."
+            return f"آیتم {item_id} با موفقیت حذف شد."
+    except Exception as e:
+        conn.rollback()
+        return f"خطا در حذف آیتم: {str(e)}"
+    finally:
+        release_db_connection(conn)
+
+# --- System Instructions ---
 CUSTOMER_SYSTEM_INSTRUCTION = """Your name is Nila, a classic, warm, and highly professional AI cafe assistant for "Cafe AI". You were developed by "Nastaran Data Algorithm".
 ... (keep existing)"""
-ADMIN_SYSTEM_INSTRUCTION = """Your name is Nila, the smart cafe administrator assistant for "Cafe AI". ... (keep existing)"""
+
+ADMIN_SYSTEM_INSTRUCTION = """Your name is Nila, the smart cafe administrator assistant for "Cafe AI". You manage the cafe menu in real time.
+You can view, add, update, delete menu items, and change availability using the following tools:
+- **get_menu_items()**: fetch the full menu with IDs, names, descriptions, prices, and availability.
+- **add_menu_item(name, description, price, available)**: add a new item (description and price optional, available is True by default).
+- **update_menu_item(item_id, name, description, price, available)**: update any fields of an existing item by its ID (only pass the fields you want to change).
+- **delete_menu_item(item_id)**: permanently remove an item.
+- **set_item_availability(item_name, available)**: quickly toggle availability of an item by its exact name.
+
+When the user asks to see the menu, use get_menu_items and show a nicely formatted list.
+When asked to add an item, ask for the necessary details and then call add_menu_item.
+For modifications, first get the menu to find the item ID, then use update_menu_item.
+For removals, use delete_menu_item with the correct ID.
+Always confirm every change clearly. Communicate primarily in Persian (Farsi), but you can understand English commands as well.
+Be polite, precise, and efficient."""
 
 # --- Routes ---
 @app.route('/')
@@ -299,7 +390,7 @@ def api_customer_chat_stream(table_number):
 
             response = chat.send_message(user_message)
 
-            # Extract text
+            # Extract text (ignore possible function calls in this simplified customer flow)
             if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
                 parts = response.candidates[0].content.parts
                 text_parts = [part.text for part in parts if hasattr(part, "text") and part.text]
@@ -431,7 +522,7 @@ def api_admin_chat():
 
         config = types.GenerateContentConfig(
             system_instruction=ADMIN_SYSTEM_INSTRUCTION,
-            tools=[get_full_menu, set_item_availability],
+            tools=[get_menu_items, add_menu_item, update_menu_item, delete_menu_item, set_item_availability],
             thinking_config=types.ThinkingConfig(thinking_budget=0)
         )
 
@@ -445,6 +536,48 @@ def api_admin_chat():
 
         response = chat.send_message(last_user_message)
 
+        # Handle function calls loop
+        max_iterations = 10  # safety net
+        while max_iterations > 0:
+            function_calls = []
+            if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
+                for part in response.candidates[0].content.parts:
+                    if hasattr(part, "function_call") and part.function_call:
+                        function_calls.append(part.function_call)
+            if not function_calls:
+                break
+
+            # Execute functions and build response
+            function_response_parts = []
+            for fc in function_calls:
+                func_name = fc.name
+                args = dict(fc.args) if hasattr(fc, "args") else {}
+                print(f"Admin function call: {func_name}({args})")
+                try:
+                    if func_name == "get_menu_items":
+                        result = get_menu_items()
+                    elif func_name == "add_menu_item":
+                        result = add_menu_item(**args)
+                    elif func_name == "update_menu_item":
+                        result = update_menu_item(**args)
+                    elif func_name == "delete_menu_item":
+                        result = delete_menu_item(**args)
+                    elif func_name == "set_item_availability":
+                        result = set_item_availability(**args)
+                    else:
+                        result = f"Unknown function: {func_name}"
+                except Exception as e:
+                    result = f"Error executing {func_name}: {str(e)}"
+                function_response_parts.append(
+                    types.Part.from_function_response(name=func_name, response={"result": result})
+                )
+            # Send function responses back to the model
+            response = chat.send_message(
+                types.Content(role="user", parts=function_response_parts)
+            )
+            max_iterations -= 1
+
+        # Extract final text
         admin_response_text = ""
         if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
             parts = response.candidates[0].content.parts
@@ -455,22 +588,20 @@ def api_admin_chat():
             try:
                 admin_response_text = response.text or ""
             except (ValueError, AttributeError):
-                admin_response_text = "تغییرات مورد نظر شما با موفقیت روی منو اعمال شد."
+                admin_response_text = "تغییرات مورد نظر شما با موفقیت اعمال شد."
 
         return jsonify({"success": True, "response": admin_response_text})
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# --- Admin Menu CRUD ---
+# --- Admin Menu CRUD (manual REST endpoints, kept for the admin UI table) ---
 @app.route('/api/admin/menu', methods=['GET'])
 def api_admin_get_menu():
-    """Get full menu for admin panel."""
     menu = load_menu()
     return jsonify(menu)
 
 @app.route('/api/admin/menu/item', methods=['POST'])
 def api_admin_add_menu_item():
-    """Add a new menu item manually."""
     data = request.json or {}
     name = data.get("name", "").strip()
     if not name:
@@ -497,12 +628,10 @@ def api_admin_add_menu_item():
 
 @app.route('/api/admin/menu/item/<int:item_id>', methods=['PUT'])
 def api_admin_update_menu_item(item_id):
-    """Update a menu item's fields."""
     data = request.json or {}
     conn = get_db_connection()
     try:
         with conn.cursor() as cur:
-            # Build dynamic SET clause
             allowed_fields = ["name", "description", "price", "available"]
             updates = []
             values = []
@@ -527,7 +656,6 @@ def api_admin_update_menu_item(item_id):
 
 @app.route('/api/admin/menu/item/<int:item_id>', methods=['DELETE'])
 def api_admin_delete_menu_item(item_id):
-    """Delete a menu item."""
     conn = get_db_connection()
     try:
         with conn.cursor() as cur: