Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import json
 import time
-from flask import Flask, request, Response, jsonify, render_template, redirect
+from flask import Flask, request, Response, jsonify, render_template, redirect, url_for
 from google import genai
 from google.genai import types
 from pydantic import BaseModel, Field
@@ -32,7 +32,7 @@ global_drafts = {}        # پیش‌نویس‌های موقت سفارش بر
 active_tasks = {}         # کنترل پردازش‌های فعال جهت متوقف کردن استریمینگ
 
 
-# --- تعریف کلاس‌های پایداری اسکیما با Pydantic (رفع قطعی خطای پارامتر) ---
+# --- تعریف کلاس‌های پایداری اسکیما با Pydantic ---
 
 class OrderItem(BaseModel):
     name: str = Field(description="نام دقیق و کامل آیتم از منوی فعال به زبان فارسی (مثال: کیک شکلاتی خیس)")
@@ -68,7 +68,6 @@ def save_orders(orders_data):
 
 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".
 Your primary goal is to help customers at their tables select drinks, food, and sweets from the menu, and prepare their orders.
-
 Rules you must strictly follow:
 1. Speak in Farsi (Persian) in a warm, polite, and welcoming tone. Use emojis suitable for a cozy cafe (☕️, 🍰, 🥐, 🌸, etc.).
 2. If anyone asks who you are or who created you, say: "من نیلا هستم، دستیار هوشمند کافه AI که توسط تیم الگوریتم داده نسترن توسعه پیدا کردم." 
@@ -81,7 +80,6 @@ Rules you must strictly follow:
 
 ADMIN_SYSTEM_INSTRUCTION = """Your name is Nila, the smart cafe administrator assistant for "Cafe AI". You were developed by "Nastaran Data Algorithm".
 Your role is to help the cafe staff manage their active menu. You can update item availabilities, edit menu details, and process commands.
-
 Rules you must strictly follow:
 1. Speak in Farsi (Persian) in a professional, clear, and polite tone.
 2. If anyone asks who you are or who created you, say: "من نیلا هستم، دستیار هوشمند مدیریت که توسط تیم الگوریتم داده نسترن توسعه پیدا کردم."
@@ -117,8 +115,27 @@ def prepare_order_draft(table_number: str, items: list[OrderItem]) -> str:
         table_number: The table number (e.g. '3')
         items: A list of ordered items, each containing a name and a quantity.
     """
