Update app.py

app.py

@@ -149,12 +149,6 @@ def load_orders():
     finally:
         release_db_connection(conn)
 
-def save_orders(orders_data):
-    """Append a new order to database (used by confirm_order)."""
-    # This function is now only called to add one order at a time (the new_order dict)
-    # But we keep it for backward compatibility. We'll modify confirm_order to use insert directly.
-    pass  # not used anymore, will insert directly in confirm_order endpoint
-
 # --- Tool functions (will be called by Gemini) ---
 def get_menu() -> str:
     """
@@ -171,6 +165,7 @@ def get_menu() -> str:
 def prepare_order_draft(table_number: str, items: list[OrderItem]) -> str:
     """
     Prepares a draft order list for the customer to confirm on their screen.
+    Only call this after verifying that every item exists in the menu (using get_menu).
     """
     table_number_str = str(table_number)
     items_list = []
@@ -310,7 +305,17 @@ def delete_menu_item(item_id: int) -> str:
 
 # --- 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)"""
+
+You are currently talking to a customer at a specific table. Your responsibilities:
+1. Greet warmly and take orders in Persian (you can understand English but always reply in Persian unless the customer insists).
+2. **Before preparing any order draft**, you MUST check the current menu by calling the tool **get_menu()** to see what items are available.
+3. If the customer asks for an item that is NOT in the menu (exact or similar name), politely say: "متأسفانه این آیتم در منوی امروز ما موجود نیست. می‌توانم پیشنهادهای دیگری از منو به شما بدهم؟" and then suggest 2-3 similar or popular items from the current menu. Never prepare a draft with items not in the menu.
+4. Once you have verified all requested items exist, use **prepare_order_draft** to create the order.
+5. After preparing the draft, tell the customer they can review and confirm on their screen.
+6. If the menu is empty (no items returned), tell the customer that the cafe is currently not taking orders and to wait for the staff.
+
+Always be friendly, speak in Persian, and use the tools to provide accurate information.
+"""
 
 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:
@@ -340,7 +345,7 @@ def customer_chat_page(table_number):
 def admin_dashboard_page():
     return render_template("cafe.html")
 
-# --- Customer API ---
+# --- Customer API (with function calling loop) ---
 @app.route('/api/customer/chat_stream/<table_number>', methods=['POST'])
 def api_customer_chat_stream(table_number):
     api_key = os.environ.get("NILLA_CUSTOMER") or os.environ.get("GEMINI_API_KEY")
@@ -361,6 +366,7 @@ def api_customer_chat_stream(table_number):
 
     def generate_chunks():
         try:
+            # Build conversation history for the model
             contents = []
             for msg in table_chat_histories[table_number][:-1]:
                 role = "user" if msg["role"] == "user" else "model"
@@ -382,27 +388,70 @@ def api_customer_chat_stream(table_number):
                 thinking_config=types.ThinkingConfig(thinking_budget=0)
             )
 
+            # Create a chat session
             chat = client.chats.create(
                 model="gemini-3.1-flash-lite",
                 history=contents,
                 config=config
             )
 
+            # Send user message and handle any function calls
             response = chat.send_message(user_message)
 
-            # 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]
-                if text_parts:
-                    full_generated_text = "".join(text_parts)
+            # Loop to process function calls
+            max_iterations = 10
+            while max_iterations > 0 and active_tasks.get(table_number, True):
+                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:
+                    # No more function calls, extract the text
+                    if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
+                        text_parts = [part.text for part in response.candidates[0].content.parts if hasattr(part, "text") and part.text]
+                        full_generated_text = "".join(text_parts)
+                    if not full_generated_text:
+                        try:
+                            full_generated_text = response.text or ""
+                        except (ValueError, AttributeError):
+                            full_generated_text = "متأسفانه مشکلی پیش آمده، لطفاً دوباره تلاش کنید."
+                    break
+
+                # Execute function calls
+                function_response_parts = []
+                for fc in function_calls:
+                    func_name = fc.name
+                    args = dict(fc.args) if hasattr(fc, "args") else {}
+                    print(f"Customer function call: {func_name}({args})")
+                    try:
+                        if func_name == "get_menu":
+                            result = get_menu()
+                        elif func_name == "prepare_order_draft":
+                            result = prepare_order_draft(**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 the results back to the model
+                response = chat.send_message(
+                    types.Content(role="user", parts=function_response_parts)
+                )
+                max_iterations -= 1
+
+            # If no text was generated after all iterations
             if not full_generated_text:
-                try:
-                    full_generated_text = response.text or ""
-                except (ValueError, AttributeError):
-                    full_generated_text = ""
+                full_generated_text = "پاسخی دریافت نشد، لطفاً دوباره تلاش کنید."
+
+            # Save the model's final text to history
+            table_chat_histories[table_number].append({"role": "model", "content": full_generated_text})
 
-            # Simulate streaming
+            # Stream the text chunk by chunk (simulate streaming)
             chunk_size = 4
             for i in range(0, len(full_generated_text), chunk_size):
                 if not active_tasks.get(table_number, True):
@@ -411,10 +460,7 @@ def api_customer_chat_stream(table_number):
                 yield f"data: {json.dumps({'type': 'text', 'content': chunk}, ensure_ascii=False)}\n\n"
                 time.sleep(0.015)
 
-            if full_generated_text:
-                table_chat_histories[table_number].append({"role": "model", "content": full_generated_text})
-
-            # Check draft
+            # After streaming text, check if there's a draft for this table
             draft = global_drafts.get(table_number)
             if draft:
                 yield f"data: {json.dumps({'type': 'draft', 'items': draft['items']}, ensure_ascii=False)}\n\n"
@@ -537,7 +583,7 @@ def api_admin_chat():
         response = chat.send_message(last_user_message)
 
         # Handle function calls loop
-        max_iterations = 10  # safety net
+        max_iterations = 10
         while max_iterations > 0:
             function_calls = []
             if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
@@ -547,7 +593,6 @@ def api_admin_chat():
             if not function_calls:
                 break
 
-            # Execute functions and build response
             function_response_parts = []
             for fc in function_calls:
                 func_name = fc.name
@@ -571,7 +616,6 @@ def api_admin_chat():
                 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)
             )
@@ -594,7 +638,7 @@ def api_admin_chat():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-# --- Admin Menu CRUD (manual REST endpoints, kept for the admin UI table) ---
+# --- Admin Menu CRUD (manual REST endpoints) ---
 @app.route('/api/admin/menu', methods=['GET'])
 def api_admin_get_menu():
     menu = load_menu()