From ca3ea0630ed7aac6a3a74a85f754cf5440d718eb Mon Sep 17 00:00:00 2001
From: CodeST <694468528@qq.com>
Date: Fri, 5 Dec 2025 21:23:54 +0800
Subject: [PATCH] 2

---
 CustomKeyboard/View/KBFunctionView.m | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/CustomKeyboard/View/KBFunctionView.m b/CustomKeyboard/View/KBFunctionView.m
index 44cc14a..3d89d40 100644
--- a/CustomKeyboard/View/KBFunctionView.m
+++ b/CustomKeyboard/View/KBFunctionView.m
@@ -204,14 +204,18 @@
     overlay.delegate = (id)self;
     [self addSubview:overlay];
     [overlay mas_makeConstraints:^(MASConstraintMaker *make) {
-        make.left.equalTo(self.mas_left).offset(12);
-        // 流式期间占满右侧宽度：不再贴 rightButtonContainer，让文本有最大宽度
-        make.right.equalTo(self.mas_right).offset(-12);
+        // 在原标签区域内展示流式文本，右侧继续保留竖排按钮栏
+        CGFloat vSpace = 4.0;
+        make.left.equalTo(self.pasteViewInternal);
+        make.right.equalTo(self).offset(-vSpace);
         make.top.equalTo(self.pasteViewInternal.mas_bottom).offset(10);
         make.bottom.equalTo(self.mas_bottom).offset(-10);
     }];
-    // 隐藏右侧按钮栏，最大化文本宽度；关闭 overlay 时再恢复
-    self.rightButtonContainer.hidden = YES;
+    // 仅隐藏删除/清空/发送按钮，保留“Paste”按钮可用
+    self.pasteButtonInternal.hidden = NO;
+    self.deleteButtonInternal.hidden = YES;
+    self.clearButtonInternal.hidden = YES;
+    self.sendButtonInternal.hidden = YES;
     // 适当缩小内部左右留白，进一步提升可用宽度
     overlay.textView.contentHorizontalPadding = 8.0;
     self.streamOverlay = overlay;
@@ -225,8 +229,11 @@
     [self.streamOverlay removeFromSuperview];
     self.streamOverlay = nil;
     self.tagListView.hidden = NO;
-    // 恢复右侧按钮栏
-    self.rightButtonContainer.hidden = NO;
+    // 恢复右侧按钮栏的全部按钮
+    self.pasteButtonInternal.hidden = NO;
+    self.deleteButtonInternal.hidden = NO;
+    self.clearButtonInternal.hidden = NO;
+    self.sendButtonInternal.hidden = NO;
 }
 
 // 叠层关闭回调