diff --git a/lib/features/chat/widgets/modern_chat_input.dart b/lib/features/chat/widgets/modern_chat_input.dart
index 4ad579d..671a15b 100644
--- a/lib/features/chat/widgets/modern_chat_input.dart
+++ b/lib/features/chat/widgets/modern_chat_input.dart
@@ -378,7 +378,8 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
                                       ref
                                           .watch(selectedToolIdsProvider)
                                           .isNotEmpty ||
-                                      ref.watch(webSearchEnabledProvider),
+                                      ref.watch(webSearchEnabledProvider) ||
+                                      ref.watch(imageGenerationEnabledProvider),
                                 ),
                                 const Spacer(),
                                 // Microphone button: call provided callback for premium voice UI