Upload 3 files

event_recommendation.py

@@ -0,0 +1,312 @@
+"""
+Event Recommendation Scenario Handler
+Recommends events based on user's vibe/mood with RAG integration
+"""
+from typing import Dict, Any
+from .base_handler import BaseScenarioHandler
+
+
+class EventRecommendationHandler(BaseScenarioHandler):
+    """
+    Handle event recommendation flow
+    
+    Steps:
+    1. Ask for vibe/mood (Chill, Sôi động, Hài, Workshop)
+    2. Search events matching vibe → RAG
+    3. Show event list, ask which to see details
+    4. Ask what info needed (price, lineup, location, time)
+    5-8. Show specific info → RAG
+    9. Ask if want to save event to email
+    10. Collect email + send summary
+    11-12. End scenario
+    """
+    
+    def start(self, initial_data: Dict = None) -> Dict[str, Any]:
+        """Start event recommendation flow"""
+        return {
+            "message": "Hello! 👋 Bạn muốn tìm sự kiện theo vibe gì nè? Chill – Sôi động – Hài – Workshop?",
+            "new_state": {
+                "active_scenario": "event_recommendation",
+                "scenario_step": 1,
+                "scenario_data": initial_data or {}
+            }
+        }
+    
+    def next_step(self, current_step: int, user_input: str, scenario_data: Dict) -> Dict[str, Any]:
+        """Process user input and advance scenario"""
+        
+        # Get expected input type for this step
+        expected_type = self._get_expected_type(current_step)
+        
+        # Check for unexpected input (off-topic questions)
+        unexpected = self.handle_unexpected_input(user_input, expected_type, current_step)
+        if unexpected:
+            return unexpected
+        
+        # ===== STEP 1: Collect interest tag =====
+        if current_step == 1:
+            scenario_data['interest_tag'] = user_input
+            
+            return {
+                "message": f"Mình hiểu rồi! Để mình tìm sự kiện hợp vibe **{user_input}** nha",
+                "new_state": {
+                    "active_scenario": "event_recommendation",
+                    "scenario_step": 2,
+                    "scenario_data": scenario_data
+                },
+                "scenario_active": True
+            }
+        
+        # ===== STEP 2: Execute RAG search for events =====
+                "new_state": {
+                    "active_scenario": "event_recommendation",
+                    "scenario_step": 3,
+                    "scenario_data": scenario_data
+                },
+                "scenario_active": True,
+                "loading_message": "⏳ Bạn đợi tôi tìm 1 chút nhé..."
+            }
+        
+        # ===== STEP 3: User picks event =====
+        elif current_step == 3:
+            scenario_data['event_name'] = user_input
+            
+            return {
+                "message": "Bạn cần xem: giá – line-up – địa điểm – hay thời gian của sự kiện?",
+                "new_state": {
+                    "active_scenario": "event_recommendation",
+                    "scenario_step": 4,
+                    "scenario_data": scenario_data
+                },
+                "scenario_active": True
+            }
+        
+        # ===== STEP 4: Branch based on info choice =====
+        elif current_step == 4:
+            choice = self._detect_choice(user_input)
+            event_name = scenario_data.get('event_name', 'sự kiện này')
+            
+            # Build RAG query based on choice
+            query_map = {
+                'price': f"giá vé {event_name}",
+                'lineup': f"lineup nghệ sĩ {event_name}",
+                'location': f"địa điểm tổ chức {event_name}",
+                'time': f"thời gian lịch diễn {event_name}"
+            }
+            
+            query = query_map.get(choice, query_map['price'])
+            print(f"🔍 RAG Search: {query}")
+            
+            results = self._search_rag(query)
+            formatted_info = self._format_rag_results(results)
+            
+            # Build response message
+            message_map = {
+                'price': f"Giá vé event {event_name} nè:\n{formatted_info}",
+                'lineup': f"Lineup / nghệ sĩ của event {event_name} là:\n{formatted_info}",
+                'location': f"Địa điểm tổ chức event {event_name}:\n{formatted_info}",
+                'time': f"Thời gian / lịch diễn của event {event_name}:\n{formatted_info}"
+            }
+            
+            return {
+                "message": message_map.get(choice, message_map['price']),
+                "new_state": {
+                    "active_scenario": "event_recommendation",
+                    "scenario_step": 9,  # Skip to email step
+                    "scenario_data": scenario_data
+                },
+                "scenario_active": True,
+                "loading_message": "⏳ Bạn đợi tôi tìm 1 chút nhé..."
+            }
+        
+        # ===== STEP 9: Ask if want to save event to email =====
+        elif current_step == 9:
+            choice = self._detect_yes_no(user_input)
+            
+            if choice == 'yes':
+                return {
+                    "message": "Cho mình xin email để gửi bản tóm tắt event kèm link mua vé?",
+                    "new_state": {
+                        "active_scenario": "event_recommendation",
+                        "scenario_step": 10,
+                        "scenario_data": scenario_data
+                    },
+                    "scenario_active": True
+                }
+            else:
+                return {
+                    "message": "Okie, bạn cần event theo vibe khác không nè? 😄",
+                    "new_state": None,
+                    "scenario_active": False,
+                    "end_scenario": True
+                }
+        
+        # ===== STEP 10: Collect email and send summary =====
+        elif current_step == 10:
+            email = user_input.strip()
+            
+            if not self._validate_email(email):
+                return {
+                    "message": "Email này có vẻ không đúng định dạng. Bạn nhập lại giúp mình nhé? (Ví dụ: name@example.com)",
+                    "new_state": None,  # Stay at same step
+                    "scenario_active": True
+                }
+            
+            # Save lead
+            scenario_data['email'] = email
+            
+            try:
+                self.lead_storage.save_lead(
+                    event_name=scenario_data.get('event_name', 'Unknown Event'),
+                    email=email,
+                    interests={
+                        "vibe": scenario_data.get('interest_tag'),
+                        "wants_event_summary": True
+                    },
+                    session_id=scenario_data.get('session_id')
+                )
+                print(f"📧 Lead saved: {email}")
+            except Exception as e:
+                print(f"⚠️ Error saving lead: {e}")
+            
+            return {
+                "message": "Đã gửi email cho bạn nha! ✨",
+                "new_state": None,
+                "scenario_active": False,
+                "end_scenario": True,
+                "action": "send_event_summary_email"
+            }
+        
+        # Fallback - unknown step
+        return {
+            "message": "Xin lỗi, có lỗi xảy ra. Bạn muốn bắt đầu lại không?",
+            "new_state": None,
+            "scenario_active": False,
+            "end_scenario": True
+        }
+    
+    def _get_expected_type(self, step: int) -> str:
+        """Get expected input type for each step"""
+        type_map = {
+            1: 'interest_tag',
+            2: None,  # Auto-advance after RAG
+            3: 'event_name',
+            4: 'choice',
+            9: 'choice',
+            10: 'email'
+        }
+        return type_map.get(step, 'text')
+    
+    def _format_event_list(self, results: list) -> str:
+        """Format event search results as numbered list"""
+        print(f"🔍 DEBUG: RAG returned {len(results)} results")
+        
+        if not results or len(results) == 0:
+            return "Hiện tại chưa có event phù hợp 😢\nBạn thử vibe khác nhé!"
+        
+        # Debug: Print first result to see structure
+        if len(results) > 0:
+            print(f"🔍 DEBUG: First result metadata: {results[0].get('metadata', {})}")
+        
+        events = []
+        for i, r in enumerate(results[:3], 1):
+            metadata = r.get('metadata', {})
+            
+            # Extract event info from metadata
+            # Your Qdrant has: {'texts': [...], 'id_use': '...'}
+            event_id = metadata.get('id_use', metadata.get('original_id'))
+            texts = metadata.get('texts', [])
+            text = texts[0] if texts and len(texts) > 0 else metadata.get('text', '')
+            
+            # Use first 60 chars of text as event name
+            name = text[:60].strip() + "..." if len(text) > 60 else text.strip()
+            
+            print(f"🔍 DEBUG: Event {i}: id={event_id}, name={name[:50]}")
+            
+            # Simple format for now (can enhance with API call later)
+            event_str = f"{i}. **{name}**"
+            
+            # Store event_id for later API call if needed
+            if event_id:
+                event_str += f" (ID: {event_id[:8]}...)"
+            
+            events.append(event_str)
+        
+        return "\n".join(events)
+    
+    async def _format_event_list_with_api(self, results: list) -> str:
+        """
+        Format event search results by calling API for full details
+        """
+        print(f"🔍 DEBUG: RAG returned {len(results)} results")
+        
+        if not results or len(results) == 0:
+            return "Hiện tại chưa có event phù hợp 😢\nBạn thử vibe khác nhé!"
+        
+        # Import event service
+        from event_service import EventService
+        event_service = EventService()
+        
+        events = []
+        for i, r in enumerate(results[:3], 1):
+            metadata = r.get('metadata', {})
+            event_id = metadata.get('id_use', metadata.get('original_id'))
+            
+            print(f"🔍 DEBUG: Fetching event {i} with ID: {event_id}")
+            
+            # Try to get full event data from API
+            event_data = None
+            if event_id:
+                event_data = await event_service.get_event_by_id(event_id)
+            
+            if event_data:
+                # Use API data
+                name = event_data.get("eventName", "Sự kiện")
+                start = event_data.get("eventStartTime", "")
+                date_str = start[:10] if start else "TBA"
+                location = event_data.get("eventAddress", "")
+                
+                event_str = f"{i}. **{name}**"
+                if date_str != "TBA":
+                    event_str += f" ({date_str})"
+                if location:
+                    event_str += f" - {location}"
+                
+                print(f"✅ Event {i}: {name} ({date_str})")
+            else:
+                # Fallback to text from Qdrant
+                texts = metadata.get('texts', [])
+                text = texts[0] if texts and len(texts) > 0 else ""
+                name = text[:60].strip() + "..." if len(text) > 60 else text.strip()
+                
+                event_str = f"{i}. **{name}**"
+                print(f"⚠️ Event {i}: Fallback to text (API failed)")
+            
+            events.append(event_str)
+        
+        await event_service.close()
+        return "\n".join(events)
+    
+    def _detect_choice(self, user_input: str) -> str:
+        """Detect what info user wants to see"""
+        input_lower = user_input.lower()
+        
+        if any(k in input_lower for k in ['giá', 'price', 'vé', 'ticket', 'bao nhiêu']):
+            return 'price'
+        elif any(k in input_lower for k in ['lineup', 'line-up', 'nghệ sĩ', 'artist', 'performer']):
+            return 'lineup'
+        elif any(k in input_lower for k in ['địa điểm', 'location', 'ở đâu', 'where', 'chỗ']):
+            return 'location'
+        elif any(k in input_lower for k in ['thời gian', 'time', 'khi nào', 'when', 'lịch', 'date']):
+            return 'time'
+        else:
+            return 'price'  # Default
+    
+    def _detect_yes_no(self, user_input: str) -> str:
+        """Detect yes/no response"""
+        input_lower = user_input.lower()
+        
+        if any(k in input_lower for k in ['có', 'yes', 'ok', 'được', 'ừ', 'oke']):
+            return 'yes'
+        else:
+            return 'no'

