AI - 重写与纠错状态切换细节调整

PDF Office/PDF Master/Class/Home/ViewController/KMAIRewritingVC.swift

@@ -247,15 +247,15 @@ class KMAIRewritingVC: NSViewController {
                 if pasteboard.string(forType: .string) != nil {
                     if pasteboard.string(forType: .string)!.count > 0 {
                         if pasteboard.string(forType: .string)! == self.aiRewritingResultStr {
-                            self.uiState = .CanCopy
-                        } else {
                             self.uiState = .Copyed
+                        } else {
+                            self.uiState = .CanCopy
                         }
                     } else {
-                        self.uiState = .Copyed
+                        self.uiState = .CanCopy
                     }
                 } else {
-                    self.uiState = .Copyed
+                    self.uiState = .CanCopy
                 }
             } else if (self.aiRewritingStr.count > 0) {
                 if self.aiRewritingStr.count > 150 {
@@ -273,14 +273,18 @@ class KMAIRewritingVC: NSViewController {
         } else if self.state == .AIErrorCorrection {
             if (self.aiErrorCorrectionStr.count > 0) && (self.aiErrorCorrectionResultStr.count > 0) {
                 let pasteboard = NSPasteboard.general
-                if pasteboard.string(forType: .string)!.count > 0 {
-                    if pasteboard.string(forType: .string)! == self.aiErrorCorrectionResultStr {
-                        self.uiState = .CanCopy
+                if pasteboard.string(forType: .string) != nil {
+                    if pasteboard.string(forType: .string)!.count > 0 {
+                        if pasteboard.string(forType: .string)! == self.aiErrorCorrectionResultStr {
+                            self.uiState = .Copyed
+                        } else {
+                            self.uiState = .CanCopy
+                        }
                     } else {
-                        self.uiState = .Copyed
+                        self.uiState = .CanCopy
                     }
                 } else {
-                    self.uiState = .Copyed
+                    self.uiState = .CanCopy
                 }
             } else if (self.aiErrorCorrectionStr.count > 0) {
                 if self.aiErrorCorrectionStr.count > 150 {
@@ -428,8 +432,12 @@ extension KMAIRewritingVC: NSTextFieldDelegate {
         
         if self.state == .AIRewriting {
             self.aiRewritingStr = object.stringValue
+            self.aiRewritingResultStr = ""
+            self.resultTextView.string = ""
         } else if self.state == .AIErrorCorrection {
             self.aiErrorCorrectionStr = object.stringValue
+            self.aiErrorCorrectionResultStr = ""
+            self.resultTextView.string = ""
         }
     }