-    # تبدیل دیتاهای شیء گرا Pydantic به دیکشنری ساده جهت نوشتن در فایل دیتابیس
-    items_list = [{"name": item.name, "quantity": item.quantity} for item in items]
+    items_list = []
+    # پردازش دفاعی برای پشتیبانی همزمان از کلاس‌های Pydantic و ساختارهای دیکشنری خام
+    for item in items:
+        if isinstance(item, dict):
+            items_list.append({
+                "name": item.get("name", ""),
+                "quantity": item.get("quantity", 1)
+            })
+        elif hasattr(item, "name"):
+            items_list.append({
+                "name": item.name,
+                "quantity": item.quantity
+            })
+        else:
+            try:
+                items_list.append({
+                    "name": getattr(item, "name", str(item)),
+                    "quantity": getattr(item, "quantity", 1)
+                })
+            except Exception:
+                pass
     
     global_drafts[table_number] = {
         "table": table_number,
@@ -163,11 +180,12 @@ def set_item_availability(item_name: str, available: bool) -> str:
         return f"آیتم '{item_name}' در منوی کافه یافت نشد."
 
 
-# --- روت‌های مسیریابی آدرس‌ها (Domain URL Mapping) ---
+# --- روت‌های مسیریابی آدرس‌ها ---
 
 @app.route('/')
-def home_redirect():
-    return redirect('/customer/1')
+def home():
+    # فرستادن مقدار None به صفحه کلاینت تا ابتدا فرم انتخاب شماره میز باز شود
+    return render_template("customer.html", table_number=None)
 
 @app.route('/customer/<table_number>')
 def customer_chat_page(table_number):
@@ -182,9 +200,10 @@ def admin_dashboard_page():
 
 @app.route('/api/customer/chat_stream/<table_number>', methods=['POST'])
 def api_customer_chat_stream(table_number):
-    api_key = os.environ.get("NILLA_CUSTOMER")
+    # پشتیبانی از کلید عمومی پیش‌فرض در صورت عدم تنظیم متغیرهای اختصاصی
+    api_key = os.environ.get("NILLA_CUSTOMER") or os.environ.get("GEMINI_API_KEY")
     if not api_key:
-        return jsonify({"error": "کلید NILLA_CUSTOMER در سرور یافت نشد."}), 500
+        return jsonify({"error": "کلید API در سرور یافت نشد. لطفاً NILLA_CUSTOMER یا GEMINI_API_KEY را بررسی کنید."}), 500
 
     data = request.json or {}
     user_message = data.get("message")
@@ -200,8 +219,9 @@ def api_customer_chat_stream(table_number):
 
     def generate_chunks():
         try:
+            # بازسازی ساختار تاریخچه به جز پیام جاری برای ارسال مستقیم به موتور چت استریمینگ
             contents = []
-            for msg in table_chat_histories[table_number]:
+            for msg in table_chat_histories[table_number][:-1]:
                 role = "user" if msg["role"] == "user" else "model"
                 contents.append(
                     types.Content(
@@ -215,18 +235,23 @@ def api_customer_chat_stream(table_number):
             
             dynamic_instruction = CUSTOMER_SYSTEM_INSTRUCTION + f"\nYou are currently serving table: {table_number}. Maintain context based on this table number."
 
+            # تنظیم بودجه تفکر روی صفر جهت دریافت سریع‌ترین پاسخ با حداقل تاخیر برای مشتری
             config = types.GenerateContentConfig(
                 system_instruction=dynamic_instruction,
                 tools=[get_menu, prepare_order_draft],
-                thinking_config=types.ThinkingConfig(thinking_level="MINIMAL")
+                thinking_config=types.ThinkingConfig(thinking_budget=0)
             )
 
-            # فراخوانی جریان استریم همگام
-            for chunk in client.models.generate_content_stream(
+            # استفاده از موتور چت SDK جهت مدیریت اتوماتیک و اجرای موازی ابزارها (Automatic Function Execution)
+            chat = client.chats.create(
                 model="gemini-3.1-flash-lite",
-                contents=contents,
+                history=contents,
                 config=config
-            ):
+            )
+
+            response_stream = chat.send_message_stream(user_message)
+
+            for chunk in response_stream:
                 if not active_tasks.get(table_number, True):
                     break
 
@@ -237,6 +262,7 @@ def api_customer_chat_stream(table_number):
             if full_generated_text:
                 table_chat_histories[table_number].append({"role": "model", "content": full_generated_text})
 
+            # بررسی و ارسال پیش‌نویس سفارش در صورتی که ابزار مربوطه در چرخه گفتگو اجرا شده باشد
             draft = global_drafts.get(table_number)
             if draft:
                 yield f"data: {json.dumps({'type': 'draft', 'items': draft['items']}, ensure_ascii=False)}\n\n"
@@ -279,6 +305,7 @@ def api_confirm_order():
     orders.append(new_order)
     save_orders(orders)
     
+    # آزاد کردن سشن و سوابق حافظه موقت
     table_chat_histories.pop(table_number, None)
     global_drafts.pop(table_number, None)
     active_tasks.pop(table_number, None)
@@ -317,17 +344,20 @@ def api_admin_complete_order():
 
 @app.route('/api/admin/chat', methods=['POST'])
 def api_admin_chat():
-    api_key = os.environ.get("NILLA_CAFE")
+    api_key = os.environ.get("NILLA_CAFE") or os.environ.get("GEMINI_API_KEY")
     if not api_key:
-        return jsonify({"error": "کلید NILLA_CAFE یافت نشد."}), 500
+        return jsonify({"error": "کلید API یافت نشد."}), 500
         
     data = request.json or {}
     messages = data.get("messages", [])
+    if not messages:
+        return jsonify({"error": "تاریخچه پیام‌ها خالی است."}), 400
     
     try:
         client = genai.Client(api_key=api_key)
         contents = []
-        for msg in messages:
+        # تبدیل ساختار پیام‌ها به فرمت Content شیءگرا، به استثنای پیام آخر
+        for msg in messages[:-1]:
             role = "user" if msg["role"] == "user" else "model"
             contents.append(
                 types.Content(
@@ -339,15 +369,20 @@ def api_admin_chat():
         config = types.GenerateContentConfig(
             system_instruction=ADMIN_SYSTEM_INSTRUCTION,
             tools=[get_full_menu, set_item_availability],
-            thinking_config=types.ThinkingConfig(thinking_level="MINIMAL")
+            thinking_config=types.ThinkingConfig(thinking_budget=0)
         )
         
-        response = client.models.generate_content(
+        last_user_message = messages[-1]["content"]
+        
+        # استفاده از مدل گفتگو چت جهت اعمال خودکار تغییرات روی منو
+        chat = client.chats.create(
             model="gemini-3.1-flash-lite",
-            contents=contents,
+            history=contents,
             config=config
         )
         
+        response = chat.send_message(last_user_message)
+        
         return jsonify({
             "success": True,
             "response": response.text
@@ -358,9 +393,9 @@ def api_admin_chat():
 
 @app.route('/api/admin/extract_menu', methods=['POST'])
 def api_extract_menu():
-    api_key = os.environ.get("NILLA_CAFE")
+    api_key = os.environ.get("NILLA_CAFE") or os.environ.get("GEMINI_API_KEY")
     if not api_key:
-        return jsonify({"error": "کلید NILLA_CAFE یافت نشد."}), 500
+        return jsonify({"error": "کلید API یافت نشد."}), 500
         
     if "file" not in request.files:
         return jsonify({"error": "فایل ارسال نشده"}), 400