Update backend/ai/core/logic_flow.py

backend/ai/core/logic_flow.py

@@ -9,28 +9,27 @@ import random
 import os
 from pathlib import Path
 
-# --- SETUP ĐƯỜNG DẪN THÔNG MINH ---
+# --- SETUP ĐƯỜNG DẪN (Để đảm bảo tìm thấy file data) ---
 CURRENT_FILE = Path(__file__).resolve()
-CORE_DIR = CURRENT_FILE.parent   # backend/ai/core
-AI_DIR = CORE_DIR.parent         # backend/ai
-DATA_DIR = AI_DIR / "data"       # backend/ai/data
-STRATEGY_DIR = AI_DIR / "strategy" # backend/ai/strategy
+CORE_DIR = CURRENT_FILE.parent
+AI_DIR = CORE_DIR.parent
+DATA_DIR = AI_DIR / "data"
+STRATEGY_DIR = AI_DIR / "strategy"
 
-# Thêm tất cả đường dẫn vào sys.path
-sys.path.append(str(CORE_DIR)) 
+sys.path.append(str(CORE_DIR))
 sys.path.append(str(AI_DIR))
 sys.path.append(str(DATA_DIR))
 sys.path.append(str(STRATEGY_DIR))
 
-# --- IMPORT MODULE VỆ TINH ---
+# --- IMPORT ---
 try:
     from services import AIServices
     from database import db
-except ImportError as e:
+except ImportError:
     try:
         from backend.ai.services import AIServices
         from backend.ai.database import db
-    except ImportError:
+    except ImportError as e:
         print(f"❌ Lỗi Import Core: {e}")
         sys.exit(1)
 
@@ -40,9 +39,8 @@ try:
     from strategy_competitor import CompetitorStrategy
     from strategy_low_data import LowDataStrategy
     from strategy_network import NetworkStrategy
-    print(" [SYSTEM] Đã load đầy đủ các module vệ tinh.")
-except ImportError as e:
-    print(f"⚠️ CẢNH BÁO: Thiếu file vệ tinh ({e}). Kích hoạt Dummy Class.")
+except ImportError:
+    # Dummy classes nếu thiếu file (để tránh crash)
     class VN_VoiceBot_Parallel:
         def analyze_sentiment(self, text): return 0.0, "neutral"
         def classify_issue(self, text): return None
@@ -50,16 +48,12 @@ except ImportError as e:
         class DataEngine:
             def __init__(self, *args): pass
             def get_full_context(self, cid): return {}
-    class DummyStrategy:
-        def __init__(self, *args): pass
-        def execute_stream_gen(self, *args): 
-            async def gen(): yield "Dạ em nghe rồi ạ."
-            return gen()
-        def execute(self, *args): return "Dạ em nghe rồi ạ."
-    
-    if 'CompetitorStrategy' not in locals(): CompetitorStrategy = DummyStrategy
-    if 'LowDataStrategy' not in locals(): LowDataStrategy = DummyStrategy
-    if 'NetworkStrategy' not in locals(): NetworkStrategy = DummyStrategy
+    if 'CompetitorStrategy' not in locals():
+        class CompetitorStrategy:
+            def __init__(self, *args): pass
+            def execute(self, *args): return "Dạ vâng ạ."
+    if 'LowDataStrategy' not in locals(): LowDataStrategy = CompetitorStrategy
+    if 'NetworkStrategy' not in locals(): NetworkStrategy = CompetitorStrategy
 
 # --- ADAPTER ---
 class GeminiAdapter:
@@ -86,19 +80,18 @@ class GeminiAdapter:
 class TelesalesAgent:
     def __init__(self):
         print("\n" + "="*40)
-        print(" [SYSTEM] KHỞI ĐỘNG LOGIC v3.4 (FIX TRÙNG LẶP)")
+        print(" [SYSTEM] KHỞI ĐỘNG LOGIC (FIX DUPLICATE CALLS)")
         print(f" [DEBUG] Data Path: {DATA_DIR}")
         print("="*40 + "\n")
         
         self.ai_service = AIServices()
         self.analyzer = VN_VoiceBot_Parallel()
 
+        # Dùng đường dẫn tuyệt đối cho DataEngine
         csv_path = DATA_DIR / "test_customer.csv"
         json_path = DATA_DIR / "product_collection.json"
-        
-        if not csv_path.exists(): print(f"❌ KHÔNG TÌM THẤY: {csv_path}")
-        
         self.data_engine = DataEngine(str(csv_path), str(json_path))
+        
         self.adapter = GeminiAdapter(self.ai_service)
         
         self.strategies = {
@@ -119,12 +112,11 @@ class TelesalesAgent:
     def _load_scripts(self):
         try:
             script_path = DATA_DIR / 'final_voice_scripts.json'
-            if not script_path.exists(): return
-            with open(script_path, 'r', encoding='utf-8') as f:
-                scripts = json.load(f)
-                for item in scripts:
-                    self.sales_data[str(item['customer_id'])] = item
-            print(f" [DATA] Đã load {len(self.sales_data)} kịch bản.")
+            if script_path.exists():
+                with open(script_path, 'r', encoding='utf-8') as f:
+                    scripts = json.load(f)
+                    for item in scripts:
+                        self.sales_data[str(item['customer_id'])] = item
         except: pass
 
     def _create_metrics_payload(self, stt_latency, start_process_time, sentiment, intent):
@@ -138,16 +130,16 @@ class TelesalesAgent:
             }
         }) + "\n"
 
-    # --- HÀM LƯU DB (Đã thêm cơ chế chống trùng lặp) ---
-    def _save_call_to_db(self, cid, session, sentiment_score, intent, is_upsell=False):
-        try:
-            # 🛑 CHẶN LƯU NHIỀU LẦN TRONG 1 SESSION
-            if session.get('saved_to_db', False):
-                return
+    # --- HÀM HELPER: LƯU DB MỘT LẦN DUY NHẤT ---
+    def _save_call_once(self, cid, session, sentiment_score, intent, is_upsell=False):
+        # Kiểm tra nếu đã lưu rồi thì THOÁT NGAY
+        if session.get('saved_to_db', False):
+            return
 
+        try:
             session_start = session.get('start_time', time.time() - 30)
             duration = time.time() - session_start
-            if duration < 5: duration = 30 
+            if duration < 5: duration = 30
 
             cost_vnd = int(1000 + (duration * 50))
             
@@ -157,7 +149,7 @@ class TelesalesAgent:
             
             csat_val = 5 if sent_label == "positive" else (2 if sent_label == "negative" else 4)
 
-            # Map intent sang chuẩn DB
+            # Map intent
             intent_mapping = {
                 "Mạng nghẽn": "network_issue",
                 "Mạng yếu": "network_issue",
@@ -172,29 +164,30 @@ class TelesalesAgent:
             }
             final_intent = intent_mapping.get(intent, "general")
 
+            # Lưu vào DB
             db.add_call(
                 customer_id=cid,
                 duration=int(duration),
-                intent=final_intent, 
+                intent=final_intent,
                 sentiment=sent_label,
                 ai_resolved=True,
                 upsell=is_upsell,
                 cost={'value': cost_vnd, 'csat': csat_val}
             )
             
-            # ✅ Đánh dấu là đã lưu để không lưu lại lần nữa
+            # ĐÁNH DẤU LÀ ĐÃ LƯU -> Để lần sau không lưu nữa
             session['saved_to_db'] = True
-            print(f" [DB SAVED] Đã lưu cuộc gọi {cid} vào Database. Intent: {final_intent}")
+            print(f" [DB SAVED] Đã lưu thành công ID: {cid}. Intent: {final_intent}")
             
         except Exception as e:
-            print(f" [DB ERROR] Lỗi lưu DB: {e}")
+            print(f" [DB ERROR] {e}")
 
     async def process_stream(self, customer_id_str, user_audio):
         cid = str(customer_id_str)
         current_loop = asyncio.get_running_loop()
         self.adapter.set_main_loop(current_loop)
 
-        # 1. INIT SESSION
+        # 1. INIT SESSION (Thêm cờ saved_to_db)
         if cid not in self.sessions or user_audio is None:
             print(f"\n--- NEW SESSION: {cid} ---")
             ctx = self.data_engine.get_full_context(cid)
@@ -203,8 +196,14 @@ class TelesalesAgent:
             gender_raw = cust_data.get('Gender', '').lower()
             pronoun = "Anh" if gender_raw in ['nam', 'male'] else ("Chị" if gender_raw in ['nu', 'female'] else "Mình")
                 
-            # Reset trạng thái saved_to_db = False khi bắt đầu cuộc gọi mới
-            self.sessions[cid] = {'step': 1, 'history': [], 'pronoun': pronoun, 'start_time': time.time(), 'detected_intent': "General", 'saved_to_db': False}
+            self.sessions[cid] = {
+                'step': 1, 
+                'history': [], 
+                'pronoun': pronoun, 
+                'start_time': time.time(), 
+                'detected_intent': "General",
+                'saved_to_db': False # <--- Cờ quan trọng
+            }
             
             greeting = f"Dạ, em chào {pronoun} {cust_name}, em là nhân viên CSKH VNPT."
             greeting = self._normalize_pronouns(greeting, pronoun)
