Enhance property search functionality by adding transaction type filtering

- Updated the extract_and_search_properties function to include transaction_type in the search filters, allowing users to specify whether they are looking to rent or buy properties.
- Modified the search_properties function to handle transaction_type filtering, ensuring appropriate price types are applied based on the user's selection.
- Added sample sale properties to the database setup for testing purposes, improving the robustness of property search functionality.

ai_chat.py

@@ -517,6 +517,10 @@ async def extract_and_search_properties(state):
     if intent.get("size_preference_sqm") is not None:
         filters["size_preference_sqm"] = intent["size_preference_sqm"]
     
+    # Add transaction type if set (rent vs buy)
+    if intent.get("transaction_type") is not None:
+        filters["transaction_type"] = intent["transaction_type"]
+    
     # Search for properties with flexible ranges
     print(f"DEBUG - Searching with filters: {filters}")
     properties = await search_properties(filters)

database.py

@@ -327,6 +327,21 @@ async def search_properties(filters: dict) -> list:
             max_size = int(size * 1.5)  # 50% above target
             query = query.gte("size_sqm", min_size).lte("size_sqm", max_size)
         
+        # d. Transaction type (rent vs buy)
+        transaction_type = filters.get("transaction_type")
+        if transaction_type:
+            print(f"DEBUG - Filtering by transaction_type: {transaction_type}")
+            if transaction_type == "rent":
+                # For rent: filter by monthly, weekly, daily price types
+                query = query.in_("price_type", ["monthly", "weekly", "daily"])
+                print(f"DEBUG - Filtering for rental properties (monthly, weekly, daily)")
+            elif transaction_type == "buy":
+                # For buy: filter by sale, purchase price types
+                query = query.in_("price_type", ["sale", "purchase", "total"])
+                print(f"DEBUG - Filtering for sale properties (sale, purchase, total)")
+        else:
+            print(f"DEBUG - No transaction_type filter applied")
+        
         # Note: Must-have features are removed from database search
         # They will be handled by the LLM in the chat response
         

supabase_setup.sql

@@ -224,4 +224,52 @@ INSERT INTO properties (title, description, location, city, address, size_sqm, p
     TRUE,
     FALSE
 )
-ON CONFLICT DO NOTHING; 
+ON CONFLICT DO NOTHING; 
+
+-- Add some sample sale properties for testing
+INSERT INTO properties (title, description, location, city, address, size_sqm, price, price_type, listing_url, images, features, is_active, is_featured) VALUES
+(
+    'Office Building for Sale in Sandton',
+    'Modern office building with excellent investment potential',
+    'Sandton',
+    'Johannesburg', 
+    '1 Sandton Drive, Sandton, 2196, South Africa',
+    500,
+    2500000,
+    'sale',
+    'https://example.com/office-building-sale',
+    ARRAY['https://example.com/building1.jpg', 'https://example.com/building2.jpg'],
+    ARRAY['Elevator', 'Parking', 'Security', 'Reception'],
+    TRUE,
+    TRUE
+),
+(
+    'Warehouse for Sale in Midrand',
+    'Industrial warehouse with great location for logistics',
+    'Midrand',
+    'Johannesburg',
+    'New Road, Midrand, 1685, South Africa', 
+    1000,
+    3200000,
+    'sale',
+    'https://example.com/warehouse-sale',
+    ARRAY['https://example.com/warehouse-sale1.jpg'],
+    ARRAY['Loading dock', 'High ceiling', '3 phase power', 'Truck access'],
+    TRUE,
+    FALSE
+),
+(
+    'Commercial Property for Sale in Cape Town',
+    'Prime commercial property in Cape Town CBD',
+    'CBD',
+    'Cape Town',
+    'Long Street, Cape Town, 8001, South Africa',
+    300,
+    1800000,
+    'sale', 
+    'https://example.com/commercial-sale-ct',
+    ARRAY['https://example.com/commercial1.jpg'],
+    ARRAY['Street frontage', 'High foot traffic', 'Parking'],
+    TRUE,
+    FALSE
+);