event_service.py

@@ -0,0 +1,120 @@
+"""
+Event Service - Get event details from MongoDB ID or eventCode
+"""
+import httpx
+from typing import Dict, Any, Optional
+
+
+class EventService:
+    """Service to fetch event details from external API"""
+    
+    def __init__(self, base_url: str = "https://hoalacrent.io.vn/api/v0"):
+        self.base_url = base_url
+        self.client = httpx.AsyncClient(timeout=10.0)
+    
+    async def get_event_by_code(self, event_code: str) -> Optional[Dict[str, Any]]:
+        """
+        Get event details by eventCode
+        
+        Args:
+            event_code: Event code (e.g., "EVENT001")
+        
+        Returns:
+            Event data or None if not found
+        """
+        try:
+            url = f"{self.base_url}/event/get-event-by-event-code"
+            response = await self.client.get(url, params={"eventCode": event_code})
+            response.raise_for_status()
+            
+            data = response.json()
+            if data.get("success") and data.get("data"):
+                event = data["data"]
+                return {
+                    "eventCode": event.get("eventCode"),
+                    "eventName": event.get("eventName"),
+                    "eventAddress": event.get("eventAddress"),
+                    "eventStartTime": event.get("eventStartTime"),
+                    "eventEndTime": event.get("eventEndTime"),
+                    "maxParticipants": event.get("maxParticipants"),
+                    "eventOrganizer": event.get("eventOrganizer"),
+                    "eventDescription": event.get("eventDescription")
+                }
+            return None
+        except Exception as e:
+            print(f"⚠️ Error fetching event by code: {e}")
+            return None
+    
+    async def get_event_by_id(self, event_id: str) -> Optional[Dict[str, Any]]:
+        """
+        Get event details by MongoDB _id
+        
+        IMPORTANT: Backend needs to implement this endpoint:
+        GET /api/v0/event/get-event-by-id?id={mongodb_id}
+        
+        Args:
+            event_id: MongoDB _id (e.g., "6900ae38eb03f29702c7fd1d")
+        
+        Returns:
+            Event data or None if not found
+        """
+        try:
+            # Option 1: If backend has this endpoint
+            url = f"{self.base_url}/event/get-event-by-id"
+            response = await self.client.get(url, params={"id": event_id})
+            response.raise_for_status()
+            
+            data = response.json()
+            if data.get("success") and data.get("data"):
+                event = data["data"]
+                return {
+                    "eventCode": event.get("eventCode"),
+                    "eventName": event.get("eventName"),
+                    "eventAddress": event.get("eventAddress"),
+                    "eventStartTime": event.get("eventStartTime"),
+                    "eventEndTime": event.get("eventEndTime"),
+                    "maxParticipants": event.get("maxParticipants"),
+                    "eventOrganizer": event.get("eventOrganizer"),
+                    "eventDescription": event.get("eventDescription")
+                }
+            return None
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 404:
+                print(f"⚠️ Event not found: {event_id}")
+            else:
+                print(f"⚠️ Error fetching event by ID: {e}")
+            return None
+        except Exception as e:
+            print(f"⚠️ Error fetching event by ID: {e}")
+            return None
+    
+    def format_event_for_display(self, event: Dict[str, Any]) -> str:
+        """
+        Format event data for chat display
+        
+        Args:
+            event: Event data dict
+        
+        Returns:
+            Formatted string
+        """
+        name = event.get("eventName", "Sự kiện")
+        
+        # Format date
+        start = event.get("eventStartTime", "")
+        date_str = start[:10] if start else "TBA"  # Extract date part
+        
+        # Format location
+        location = event.get("eventAddress", "")
+        
+        result = f"**{name}**"
+        if date_str != "TBA":
+            result += f" ({date_str})"
+        if location:
+            result += f" - {location}"
+        
+        return result
+    
+    async def close(self):
+        """Close HTTP client"""
+        await self.client.aclose()