Update app.py

app.py

@@ -115,6 +115,8 @@ 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.
     """
+    # اطمینان از تبدیل شماره میز به نوع رشته جهت سازگاری با کلیدهای دیکشنری
+    table_number_str = str(table_number)
     items_list = []
     # پردازش دفاعی برای پشتیبانی همزمان از کلاس‌های Pydantic و ساختارهای دیکشنری خام
     for item in items:
@@ -137,12 +139,12 @@ def prepare_order_draft(table_number: str, items: list[OrderItem]) -> str:
             except Exception:
                 pass
     
-    global_drafts[table_number] = {
-        "table": table_number,
+    global_drafts[table_number_str] = {
+        "table": table_number_str,
         "items": items_list,
         "timestamp": time.time()
     }
-    return f"من پیش‌نویس سفارش شما رو برای میز {table_number} آماده کردم."
+    return f"من پیش‌نویس سفارش شما رو برای میز {table_number_str} آماده کردم."
 
 def get_full_menu() -> str:
     """
@@ -180,6 +182,28 @@ def set_item_availability(item_name: str, available: bool) -> str:
         return f"آیتم '{item_name}' در منوی کافه یافت نشد."
 
 
+# --- توابع کمکی استخراج متن ---
+
+def get_chunk_text(chunk) -> str:
+    """
+    استخراج ایمن متن از تکه‌های استریم جهت جلوگیری از خطای ValueError روی تکه‌های حاوی فراخوانی ابزار.
+    """
+    try:
+        # ابتدا بخش کاندیداها بررسی می‌شود تا در صورت وجود متن، مستقیماً استخراج شود
+        if chunk.candidates and chunk.candidates[0].content and chunk.candidates[0].content.parts:
+            parts = chunk.candidates[0].content.parts
+            text_parts = [part.text for part in parts if hasattr(part, "text") and part.text]
+            if text_parts:
+                return "".join(text_parts)
+    except Exception:
+        pass
+        
+    try:
+        return chunk.text or ""
+    except (ValueError, AttributeError):
+        return ""
+
+
 # --- روت‌های مسیریابی آدرس‌ها ---
 
 @app.route('/')
@@ -255,9 +279,12 @@ def api_customer_chat_stream(table_number):
                 if not active_tasks.get(table_number, True):
                     break
 
-                if chunk.text:
-                    full_generated_text += chunk.text
-                    yield f"data: {json.dumps({'type': 'text', 'content': chunk.text}, ensure_ascii=False)}\n\n"
+                # استفاده از تابع استخراج دفاعی به جای ارجاع مستقیم به chunk.text
+                chunk_text = get_chunk_text(chunk)
+
+                if chunk_text:
+                    full_generated_text += chunk_text
+                    yield f"data: {json.dumps({'type': 'text', 'content': chunk_text}, ensure_ascii=False)}\n\n"
 
             if full_generated_text:
                 table_chat_histories[table_number].append({"role": "model", "content": full_generated_text})
@@ -383,9 +410,26 @@ def api_admin_chat():
         
         response = chat.send_message(last_user_message)
         
+        # استخراج متن به صورت دفاعی در صورت اجرای ابزارها در گام‌های میانی
+        admin_response_text = ""
+        try:
+            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:
+                    admin_response_text = "".join(text_parts)
+        except Exception:
+            pass
+            
+        if not admin_response_text:
+            try:
+                admin_response_text = response.text or ""
+            except (ValueError, AttributeError):
+                admin_response_text = "تغییرات مورد نظر شما با موفقیت روی منو اعمال شد."
+        
         return jsonify({
             "success": True,
-            "response": response.text
+            "response": admin_response_text
         })
         
     except Exception as e: