Enhance image request handling and intent classification in AI chat

- Updated the `chat_with_session_memory` function to clarify the inclusion of property data without image URLs, improving user understanding of available information.
- Modified the `extract_and_update_intent` function to recognize requests for images, allowing the system to handle both property and image inquiries more effectively.
- Adjusted the `process_message` function to check for image requests based on user input, ensuring appropriate responses are generated when users ask for images or photos.
- These changes aim to improve user interaction by providing clearer communication regarding available listings and enhancing the system's ability to respond to image requests.

ai_chat.py

@@ -50,7 +50,7 @@ def chat_with_session_memory(state):
     if search_status:
         system_message += f"\n\n{search_status}"
     
-    # Include property data if available
+    # Include property data if available (without image URLs)
     if props:
         system_message += "\n\nAvailable listings:\n"
         for p in props[:5]:
@@ -58,7 +58,7 @@ def chat_with_session_memory(state):
                 f"- {p.get('title')} in {p.get('location')}, {p.get('city')}: "
                 f"{p.get('size_sqm')} sqm, {p.get('price')} ({p.get('price_type')})\n"
             )
-            # Include all available data for the LLM to use (but not image URLs in listings)
+            # Include available data but NOT image URLs
             if p.get("listing_url"):
                 system_message += f"  URL: {p.get('listing_url')}\n"
             if p.get("features"):
@@ -67,7 +67,7 @@ def chat_with_session_memory(state):
                 system_message += f"  Floorplan: {p.get('floorplan_pdf')}\n"
             if p.get("video_url"):
                 system_message += f"  Video: {p.get('video_url')}\n"
-            # Note: Images are available but not shown in listings - users must ask for them
+            # Images are available but not shown - users must ask for them specifically
     
     system_message += "\n\nIMPORTANT: You may only reference listings passed in state['properties']. If the user requests more detail, respond with whatever is in that listing dict (URL, images, features, etc.). If information is not available, respond with 'For this listing, I don't have [specific detail] available right now'. When users ask for images, photos, or pictures, let them know that images are available and will be sent separately."
     
@@ -269,12 +269,12 @@ async def extract_and_update_intent(state):
     missing = [f for f in intent_fields if state["intent"].get(f) is None]
     print(f"DEBUG - Missing intent fields: {missing}")
     if missing:
-        # Check if user is asking for properties (using AI classification)
+        # Check if user is asking for properties or images (using AI classification)
         classification = state.get("classification")
         print(f"DEBUG - Intent update classification check: '{classification}'")
-        if classification == "search_listings":
-            # User is asking for properties, don't interrupt with preference questions
-            print("DEBUG - User asking for properties, skipping preference questions")
+        if classification in ["search_listings", "request_images"]:
+            # User is asking for properties or images, don't interrupt with preference questions
+            print(f"DEBUG - User asking for {classification}, skipping preference questions")
             return {"response": None}
         
         # User is setting preferences, ask for missing fields
@@ -300,7 +300,8 @@ async def classify_user_intent(state):
     prompt = f"""
 Classify the user's message into exactly one of:
 - search_listings  (user wants to see property listings)
-- request_details  (user wants more info on a shown listing)
+- request_images   (user wants to see images/photos/pictures of a listing)
+- request_details  (user wants other info like features, floorplan, video)
 - other             (anything else)
 Return only the tag.
 Message: {user_message}
@@ -319,7 +320,7 @@ async def extract_and_search_properties(state):
     # Only search when the LLM tagged this as a listings request
     classification = state.get("classification")
     print(f"DEBUG - Property search classification check: '{classification}'")
-    if classification != "search_listings":
+    if classification not in ["search_listings", "request_images"]:
         print(f"DEBUG - Skipping property search, classification is '{classification}'")
         return {"response": None}
 
@@ -426,7 +427,8 @@ async def process_message(user_message: str, user_info: dict = None, session_id:
     
     return {
         "response": result["response"],
-        "properties": result.get("properties", [])
+        "properties": result.get("properties", []),
+        "classification": result.get("classification", "")
     }
 
 async def handle_image_request(state):

main.py

@@ -72,7 +72,12 @@ async def receive_message(req: Request):
                 "properties": properties
             }
             
-            image_messages = await handle_image_request(state)
+            # Check if this is an image request
+            classification = ai_result.get("classification", "")
+            image_messages = None
+            
+            if classification == "request_images" or "image" in user_message.lower() or "photo" in user_message.lower():
+                image_messages = await handle_image_request(state)
             
             if image_messages:
                 # Send images