Update chat_handler.py

chat_handler.py

@@ -335,12 +335,25 @@ class ChatHandler:
                         }
                     ))
                     
-                    # Add clean media display only if we extracted a valid URL and it's different from raw result
-                    if media_url and media_url not in result_text:
-                        chat_messages.append(ChatMessage(
-                            role="assistant", 
-                            content={"path": media_url}
-                        ))
+                    # Only add separate media display if:
+                    # 1. We extracted a valid URL 
+                    # 2. The result text is NOT just a URL (to avoid double rendering)
+                    # 3. The result text doesn't contain the exact URL
+                    if media_url:
+                        # Check if the result is essentially just the URL (with possible whitespace)
+                        cleaned_result = result_text.strip()
+                        is_just_url = (cleaned_result == media_url or 
+                                     cleaned_result.startswith('http') and len(cleaned_result.split()) == 1)
+                        
+                        # Only add separate media if the result is more than just the URL
+                        if not is_just_url and media_url not in result_text:
+                            logger.info(f"🎯 Adding separate media display for: {media_url}")
+                            chat_messages.append(ChatMessage(
+                                role="assistant", 
+                                content={"path": media_url}
+                            ))
+                        else:
+                            logger.info(f"🚫 Skipping separate media - result is just URL: {is_just_url}, contains URL: {media_url in result_text}")
                         
                 else:
                     # Add error message for failed tool call