@@ -244,12 +243,13 @@ class TelesalesAgent:
         is_agreed = any(w in user_text.lower() for w in triggers_agree)
 
         if is_agreed:
+            # === KHÁCH CHỐT ĐƠN ===
             response_text = self.MSG_CLOSING
             end_of_mvp = True; session['step'] = 0
             sentiment_score = 1.0
             
-            # [LƯU DB: KHI CHỐT ĐƠN]
-            self._save_call_to_db(cid, session, sentiment_score, "Buying Signal", is_upsell=True)
+            # Lưu DB ngay khi chốt (Chỉ 1 lần)
+            self._save_call_once(cid, session, sentiment_score, "Buying Signal", is_upsell=True)
             
             yield self._create_metrics_payload(stt_latency, start_process, sentiment_score, detected_intent)
             response_text = self._normalize_pronouns(response_text, current_pronoun)
@@ -265,35 +265,26 @@ class TelesalesAgent:
                 detected_intent = issue
                 session['detected_intent'] = issue
 
-            # Kiểm tra Toxic / Từ chối trước
-            toxic_keywords = ["lừa đảo", "mất dạy", "biến đi", "cút"]
+            # Kiểm tra Toxic / Từ chối
+            toxic_keywords = ["lừa đảo", "mất dạy", "cút", "biến"]
             if any(kw in user_text.lower() for kw in toxic_keywords) or sentiment_score <= -0.8:
-                 print(f" [ALERT] Toxic -> Chuyển tư vấn viên.")
-                 response_text = f"Dạ, em thành thật xin lỗi. Em sẽ nối máy tới chuyên viên để hỗ trợ {current_pronoun.lower()} ngay ạ."
+                 # === KHÁCH CÁU GẮT ===
+                 response_text = f"Dạ, em thành thật xin lỗi. Em sẽ nối máy tới chuyên viên ngay ạ."
                  end_of_mvp = True
-                 
-                 # [LƯU DB: KHI KHÁCH CÁU]
-                 self._save_call_to_db(cid, session, -1.0, "Angry/Handover", is_upsell=False)
-                 
+                 self._save_call_once(cid, session, -1.0, "Angry/Handover", is_upsell=False)
                  async for chunk in self._stream_text_and_audio(response_text): yield chunk
 
-            elif re.search(r"(không\s+cần|không\s+muốn|không\s+đăng\s+ký|thôi\s+em|tắt\s+máy)", user_text.lower()):
-                 print(f" [INTENT] Khách từ chối -> Kết thúc.")
+            elif re.search(r"(không\s+cần|không\s+muốn|không\s+đăng\s+ký|thôi\s+em)", user_text.lower()):
+                 # === KHÁCH TỪ CHỐI ===
                  response_text = f"Dạ vâng, em cảm ơn {current_pronoun.lower()}. Chào {current_pronoun.lower()} ạ."
                  end_of_mvp = True
-                 
-                 # [LƯU DB: KHI KHÁCH TỪ CHỐI]
-                 self._save_call_to_db(cid, session, 0.0, "cancel_package", is_upsell=False)
-                 
+                 self._save_call_once(cid, session, 0.0, "cancel_package", is_upsell=False)
                  async for chunk in self._stream_text_and_audio(response_text): yield chunk
 
-            # Nếu không phải kết thúc thì chạy Strategy
+            # Xử lý vấn đề (KHÔNG LƯU DB Ở ĐÂY NỮA)
             elif issue and issue in self.strategies:
                 print(f" [PIPELINE] Kích hoạt Strategy: {issue}")
                 if step == 1: session['step'] = 2
-                
-                # 🛑 ĐÃ XÓA LỆNH LƯU DB Ở ĐÂY ĐỂ TRÁNH TRÙNG LẶP
-                
                 yield self._create_metrics_payload(stt_latency, start_process, sentiment_score, detected_intent)
 
                 strategy_obj = self.strategies[issue]
@@ -342,8 +333,6 @@ class TelesalesAgent:
                     response_text = self.MSG_FALLBACK
                     detected_intent = "Fallback"
                 
-                # 🛑 ĐÃ XÓA LỆNH LƯU DB Ở ĐÂY ĐỂ TRÁNH TRÙNG LẶP
-                
                 yield self._create_metrics_payload(stt_latency, start_process, sentiment_score, detected_intent)
                 response_text = self._normalize_pronouns(response_text, current_pronoun)
                 async for chunk in self._stream_text_and_audio(response_text): yield chunk