Sửa để animation chạy liên tục cho đến khi có response từ API hoặc timeout

app.py

@@ -537,14 +537,14 @@ class UIService:
     
     @staticmethod
     async def create_typing_animation():
-        """Create typing animation effect"""
+        """Create typing animation effect (legacy method - kept for compatibility)"""
         msg = cl.Message(content="", author="assistant")
         await msg.send()
         
         # Typing animation frames
         typing_frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
         
-        for i in range(40):  # Show animation for ~10 seconds
+        for i in range(27):  # Show animation for ~2 seconds
             frame = typing_frames[i % len(typing_frames)]
             msg.content = f"{frame} Đang suy nghĩ..."
             await msg.update()
@@ -553,6 +553,25 @@ class UIService:
         return msg
 
 
+async def run_typing_animation(msg: cl.Message):
+    """Run typing animation until cancelled"""
+    typing_frames = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"]
+    frame_index = 0
+    
+    try:
+        while True:  # Run indefinitely until cancelled
+            frame = typing_frames[frame_index % len(typing_frames)]
+            msg.content = f"{frame} Đang suy nghĩ..."
+            await msg.update()
+            await asyncio.sleep(0.25)
+            frame_index += 1
+            
+    except asyncio.CancelledError:
+        # Animation was cancelled, this is expected
+        print("🎬 Animation cancelled - API response received")
+        raise
+
+
 # HELPER FUNCTIONS: Session management with proper async error handling
 async def ensure_session_state() -> Optional[ConversationState]:
     """Ensure session state exists, create if not"""
@@ -803,7 +822,7 @@ async def on_debug_sessions(action):
 
 @cl.on_message
 async def main(message: cl.Message):
-    """Main message handler with proper async flow"""
+    """Main message handler with concurrent animation and API call"""
     app_state = await ensure_session_state()
     if app_state is None:
         await cl.Message(content="Error: Session state not found", author="assistant").send()
@@ -817,14 +836,37 @@ async def main(message: cl.Message):
                 image_path = element.path
                 break
     
-    # Show typing animation
-    typing_msg = await UIService.create_typing_animation()
+    # Create initial message for animation
+    msg = cl.Message(content="", author="assistant")
+    await msg.send()
     
-    # Get response from API - now fully async, won't block event loop
-    response = await ChatService.respond_to_chat(app_state, message.content, image_path)
+    # Create concurrent tasks for animation and API call
+    animation_task = asyncio.create_task(run_typing_animation(msg))
+    api_task = asyncio.create_task(ChatService.respond_to_chat(app_state, message.content, image_path))
+    
+    try:
+        # Wait for API response (this will complete first usually)
+        response = await api_task
+        
+        # Cancel animation task since we have the response
+        animation_task.cancel()
+        
+        # Wait a bit for graceful animation cancellation
+        try:
+            await asyncio.wait_for(animation_task, timeout=0.1)
+        except (asyncio.CancelledError, asyncio.TimeoutError):
+            pass
+        
+    except Exception as e:
+        # If API fails, cancel animation and show error
+        animation_task.cancel()
+        try:
+            await asyncio.wait_for(animation_task, timeout=0.1)
+        except (asyncio.CancelledError, asyncio.TimeoutError):
+            pass
+        response = f"Error: {e}"
     
-    # Update the typing message with final response and buttons
-    typing_msg.content = response
-    typing_msg.actions = UIService.create_action_buttons(app_state)
-    typing_msg.author = "assistant"
-    await typing_msg.update()
+    # Update message with final response and buttons
+    msg.content = response
+    msg.actions = UIService.create_action_buttons(app_state)
+    await msg.update()