fyp

app/ai/agent/brain.py

@@ -46,7 +46,21 @@ AVAILABLE_TOOLS = """
 
 1. update_listing(fields: dict)
    - Use this when user provides property details (price, bedrooms, location, etc.)
-   - Fields: title, description, price, bedrooms, bathrooms, location, amenities (list), requirements (list), type (rent/sale)
+   - Fields: title, description, price, bedrooms, bathrooms, location, amenities (list), requirements (list), listing_type
+   
+   LISTING TYPES (4 total):
+   - "rent" = Regular long-term rental (monthly/yearly payment)
+   - "sale" = Property for sale (one-time payment)
+   - "short-stay" = Short-term like Airbnb (daily/weekly/nightly)
+   - "roommate" = Someone ALREADY living in a place wants to find a person to SHARE the space with
+   
+   ROOMMATE EXPLAINED: If user says "I want to share my apartment", "looking for someone to split rent", 
+   "need a roommate", "want to find someone to live with" → listing_type = "roommate"
+   
+   ROLE-BASED PERMISSIONS:
+   - Landlord: Can list ALL types (rent, sale, short-stay, roommate)
+   - Renter: Can ONLY list "roommate" type. If renter tries to list rent/sale/short-stay, 
+     politely explain they can only list roommate listings.
 
 2. search_properties(query: str, location: str, min_price: int, max_price: int, beds: int)
    - Use this when user wants to FIND/SEARCH for properties

app/ai/agent/nodes/listing_collect.py

@@ -689,6 +689,37 @@ Just return the message, no quotes."""
         state.temp_data["response_text"] = smart_response.get("response_text")
         state.temp_data["action"] = action
         
+        # ✅ NEW: Generate draft_ui for NEW listings if we have enough data to display
+        # This ensures users see a visual card, not just text, when providing images/fields
+        has_images = len(state.provided_fields.get("images", [])) > 0
+        has_key_fields = any([
+            state.provided_fields.get("location"),
+            state.provided_fields.get("bedrooms"),
+            state.provided_fields.get("price"),
+        ])
+        
+        if has_images or has_key_fields:
+            from app.ai.agent.nodes.listing_validate import build_draft_ui_from_dict
+            draft_data = {
+                "title": state.provided_fields.get("title", "New Listing"),
+                "description": state.provided_fields.get("description", ""),
+                "location": state.provided_fields.get("location", ""),
+                "address": state.provided_fields.get("address", ""),
+                "bedrooms": state.provided_fields.get("bedrooms", 0),
+                "bathrooms": state.provided_fields.get("bathrooms", 0),
+                "price": state.provided_fields.get("price", 0),
+                "currency": state.provided_fields.get("currency", ""),
+                "price_type": state.provided_fields.get("price_type", "monthly"),
+                "listing_type": state.provided_fields.get("listing_type", "rent"),
+                "amenities": state.provided_fields.get("amenities", []),
+                "images": state.provided_fields.get("images", []),
+                "requirements": state.provided_fields.get("requirements", []),
+            }
+            draft_ui = build_draft_ui_from_dict(draft_data)
+            draft_ui["status"] = "draft"  # Mark as draft (not editing existing)
+            state.temp_data["draft_ui"] = draft_ui
+            logger.info("Draft UI generated for new listing", has_images=has_images, image_count=len(draft_data.get("images", [])))
+        
         logger.info("Continuing collection", 
                    action=action,
                    missing_count=len(missing_required))

app/ai/tools/listing_tool.py

@@ -415,15 +415,12 @@ async def auto_detect_listing_type(price_type: str, user_role: str, user_message
     if user_role in ["renter", "roommate"]:
         return "roommate"
         
-    # If LLM explicitly extracted a valid type, trust it (but validate)
+    # If LLM explicitly extracted a valid type, trust it
     if extracted_type:
         valid_types = ["rent", "sale", "short-stay", "roommate"]
         if extracted_type in valid_types:
-            # Landlords can't list roommates usually, but if explicitly said...
-            if extracted_type == "roommate" and user_role == "landlord":
-                 pass # Allow or block? Let's strictly block for now
-            else:
-                return extracted_type
+            # Landlords can list ALL types including roommate
+            return extracted_type
     
     # LANDLORD: Detect from price_type as backup
     price_type_lower = (price_type or "").lower().strip()
@@ -470,21 +467,26 @@ def validate_listing_type_for_role(listing_type: str, user_role: str) -> tuple[b
     """
     Validate if user can create this listing type.
     
+    Permissions:
+    - Landlord: Can list ALL types (rent, sale, short-stay, roommate)
+    - Renter: Can ONLY list roommate
+    
     Returns:
         (is_valid, error_message)
     """
+    valid_types = ["rent", "sale", "short-stay", "roommate"]
+    
+    if listing_type not in valid_types:
+        return False, f"Invalid listing type: {listing_type}. Please choose rent, sale, short-stay, or roommate."
     
     if user_role == "renter":
         if listing_type != "roommate":
-            return False, "As a renter, you can only create roommate listings. Would you like to list a room for a roommate instead?"
+            return False, "As a renter, you can only create roommate listings. Roommate listings are for finding someone to share your living space with. Would you like to list a room for a roommate instead?"
         return True, ""
     
+    # Landlord can list ANY type including roommate
     if user_role == "landlord":
-        if listing_type == "roommate":
-            return False, "As a landlord, you cannot create roommate listings. You can list properties for rent, short-stay, or sale."
-        if listing_type in ["rent", "short-stay", "sale"]:
-            return True, ""
-        return False, f"Invalid listing type: {listing_type}. Please choose rent, short-stay, or sale."
+        return True, ""
     
     # Default: allow
     return True, ""