images still have bugs

ai_chat.py

@@ -889,13 +889,19 @@ async def handle_image_request(state):
                         print(f"DEBUG - Found user selected option {option_num}: {selected_property_context.get('title')}")
                         break
             
-            # Look for property type mentions that user specifically asked about
-            for i, prop in enumerate(props):
-                title_words = prop.get("title", "").lower().split()
-                for word in ["warehouse", "office", "space", "unit"]:
-                    if word in content and word in title_words:
+            # Look for specific property type mentions that user explicitly asked about
+            # Only if they specifically mentioned the property type in their image request
+            if "warehouse" in content and "warehouse" in user_message:
+                for i, prop in enumerate(props):
+                    if "warehouse" in prop.get("title", "").lower():
+                        selected_property_context = prop
+                        print(f"DEBUG - Found user specifically asking for warehouse images: {prop.get('title')}")
+                        break
+            elif "office" in content and "office" in user_message:
+                for i, prop in enumerate(props):
+                    if "office" in prop.get("title", "").lower():
                         selected_property_context = prop
-                        print(f"DEBUG - Found user interest in {word}: {prop.get('title')}")
+                        print(f"DEBUG - Found user specifically asking for office images: {prop.get('title')}")
                         break
 
     # Use the property the user specifically selected/discussed
@@ -954,66 +960,15 @@ async def handle_image_request(state):
                 selected_property = props[most_mentioned[0]]
                 print(f"DEBUG - Selected property from pronoun context: {selected_property.get('title')}")
     
-    # Fallback: Use conversation context to find which property user was discussing
-    if not selected_property and len(props) > 1:
-        # Check conversation history for property context
-        session_messages = state.get("session_messages", [])
-        print(f"DEBUG - Checking {len(session_messages)} session messages for property context")
-        
-        # Look for property mentions in recent conversation
-        recent_messages = session_messages[-10:]  # Last 10 messages
-        property_mentions = {}
-        
-        for msg in recent_messages:
-            if msg.get("role") == "assistant":
-                content = msg.get("content", "").lower()
-                for i, prop in enumerate(props):
-                    title = prop.get("title", "").lower()
-                    location = prop.get("location", "").lower()
-                    # Check if this property was mentioned in AI response
-                    title_words = title.split()
-                    if len(title_words) >= 2:  # Use first 2 words for matching
-                        key_phrase = " ".join(title_words[:2])
-                        if key_phrase in content or location in content:
-                            property_mentions[i] = property_mentions.get(i, 0) + 1
-                            print(f"DEBUG - Found mention of property {i}: {title}")
+    # If still no property selected and user didn't specify, ask for clarification
+    if not selected_property:
+        print("DEBUG - No clear property context found, asking user to specify")
+        prop_options = []
+        for i, prop in enumerate(props[:3], 1):  # Show first 3 options
+            prop_options.append(f"Option {i}: {prop.get('title')}")
         
-        # Use most mentioned property from conversation
-        if property_mentions:
-            most_mentioned = max(property_mentions.items(), key=lambda x: x[1])
-            selected_property = props[most_mentioned[0]]
-            print(f"DEBUG - Selected property from conversation context: {selected_property.get('title')}")
-        else:
-            # Check if user is asking for images in a general way (like "show me images")
-            # and there's only one property or clear context from recent messages
-            if len(props) == 1:
-                # Only one property available - use it
-                selected_property = props[0]
-                print(f"DEBUG - Only one property available, using: {selected_property.get('title')}")
-            else:
-                # Look for any recent property discussion in the last few messages
-                very_recent_messages = session_messages[-5:]  # Last 5 messages
-                for msg in very_recent_messages:
-                    if msg.get("role") == "assistant":
-                        content = msg.get("content", "").lower()
-                        # Check if any property was mentioned in the last AI response
-                        for i, prop in enumerate(props):
-                            title = prop.get("title", "").lower()
-                            if any(word in content for word in title.split()[:3]):  # Check first 3 words of title
-                                selected_property = prop
-                                print(f"DEBUG - Found recent mention of property: {prop.get('title')}")
-                                break
-                        if selected_property:
-                            break
-                
-                # If still no property selected, only then ask user to specify
-                if not selected_property:
-                    prop_options = []
-                    for i, prop in enumerate(props[:3], 1):  # Show first 3 options
-                        prop_options.append(f"Option {i}: {prop.get('title')}")
-                    
-                    options_text = "\n".join(prop_options)
-                    return [f"I have multiple properties available. Which one would you like to see images of?\n\n{options_text}\n\nPlease let me know which option you'd like images for."]
+        options_text = "\n".join(prop_options)
+        return [f"I have multiple properties available. Which one would you like to see images of?\n\n{options_text}\n\nPlease let me know which option you'd like images for."]
     
     # Final fallback: use first property if only one or no context found
     if not selected_property: