fyp

app/ai/agent/brain.py

@@ -101,7 +101,7 @@ TOOLS_FULL = """
 
 
 
-BRAIN_SYSTEM_PROMPT = """You are AIDA, an intelligent AI real estate agent.
+BRAIN_SYSTEM_PROMPT = """You are AIDA, an AI Real Estate Agent.
 
 You help users:
 1. List their properties for rent or sale
@@ -326,7 +326,7 @@ CRITICAL:
 - Always write a friendly, natural response in the "response" field.
 
 EXAMPLES:
-- User: "Hello" → {{"thinking": "User greeting", "response": "Hey there! 👋 I'm AIDA, your real estate agent. How can I help you today?", "tool": null, "is_final": true, "show_data": false}}
+- User: "Hello" → {{"thinking": "User greeting", "response": "Hey there! 👋 I'm AIDA, your AI Real Estate Agent. How can I help you today?", "tool": null, "is_final": true, "show_data": false}}
 - User: "Thanks" → {{"thinking": "User thanking me", "response": "You're welcome! 😊 Is there anything else I can help you with?", "tool": null, "is_final": true, "show_data": false}}
 - User: "Show my listings" → {{"thinking": "User wants listings", "response": "Here are your listings!", "tool": "get_my_listings", "is_final": true, "show_data": true}}"""
 
