diff --git a/assets/gemini.png b/assets/gemini_icon.png
similarity index 100%
rename from assets/gemini.png
rename to assets/gemini_icon.png
diff --git a/gemini-input-glow.png b/gemini-input-glow.png
deleted file mode 100644
index 42c3afc4..00000000
Binary files a/gemini-input-glow.png and /dev/null differ
diff --git a/lib/screens/tasks/task_detail_screen.dart b/lib/screens/tasks/task_detail_screen.dart
index 9d9a4b42..19d2c937 100644
--- a/lib/screens/tasks/task_detail_screen.dart
+++ b/lib/screens/tasks/task_detail_screen.dart
@@ -3763,11 +3763,7 @@ class _TaskDetailScreenState extends ConsumerState<TaskDetailScreen>
     final plainText = _actionController?.document.toPlainText().trim() ?? '';
     if (plainText.isEmpty) {
       if (!context.mounted) return;
-      ScaffoldMessenger.of(context).showSnackBar(
-        const SnackBar(
-          content: Text('Please enter some action taken text first'),
-        ),
-      );
+      showWarningSnackBar(context, 'Please enter some action taken text first');
       return;
     }
 
@@ -3829,15 +3825,11 @@ class _TaskDetailScreenState extends ConsumerState<TaskDetailScreen>
       }
 
       if (context.mounted) {
-        ScaffoldMessenger.of(context).showSnackBar(
-          const SnackBar(content: Text('Action taken improved successfully')),
-        );
+        showSuccessSnackBar(context, 'Action taken improved successfully');
       }
     } catch (e) {
       if (context.mounted) {
-        ScaffoldMessenger.of(
-          context,
-        ).showSnackBar(SnackBar(content: Text('Error: $e')));
+        showErrorSnackBar(context, 'Error: $e');
       }
     } finally {
       if (mounted) {
diff --git a/lib/widgets/gemini_button.dart b/lib/widgets/gemini_button.dart
index be970a9c..e4054035 100644
--- a/lib/widgets/gemini_button.dart
+++ b/lib/widgets/gemini_button.dart
@@ -1,5 +1,6 @@
 import 'package:flutter/material.dart';
 import '../services/ai_service.dart';
+import '../utils/snackbar.dart';
 
 typedef TextUpdateCallback = void Function(String updatedText);
 typedef ProcessingStateCallback = void Function(bool isProcessing);
@@ -52,9 +53,7 @@ class _GeminiButtonState extends State<GeminiButton> {
     final text = widget.textController.text.trim();
     if (text.isEmpty) {
       if (!context.mounted) return;
-      ScaffoldMessenger.of(context).showSnackBar(
-        const SnackBar(content: Text('Please enter some text first')),
-      );
+      showWarningSnackBar(context, 'Please enter some text first');
       return;
     }
 
@@ -78,14 +77,10 @@ class _GeminiButtonState extends State<GeminiButton> {
       widget.textController.text = trimmed;
       widget.onTextUpdated(trimmed);
       if (!context.mounted) return;
-      ScaffoldMessenger.of(context).showSnackBar(
-        const SnackBar(content: Text('Text improved successfully')),
-      );
+      showSuccessSnackBar(context, 'Text improved successfully');
     } catch (e) {
       if (!context.mounted) return;
-      ScaffoldMessenger.of(
-        context,
-      ).showSnackBar(SnackBar(content: Text('Error: $e')));
+      showErrorSnackBar(context, 'Error: $e');
     } finally {
       if (mounted) {
         setState(() {