enhance UI a bit

Author: daltonnyx

AgentCrew/modules/chat/message/handler.py

@@ -348,6 +348,8 @@ async def get_assistant_response(
 
             return assistant_response, input_tokens, output_tokens
 
+        except GeneratorExit as e:
+            return assistant_response, input_tokens, output_tokens
         except Exception as e:
             if self.current_user_input:
                 self.conversation_manager.store_conversation_turn(

AgentCrew/modules/gui/components/chat_components.py

@@ -33,7 +33,8 @@ def _setup_chat_area(self):
         self.chat_window.chat_container.setObjectName("chatContainer")
         self.chat_window.chat_layout = QVBoxLayout(self.chat_window.chat_container)
         self.chat_window.chat_layout.setAlignment(Qt.AlignmentFlag.AlignTop)
-        self.chat_window.chat_layout.setSpacing(10)
+        self.chat_window.chat_layout.setSpacing(15)
+        self.chat_window.chat_layout.setContentsMargins(5, 5, 10, 10)
 
         # Create a scroll area for messages
         self.chat_window.chat_scroll = QScrollArea()

AgentCrew/modules/gui/components/input_components.py

@@ -7,6 +7,7 @@
     QPushButton,
     QCompleter,
     QFileDialog,
+    QSizePolicy,
 )
 from PySide6.QtCore import Qt, QStringListModel
 from PySide6.QtGui import QTextCursor
@@ -41,6 +42,9 @@ def _setup_input_area(self):
         self.chat_window.message_input.setStyleSheet(
             self.chat_window.style_provider.get_input_style()
         )
+        self.chat_window.message_input.setSizePolicy(
+            QSizePolicy.Policy.Minimum, QSizePolicy.Policy.Expanding
+        )
 
         # Create buttons layout
         buttons_layout = QVBoxLayout()

AgentCrew/modules/gui/components/ui_state_manager.py

@@ -81,7 +81,7 @@ def _set_send_button_state(self, is_stop_stated: bool = False):
             self.chat_window.send_button.setStyleSheet(
                 self.chat_window.style_provider.get_button_style("stop")
             )
-            self.animation_timer.setInterval(150)  # Update every 150ms for spinner
+            self.animation_timer.setInterval(80)  # Update every 150ms for spinner
             self.animation_timer.start()
 
             # Change the button to stop functionality

AgentCrew/modules/gui/widgets/message_bubble.py

@@ -140,7 +140,7 @@ def __init__(
             self.set_text(text)
 
         # self.message_label.setSizePolicy(
-        #     QSizePolicy.Policy.Maximum, QSizePolicy.Policy.Minimum
+        #     QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Minimum
         # )
 
         if text is not None:
@@ -314,7 +314,7 @@ def add_streaming_chunk(self, chunk_queue: list):
 
         # Start the streaming timer if not active
         if not self.streaming_timer.isActive():
-            self.streaming_timer.start(24)
+            self.streaming_timer.start(40)
 
         # Add characters to queue for smooth rendering
         self.character_queue = chunk_queue
@@ -328,11 +328,11 @@ def _render_next_character(self):
 
         # Adaptive rendering speed based on queue size
         if len(self.character_queue) > 100:
-            chars_per_frame = 21  # Speed up for large queues
+            chars_per_frame = 40  # Speed up for large queues
         elif len(self.character_queue) > 50:
-            chars_per_frame = 15
+            chars_per_frame = 30
         else:
-            chars_per_frame = 10  # Slower for natural effect
+            chars_per_frame = 20  # Slower for natural effect
 
         # Render characters for this frame
         new_chars = ""