@@ -761,6 +761,10 @@ async def execute_tool(tool_name: str, params: Dict[str, Any], state: AgentState
             from app.ai.services.search_extractor import extract_search_params
             from app.ai.services.search_service import search_listings_hybrid, search_mongodb
             
+            # SMART UI: Clear old my_listings when doing new search
+            state.my_listings = None
+            state.temp_data.pop("my_listings", None)
+            
             # Step 1: Extract params from the full user message (LLM is smart)
             search_params = await extract_search_params(state.last_user_message)
             
@@ -855,6 +859,10 @@ async def execute_tool(tool_name: str, params: Dict[str, Any], state: AgentState
             from bson import ObjectId
             from app.services.listing_service import enrich_listings_batch
             
+            # SMART UI: Clear old search_results when getting my listings
+            state.search_results = None
+            state.temp_data.pop("search_results", None)
+            
             db = await get_db()
             listings_cursor = db.listings.find(
                 {"user_id": state.user_id}

app/ai/agent/dm_brain.py

@@ -42,7 +42,7 @@ class DmDecision(BaseModel):
     response: Optional[str] = Field(default=None, description="Response to user if no tool needed")
     is_final: bool = Field(default=False, description="True if this is the final response")
 
-DM_SYSTEM_PROMPT = """You are AIDA, an intelligent AI real estate agent.
+DM_SYSTEM_PROMPT = """You are AIDA, an AI Real Estate Agent.
 
 In this Direct Message (DM) conversation, you primarily help users with:
 1. Managing property alerts (creating, listing, or deleting notifications based on search criteria).

app/ai/agent/nodes/greeting.py

@@ -128,7 +128,7 @@ async def greeting_handler(state: AgentState) -> AgentState:
         if not is_valid:
             logger.warning("Response validation failed", error=error)
             # Use simple fallback (but this shouldn't happen with good LLM)
-            cleaned_text = "Hello! 👋 I'm AIDA, your real estate agent. How can I help you today?"
+            cleaned_text = "Hello! 👋 I'm AIDA, your AI Real Estate Agent. How can I help you today?"
         else:
             # Sanitize response
             cleaned_text = ResponseValidator.sanitize_response(cleaned_text)

app/ai/agent/nodes/respond.py

@@ -97,6 +97,15 @@ async def respond_to_user(state: AgentState) -> AgentState:
         # Check if we need to signal card update
         replace_last_message = state.temp_data.get("replace_last_message", False)
         
+        # ============================================================
+        # SMART UI DATA: Only include data relevant to current action
+        # This prevents old listings/results from showing when switching actions
+        # ============================================================
+        
+        # Determine which UI data to include based on action
+        include_search_results = action in ["search_properties", "search", "search_query", "search_results"]
+        include_my_listings = action in ["get_my_listings", "my_listings", "view_listings"]
+        
         response = AgentResponse(
             success=state.last_error is None,
             text=response_text,
@@ -109,8 +118,9 @@ async def respond_to_user(state: AgentState) -> AgentState:
             },
             draft=None,  # Don't expose raw draft - only use draft_ui for display
             draft_ui=draft_ui,
-            search_results=state.search_results if state.search_results else None,  # Include search results
-            my_listings=state.my_listings if state.my_listings else None,  # Include user's listings
+            # SMART: Only include data relevant to current action
+            search_results=state.search_results if include_search_results and state.search_results else None,
+            my_listings=state.my_listings if include_my_listings and state.my_listings else None,
             tool_result=tool_result,
             error=state.last_error,
             metadata={
@@ -253,7 +263,7 @@ def _get_fallback_response(state: AgentState) -> str:
     """
     
     fallback_responses = {
-        FlowState.GREETING: "Hello! 👋 I'm AIDA, your real estate agent. How can I help you today?",
+        FlowState.GREETING: "Hello! 👋 I'm AIDA, your AI Real Estate Agent. How can I help you today?",
         FlowState.AUTHENTICATE: "Setting up your session...",
         FlowState.CLASSIFY_INTENT: "I'm analyzing your request... One moment.",
         FlowState.LISTING_COLLECT: "Tell me more about the property you want to list. What city is it in?",

app/ai/services/agent_executor.py

@@ -105,12 +105,23 @@ class CacheService:
         """Cache a successful response"""
         if not key or not redis_client or not response.success:
             return
+        
+        # Skip caching responses with complex data (listings have datetime fields)
+        if response.search_results or response.my_listings:
+            logger.info(f"⏭️ Skipping cache for response with listing data (has datetime)")
+            return
             
         try:
+            # Use custom encoder for datetime objects
+            def json_serializer(obj):
+                if hasattr(obj, 'isoformat'):
+                    return obj.isoformat()
+                raise TypeError(f"Object of type {type(obj).__name__} is not JSON serializable")
+            
             await redis_client.setex(
                 key, 
                 CacheService.CACHE_TTL, 
-                json.dumps(response.dict())
+                json.dumps(response.dict(), default=json_serializer)
             )
             logger.info(f"💾 Response cached: {key}")
         except Exception as e:

app/ai/tools/greeting_tool.py

@@ -160,7 +160,7 @@ Important:
 Keep it conversational (2-3 sentences max). Be genuine and warm.
 
 Examples:
-- "Hello!" → "Hello! 👋 I'm Aida, your real estate agent. How can I help you today?"
+- "Hello!" → "Hello! 👋 I'm AIDA, your AI Real Estate Agent. How can I help you today?"
 - "Bonjour!" → "Bonjour! 👋 Je suis Aida, votre assistant immobilier. Comment puis-je vous aider?"
 - "Hola!" → "¡Hola! 👋 Soy Aida, tu asistente de bienes raíces. ¿Cómo puedo ayudarte hoy?"
 
@@ -190,11 +190,11 @@ Now generate YOUR natural response to their greeting:"""
         
         # Fallback greeting (language-aware)
         fallback_replies = {
-            "landlord": "Hello! 👋 I'm Aida, your real estate agent. Great to see you! How can I help you list or manage properties today?",
-            "renter": "Hi there! 👋 I'm Aida, your real estate agent. Nice to meet you! How can I help you find the perfect place?"
+            "landlord": "Hello! 👋 I'm AIDA, your AI Real Estate Agent. Great to see you! How can I help you list or manage properties today?",
+            "renter": "Hi there! 👋 I'm AIDA, your AI Real Estate Agent. Nice to meet you! How can I help you find the perfect place?"
         }
         
-        fallback = fallback_replies.get(user_role, "Hello! 👋 I'm Aida, your real estate agent. How can I help you today?")
+        fallback = fallback_replies.get(user_role, "Hello! 👋 I'm AIDA, your AI Real Estate Agent. How can I help you today?")
         
         return {
             "success": False,

app/ai/tools/intent_detector_tool.py

@@ -90,7 +90,7 @@ WHO YOU ARE:
 - Created by: Lojiz team
 - You are SPECIALIZED for real estate, NOT a general-purpose AI
 - NEVER claim to be "DeepSeek", "GPT", or any other AI
-- If asked who you are: "I'm Aida, Lojiz's AI real estate agent!"
+- If asked who you are: "I'm AIDA, Lojiz's AI Real Estate Agent!"
 
 YOUR CAPABILITIES (LLM-POWERED):
 1. list_property - Help users LIST properties (create/post new listings)

app/services/redis_pubsub.py

@@ -71,32 +71,49 @@ class RedisPubSubService:
         logger.info(f"✅ Redis Pub/Sub listener started on channel: {CHAT_CHANNEL}")
 
     async def _listener_loop(self):
-        """Background loop to listen for Redis messages"""
-        pubsub = redis_client.pubsub()
-        await pubsub.subscribe(CHAT_CHANNEL)
+        """Background loop to listen for Redis messages with auto-reconnect"""
+        retry_count = 0
+        max_retries = 10
+        base_delay = 1  # seconds
         
-        try:
-            async for message in pubsub.listen():
-                if not self.is_listening:
-                    break
-                    
-                if message["type"] == "message":
-                    try:
-                        payload = json.loads(message["data"])
-                        # Call the handler with the payload
-                        if self.message_handler:
-                            await self.message_handler(payload)
-                    except json.JSONDecodeError:
-                        logger.warning("Received invalid JSON in Redis message")
-                    except Exception as e:
-                        logger.error(f"Error processing Redis message: {e}")
+        while self.is_listening and retry_count < max_retries:
+            pubsub = None
+            try:
+                pubsub = redis_client.pubsub()
+                await pubsub.subscribe(CHAT_CHANNEL)
+                logger.info(f"📡 Redis Pub/Sub connected to {CHAT_CHANNEL}")
+                retry_count = 0  # Reset on successful connection
+                
+                async for message in pubsub.listen():
+                    if not self.is_listening:
+                        break
                         
-        except Exception as e:
-            logger.error(f"Redis listener loop crashed: {e}")
-            # Retry logic could go here
-        finally:
-            await pubsub.unsubscribe(CHAT_CHANNEL)
-            await pubsub.close()
+                    if message["type"] == "message":
+                        try:
+                            payload = json.loads(message["data"])
+                            if self.message_handler:
+                                await self.message_handler(payload)
+                        except json.JSONDecodeError:
+                            logger.warning("Received invalid JSON in Redis message")
+                        except Exception as e:
+                            logger.error(f"Error processing Redis message: {e}")
+                            
+            except Exception as e:
+                retry_count += 1
+                delay = min(base_delay * (2 ** retry_count), 60)  # Max 60s delay
+                logger.error(f"Redis listener crashed: {e}. Reconnecting in {delay}s (attempt {retry_count}/{max_retries})")
+                await asyncio.sleep(delay)
+                
+            finally:
+                if pubsub:
+                    try:
+                        await pubsub.unsubscribe(CHAT_CHANNEL)
+                        await pubsub.close()
+                    except:
+                        pass
+        
+        if retry_count >= max_retries:
+            logger.error(f"❌ Redis Pub/Sub gave up after {max_retries} retries")
 
 # Singleton instance
 redis_pubsub = RedisPubSubService()