From d549d5ecabb21c9bfdb8d55cf9b8dc3fa9f8098f Mon Sep 17 00:00:00 2001
From: Paul Gauthier <aider@paulg.org>
Date: Fri, 26 Apr 2024 14:43:27 -0700
Subject: [PATCH] cleanup

---
 aider/gui.py | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/aider/gui.py b/aider/gui.py
index d59168213..f0121f21a 100755
--- a/aider/gui.py
+++ b/aider/gui.py
@@ -235,21 +235,18 @@ class GUI:
         self.do_messages_container()
 
         self.prompt = st.chat_input("Say something", on_submit=self.clear_controls)
-        self.chat()
 
-    def chat(self):
+        if self.prompt:
+            self.chat(self.prompt)
+            return
+
         if self.old_prompt:
             prompt = self.old_prompt
             st.session_state.recent_msgs_num += 1
             self.reset_recent_msgs()
-        elif self.prompt:
-            prompt = self.prompt
-        else:
-            prompt = None
-
-        if not prompt:
-            return
+            self.chat(prompt)
 
+    def chat(self, prompt):
         self.clear_controls()
 
         st.session_state.messages.append({"role": "user", "content": prompt})