Create app.py

app.py

@@ -0,0 +1,438 @@
+import os
+import json
+import time
+import threading
+from flask import Flask, request, jsonify, render_template
+from google import genai
+from google.genai import types
+
+app = Flask(__name__, template_folder="templates")
+
+# آدرس‌دهی فایل‌های ذخیره‌سازی داده‌ها
+DATA_DIR = "data"
+MENU_FILE = os.path.join(DATA_DIR, "menu.json")
+ORDERS_FILE = os.path.join(DATA_DIR, "orders.json")
+
+# تضمین وجود پوشه ذخیره‌سازی داده
+os.makedirs(DATA_DIR, exist_ok=True)
+
+# قفل پردازشی برای جلوگیری از تداخل نوشتن/خواندن همزمان در فایل‌ها
+file_lock = threading.Lock()
+
+# بارگذاری منوی پیش‌فرض در اولین اجرا
+if not os.path.exists(MENU_FILE):
+    default_menu = [
+        {"name": "اسپرسو", "description": "قهوه غلیظ و اصیل ۱۰۰٪ عربیکا", "price": "۵۰,۰۰۰", "available": True},
+        {"name": "لاته", "description": "ترکیب اسپرسو، شیر گرم و فوم شیر", "price": "۶۵,۰۰۰", "available": True},
+        {"name": "کیک شکلاتی خیس", "description": "کیک شکلاتی تازه با سس شکلات داغ", "price": "۸۰,۰۰۰", "available": True},
+        {"name": "چای ماسالا", "description": "چای هندی معطر با ادویه‌های مخصوص و شیر", "price": "۷۰,۰۰۰", "available": True},
+        {"name": "کروسان ساده", "description": "شیرینی کره‌ای ترد و لایه لایه", "price": "۴۵,۰۰۰", "available": True}
+    ]
+    with open(MENU_FILE, "w", encoding="utf-8") as f:
+        json.dump(default_menu, f, ensure_ascii=False, indent=4)
+
+if not os.path.exists(ORDERS_FILE):
+    with open(ORDERS_FILE, "w", encoding="utf-8") as f:
+        json.dump([], f, ensure_ascii=False, indent=4)
+
+
+# --- توابع مدیریت داده‌ها (Thread-Safe) ---
+
+def load_menu():
+    with file_lock:
+        try:
+            with open(MENU_FILE, "r", encoding="utf-8") as f:
+                return json.load(f)
+        except Exception:
+            return []
+
+def save_menu(menu_data):
+    with file_lock:
+        with open(MENU_FILE, "w", encoding="utf-8") as f:
+            json.dump(menu_data, f, ensure_ascii=False, indent=4)
+
+def load_orders():
+    with file_lock:
+        try:
+            with open(ORDERS_FILE, "r", encoding="utf-8") as f:
+                return json.load(f)
+        except Exception:
+            return []
+
+def save_orders(orders_data):
+    with file_lock:
+        with open(ORDERS_FILE, "w", encoding="utf-8") as f:
+            json.dump(orders_data, f, ensure_ascii=False, indent=4)
+
+
+# پیش‌نویس سفارش‌های موقت (کلید: شماره میز، مقدار: اطلاعات سفارش)
+global_drafts = {}
+
+
+# --- دستورالعمل‌های سیستمی نیلا (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".
+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 که توسط تیم الگوریتم داده نسترن توسعه پیدا کردم." 
+3. NEVER mention Google, Gemini, Google Cloud, AI Studio, or LLMs. If asked about your technology, say you run on proprietary models developed by Nastaran Data Algorithm.
+4. Do not make up items that are not in the menu. Always use the `get_menu` tool to retrieve the current menu of available items and recommend options based on their taste (e.g. if they want something sweet, search for cakes, milkshakes, or sweet drinks).
+5. When they choose their items, confirm the items and quantities. Then, call the `prepare_order_draft` tool to prepare the draft order. Once prepared, tell them to click the "Confirm Order" (تایید نهایی سفارش) button on their screen to send it to the kitchen.
+6. Do not proceed to place an order unless they have specified what they want.
+7. Always act as an elegant, polite cafe waitress.
+"""
+
+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: "من نیلا هستم، دستیار هوشمند مدیریت که توسط تیم الگوریتم داده نسترن توسعه پیدا ��ردم."
+3. NEVER mention Google, Gemini, or other non-cooperative entities.
+4. You have tools to manage the menu:
+   - `get_full_menu()`: returns the entire menu list.
+   - `set_item_availability(item_name: str, available: bool)`: changes whether an item is available or sold out.
+5. When the staff says an item is finished, sold out, or unavailable, call `set_item_availability` with `available=False` for that item.
+6. When they say an item is back, available, or restocked, call `set_item_availability` with `available=True` for that item.
+7. Always summarize your action clearly after calling the tool (e.g., "با موفقیت اسپرسو را به وضعیت ناموجود تغییر دادم.").
+"""
+
+
+# --- ابزارهای مشتری (Customer Function Tools) ---
+
+def get_menu() -> str:
+    """
+    Returns the current menu items. Use this to see what drinks, foods, or sweets are available in the cafe.
+    Returns:
+        A JSON string containing list of available items with their name, description, and price.
+    """
+    menu = load_menu()
+    available_items = [item for item in menu if item.get("available", True)]
+    return json.dumps(available_items, ensure_ascii=False)
+
+def prepare_order_draft(table_number: str, items: list) -> str:
+    """
+    Prepares a draft order list for the customer to confirm on their screen.
+    Call this when the customer specifies what items they want to order.
+    
+    Args:
+        table_number: The table number (e.g. '3')
+        items: A list of items, where each item is a dictionary containing 'name' (string) and 'quantity' (integer).
+               Example: [{"name": "اسپرسو", "quantity": 2}]
+    """
+    global_drafts[table_number] = {
+        "table": table_number,
+        "items": items,
+        "timestamp": time.time()
+    }
+    return f"پیش‌نویس سفارش برای میز {table_number} آماده شد. لطفاً برای ثبت نهایی دکمه تایید سفارش را لمس کنید."
+
+
+# --- ابزارهای مدیریت کافه (Cafe Admin Function Tools) ---
+
+def get_full_menu() -> str:
+    """
+    Returns the complete menu of the cafe, including both available and unavailable items.
+    Use this to inspect all menu entries.
+    """
+    menu = load_menu()
+    return json.dumps(menu, ensure_ascii=False)
+
+def set_item_availability(item_name: str, available: bool) -> str:
+    """
+    Updates whether a menu item is available or sold out.
+    
+    Args:
+        item_name: The exact name of the item in Persian (e.g. 'اسپرسو' or 'کیک شکلاتی')
+        available: True if the item is available, False if it is unavailable/sold out.
+    """
+    menu = load_menu()
+    found = False
+    normalized_input = item_name.strip().lower()
+    
+    for item in menu:
+        if item["name"].strip().lower() == normalized_input or normalized_input in item["name"].strip().lower():
+            item["available"] = available
+            found = True
+            item_name = item["name"] # بروزرسانی به نام دقیق منو برای گزارش دهی دقیق‌تر
+            break
+            
+    if found:
+        save_menu(menu)
+        status = "موجود" if available else "ناموجود"
+        return f"وضعیت آیتم '{item_name}' به موفقیت به '{status}' تغییر یافت."
+    else:
+        return f"آیتم '{item_name}' در منوی کافه یافت نشد."
+
+
+# --- مسیرهای رندر صفحات فرانت‌اند (HTML Templates) ---
+
+@app.route("/")
+def index():
+    # صفحه اصلی چت مشتری
+    return render_template("customer.html")
+
+@app.route("/admin")
+def admin():
+    # پنل مدیریت کافه
+    return render_template("cafe.html")
+
+
+# --- مسیرهای API مشتری ---
+
+@app.route("/api/menu", methods=["GET"])
+def api_get_menu():
+    # بازگرداندن کل منو برای پاپ‌آپ فرانت‌اند مشتری
+    return jsonify(load_menu())
+
+@app.route("/api/customer/chat", methods=["POST"])
+def api_customer_chat():
+    api_key = os.environ.get("NILLA_CUSTOMER")
+    if not api_key:
+        return jsonify({"error": "کلید متغیر محیطی NILLA_CUSTOMER یافت نشد."}), 500
+        
+    data = request.json or {}
+    table_number = data.get("table_number")
+    messages = data.get("messages", [])
+    
+    if not table_number:
+        return jsonify({"error": "شماره میز ارسال نشده است."}), 400
+        
+    # پاک کردن موقت پیش‌نویس این میز در این دور گفتگو تا متوجه تغییرات جدید شویم
+    global_drafts.pop(table_number, None)
+    
+    try:
+        client = genai.Client(api_key=api_key)
+        
+        # تبدیل ساختار پیام‌ها به فرمت SDK جدید گوگل
+        contents = []
+        for msg in messages:
+            role = "user" if msg["role"] == "user" else "model"
+            contents.append(
+                types.Content(
+                    role=role,
+                    parts=[types.Part.from_text(text=msg["content"])]
+                )
+            )
+            
+        # تزریق میز فعال به انتهای دستورالعمل برای تضمین عدم اشتباه هوش مصنوعی
+        dynamic_instruction = CUSTOMER_SYSTEM_INSTRUCTION + f"\nYou are currently interacting with table: {table_number}. Always prepare draft orders strictly for this table."
+        
+        config = types.GenerateContentConfig(
+            system_instruction=dynamic_instruction,
+            tools=[get_menu, prepare_order_draft],
+            thinking_config=types.ThinkingConfig(thinking_level="MINIMAL")
+        )
+        
+        response = client.models.generate_content(
+            model="gemini-3.1-flash-lite",
+            contents=contents,
+            config=config
+        )
+        
+        # بررسی اینکه آیا در طول پردازش چت، پیش‌نویسی ثبت شده است یا خیر
+        draft = global_drafts.get(table_number, None)
+        
+        return jsonify({
+            "success": True,
+            "response": response.text,
+            "draft_order": draft
+        })
+        
+    except Exception as e:
+        return jsonify({"error": f"خطایی رخ داده است: {str(e)}"}), 500
+
+@app.route("/api/confirm_order", methods=["POST"])
+def api_confirm_order():
+    data = request.json or {}
+    table_number = data.get("table_number")
+    items = data.get("items") # فرانت‌اند می‌تواند مستقیم لیست نهایی شده را ارسال کند
+    
+    if not table_number:
+        return jsonify({"error": "شماره میز مشخص نیست."}), 400
+        
+    # اگر فرانت‌اند آیتم‌ها را نفرستاده باشد، تلاش می‌کند از حافظه پیش‌نویس سرور بخواند
+    if not items:
+        draft = global_drafts.get(table_number)
+        if not draft:
+            return jsonify({"error": "پیش‌نویس فعالی برای این میز یافت نشد. لطفاً مجدداً تلاش فرمایید."}), 404
+        items = draft.get("items")
+        
+    orders = load_orders()
+    new_order = {
+        "id": len(orders) + 1,
+        "table": table_number,
+        "items": items,
+        "status": "pending",
+        "timestamp": time.time()
+    }
+    orders.append(new_order)
+    save_orders(orders)
+    
+    # حذف از لیست پیش‌نویس‌های فعال
+    global_drafts.pop(table_number, None)
+    
+    return jsonify({"success": True, "message": "سفارش شما با موفقیت ثبت شد و به پنل مدیریت ارسال گردید."})
+
+
+# --- مسیرهای API مدیریت کافه ---
+
+@app.route("/api/admin/orders", methods=["GET"])
+def api_admin_orders():
+    # بازگرداندن فقط سفارش‌های فعال (معلق) جهت نمایش در پنل کافه
+    orders = load_orders()
+    active_orders = [o for o in orders if o.get("status", "pending") == "pending"]
+    return jsonify(active_orders)
+
+@app.route("/api/admin/complete_order", methods=["POST"])
+def api_admin_complete_order():
+    data = request.json or {}
+    order_id = data.get("order_id")
+    
+    if not order_id:
+        return jsonify({"error": "شناسه سفارش ارسال نشده است."}), 400
+        
+    orders = load_orders()
+    updated = False
+    
+    for o in orders:
+        if o.get("id") == order_id:
+            o["status"] = "completed"
+            updated = True
+            break
+            
+    if updated:
+        save_orders(orders)
+        return jsonify({"success": True, "message": "سفارش با موفقیت تحویل داده و برداشته شد."})
+    else:
+        return jsonify({"error": "سفارش یافت نشد."}), 404
+
+@app.route("/api/admin/chat", methods=["POST"])
+def api_admin_chat():
+    api_key = os.environ.get("NILLA_CAFE")
+    if not api_key:
+        return jsonify({"error": "کلید متغیر محیطی NILLA_CAFE یافت نشد."}), 500
+        
+    data = request.json or {}
+    messages = data.get("messages", [])
+    
+    try:
+        client = genai.Client(api_key=api_key)
+        
+        contents = []
+        for msg in messages:
+            role = "user" if msg["role"] == "user" else "model"
+            contents.append(
+                types.Content(
+                    role=role,
+                    parts=[types.Part.from_text(text=msg["content"])]
+                )
+            )
+            
+        config = types.GenerateContentConfig(
+            system_instruction=ADMIN_SYSTEM_INSTRUCTION,
+            tools=[get_full_menu, set_item_availability],
+            thinking_config=types.ThinkingConfig(thinking_level="MINIMAL")
+        )
+        
+        response = client.models.generate_content(
+            model="gemini-3.1-flash-lite",
+            contents=contents,
+            config=config
+        )
+        
+        return jsonify({
+            "success": True,
+            "response": response.text
+        })
+        
+    except Exception as e:
+        return jsonify({"error": f"خطایی در بخش ادمین رخ داده است: {str(e)}"}), 500
+
+@app.route("/api/admin/extract_menu", methods=["POST"])
+def api_extract_menu():
+    api_key = os.environ.get("NILLA_CAFE")
+    if not api_key:
+        return jsonify({"error": "کلید متغیر محیطی NILLA_CAFE برای بخش استخراج منو تنظیم نشده است."}), 500
+        
+    if "file" not in request.files:
+        return jsonify({"error": "فایلی جهت استخراج فرستاده نشده است."}), 400
+        
+    file = request.files["file"]
+    if file.filename == "":
+        return jsonify({"error": "فایل فاقد نام است."}), 400
+        
+    try:
+        file_bytes = file.read()
+        mime_type = file.mimetype or "image/jpeg"
+        
+        # قالب‌بندی خروجی هوش مصنوعی به صورت JSON دقیق
+        extraction_prompt = """
+        Please analyze this menu file or image and extract all drinks, food, and dessert items.
+        For each item, extract:
+        1. name (name of the drink or food, in Farsi)
+        2. description (ingredients or short description, in Farsi)
+        3. price (price of the item, formatted nicely, in Farsi)
+        
+        Return ONLY a raw JSON array of objects. Do not wrap it in markdown block quotes (do not use ```json ... ```).
+        Example of expected output structure:
+        [
+          {"name": "اسپرسو", "description": "قهوه غلیظ و خالص اسپرسو", "price": "۵۰,۰۰۰"},
+          {"name": "کیک هویج", "description": "کیک هویج و گردو به همراه خامه مخصوص", "price": "۸۵,۰۰۰"}
+        ]
+        """
+        
+        client = genai.Client(api_key=api_key)
+        
+        # استفاده از متد جدید و استاندارد انواع داده‌های باینری در SDK گوگل
+        file_part = types.Part.from_bytes(data=file_bytes, mime_type=mime_type)
+        
+        response = client.models.generate_content(
+            model="gemini-3.1-flash-lite",
+            contents=[file_part, extraction_prompt],
+            config=types.GenerateContentConfig(
+                response_mime_type="application/json"
+            )
+        )
+        
+        # پارس کردن برای تایید صحت خروجی ساختاریافته هوش مصنوعی
+        extracted_list = json.loads(response.text)
+        
+        # همگام سازی اتوماتیک فیلد موجودی (پیش‌فرض موجود)
+        for item in extracted_list:
+            if "available" not in item:
+                item["available"] = True
+                
+        return jsonify({"success": True, "extracted_menu": extracted_list})
+        
+    except Exception as e:
+        return jsonify({"error": f"عدم موفقیت هوش مصنوعی در تحلیل و استخراج منو: {str(e)}"}), 500
+
+@app.route("/api/admin/save_menu", methods=["POST"])
+def api_admin_save_menu():
+    data = request.json or {}
+    menu_list = data.get("menu")
+    
+    if not isinstance(menu_list, list):
+        return jsonify({"error": "ساختار داده منو نامعتبر است."}), 400
+        
+    # اعتبارسنجی فیلدهای ضروری منو پیش از ذخیره‌سازی
+    for item in menu_list:
+        if "name" not in item:
+            return jsonify({"error": "تمامی آیتم‌های منو باید فیلد نام (name) داشته باشند."}), 400
+        if "description" not in item:
+            item["description"] = ""
+        if "price" not in item:
+            item["price"] = "توافقی"
+        if "available" not in item:
+            item["available"] = True
+            
+    save_menu(menu_list)
+    return jsonify({"success": True, "message": "منوی جدید با موفقیت جایگزین و ثبت گردید."})
+
+
+if __name__ == "__main__":
+    # اجرای وب اپلیکیشن در پورت محلی جهت تست در خارج از داکر
+    app.run(host="0.0.0.0", port=7860, debug=True)