From ede3cbb372612e7bbf75352cf5eecca8c59ae3fd Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Fri, 27 Sep 2024 19:47:40 -0700
Subject: [PATCH] refactor: simplify _get_style method in InputOutput class

---
 aider/io.py | 34 ++++++++++++++++++++--------------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/aider/io.py b/aider/io.py
index 3dc0c352d..6583737ff 100644
--- a/aider/io.py
+++ b/aider/io.py
@@ -251,26 +251,32 @@ class InputOutput:
         else:
             self.console = Console(force_terminal=False, no_color=True)  # non-pretty
 
-    def _get_style(self, style_dict=None):
-        if style_dict is None:
-            style_dict = {}
-        if self.pretty and self.user_input_color:
+    def _get_style(self):
+        style_dict = {}
+        if not self.pretty:
+            return Style.from_dict(style_dict)
+
+        if self.user_input_color:
             style_dict.setdefault("", self.user_input_color)
             style_dict.update(
                 {
                     "pygments.literal.string": f"bold italic {self.user_input_color}",
-                    "completion-menu": (
-                        f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}"
-                    ),
-                    "completion-menu.completion.current": (
-                        f"bg:{self.completion_menu_current_bg_color} "
-                        f"{self.completion_menu_current_color}"
-                    ),
                 }
             )
-            return Style.from_dict(style_dict)
-        else:
-            return {}
+
+        style_dict.update(
+            {
+                "completion-menu": (
+                    f"bg:{self.completion_menu_bg_color} {self.completion_menu_color}"
+                ),
+                "completion-menu.completion.current": (
+                    f"bg:{self.completion_menu_current_bg_color} "
+                    f"{self.completion_menu_current_color}"
+                ),
+            }
+        )
+
+        return Style.from_dict(style_dict)
 
     def read_image(self, filename):
         try: