Update scenario_handlers/event_recommendation.py

scenario_handlers/event_recommendation.py

@@ -63,18 +63,18 @@ class EventRecommendationHandler(BaseScenarioHandler):
             query = f"sự kiện phù hợp với {scenario_data.get('interest_tag', 'mọi người')}"
             print(f"🔍 RAG Search: {query}")
             
-            results = self._search_rag(query, limit=3)
+            results = self._search_rag(query, limit=5)  # Get up to 5
             formatted_events = self._format_event_list(results)
             
             # Save results to scenario data
             scenario_data['rag_results'] = formatted_events
             scenario_data['available_events'] = [
-                r['metadata'].get('event_name', f'Event {i+1}') 
-                for i, r in enumerate(results[:3])
+                r['metadata'].get('id_use', metadata.get('original_id', f'event_{i+1}')) 
+                for i, r in enumerate(results)
             ]
             
             return {
-                "message": f"Đây là 2–3 event hợp với bạn nè:\n{formatted_events}\n\nBạn có muốn xem chi tiết event nào không?",
+                "message": f"Đây là các event hợp với bạn nè:\n{formatted_events}\n\nBạn có muốn xem chi tiết event nào không?",
                 "new_state": {
                     "active_scenario": "event_recommendation",
                     "scenario_step": 3,
@@ -215,34 +215,83 @@ class EventRecommendationHandler(BaseScenarioHandler):
         return type_map.get(step, 'text')
     
     def _format_event_list(self, results: list) -> str:
-        """Format event search results as numbered list"""
+        """Format event search results - IMPROVED to show actual event count"""
         print(f"🔍 DEBUG: RAG returned {len(results)} results")
         
         if not results or len(results) == 0:
+            print("⚠️ DEBUG: No results found")
             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', {})}")
+        print(f"🔍 DEBUG: First result metadata: {results[0].get('metadata', {})}")
         
         events = []
-        for i, r in enumerate(results[:3], 1):
+        # FIXED: Use actual length instead of hardcoded [:3]
+        num_events = min(len(results), 5)  # Show max 5 events
+        print(f"🔍 DEBUG: Formatting {num_events} events")
+        
+        for i, r in enumerate(results[:num_events], 1):
             metadata = r.get('metadata', {})
-            name = metadata.get('event_name', f'Event {i}')
-            date = metadata.get('date', 'TBA')
-            location = metadata.get('location', '')
             
-            print(f"🔍 DEBUG: Event {i}: name={name}, date={date}, location={location}")
+            # Extract event info from Qdrant metadata
+            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', '')
+            
+            print(f"🔍 DEBUG Event {i}:")
+            print(f"  - ID: {event_id}")
+            print(f"  - Text length: {len(text)} chars")
+            print(f"  - Text preview: {text[:100]}")
+            
+            # Extract event name from text
+            # Try to find event name pattern in text
+            name = self._extract_event_name_from_text(text)
+            
+            if not name:
+                # Fallback to shortened text
+                name = text[:60].strip()
+                if len(text) > 60:
+                    name += "..."
+            
+            print(f"  - Extracted name: {name}")
             
+            # Format event string
             event_str = f"{i}. **{name}**"
-            if date != 'TBA':
-                event_str += f" ({date})"
-            if location:
-                event_str += f" - {location}"
             
             events.append(event_str)
         
-        return "\n".join(events)
+        result = "\n".join(events)
+        print(f"✅ DEBUG: Formatted {len(events)} events successfully")
+        return result
+    
+    def _extract_event_name_from_text(self, text: str) -> str:
+        """
+        Try to extract event name from description text
+        
+        Patterns to match:
+        - "Sự kiện văn hóa hài kịch" → "Văn hóa hài kịch"
+        - "Stand-up Comedy Night - ..." → "Stand-up Comedy Night"
+        """
+        if not text:
+            return ""
+        
+        # Remove "Sự kiện" prefix if exists
+        clean_text = text
+        if clean_text.startswith("Sự kiện "):
+            clean_text = clean_text[8:].strip()
+        
+        # Take first sentence or first 80 chars
+        sentences = clean_text.split('.')
+        if sentences and len(sentences) > 0:
+            first_sentence = sentences[0].strip()
+            if len(first_sentence) > 80:
+                return first_sentence[:77] + "..."
+            return first_sentence
+        
+        # Fallback
+        if len(clean_text) > 80:
+            return clean_text[:77] + "..."
+        return clean_text
     
     def _detect_choice(self, user_input: str) -> str:
         """Detect what info user wants to see"""