// AI

galaxy/galaxy/Classes/Modules/AI/AI-Dialogue(AI对话)/C/YHAIRobotChatViewController.swift

@@ -84,12 +84,7 @@ class YHAIRobotChatViewController: YHBaseViewController {
         let bgImgHeight = 242.0/335.0 * (KScreenWidth-40.0)
         let height = 360.0-242.0+bgImgHeight
         let view = YHAIChatBannerView(frame: CGRectMake(0, 0, KScreenWidth, height))
-        view.bgImgV.image = getBannerBg()
-        view.bgCardView.backgroundColor = getBannerBgColor()
-        view.titleLabel.text = getHeaderTitle()
-        view.descLabel.text = getHeaderDesc()
-        view.bannerArr = self.getBannerForRobotType(robotType)
-        view.messages = getFlowMessages()
+        view.config = getBannerConfig()
         view.selectFlowMsgBlock = {
             [weak self] text in
             guard let self = self else { return }
@@ -318,75 +313,33 @@ class YHAIRobotChatViewController: YHBaseViewController {
         }
     }
     
-    func getBannerBg() -> UIImage? {
-        if robotType == YHAIRobotType.education.rawValue {
-            return UIImage(named: "ai_chat_header_bg_1")
+    func getBannerConfig() -> YHAIChatBannerViewConfig {
         
-        } else if robotType == YHAIRobotType.sale.rawValue {
-            return UIImage(named: "ai_chat_header_bg_0")
-        }
-        return UIImage(named: "ai_chat_header_bg_0")
-    }
-    
-    func getBannerBgColor() -> UIColor {
-        if robotType == YHAIRobotType.education.rawValue {
-            return .init(hex: 0xDAF6FC)
-            
-
-        } else if robotType == YHAIRobotType.sale.rawValue {
-            return .init(hex: 0xE6F4FF)
-        }
-        return .init(hex: 0xE6F4FF)
-    }
-    
-    func getFlowMessages() -> [String] {
-        
-        if robotType == YHAIRobotType.education.rawValue {
-            return  ["香港教育有哪些优势？", "去香港读书有哪些条件？", "申请香港学校费用是多少？", "了解银河教育插班服务流程", "了解银河教育插班录取率"]
-
-        } else if robotType == YHAIRobotType.sale.rawValue {
-            return  ["优才学历加分要求是什么？", "优才现在要怎么申请？", "墨尔本大学是优才合资格大学吗？", "推荐一些优才产品"]
-
-        }
-        return []
-    }
-    
-    func getHeaderTitle() -> String {
-        
-        if robotType == YHAIRobotType.education.rawValue {
-            return "Hello，我是香港教育宝"
-
-        } else if robotType == YHAIRobotType.sale.rawValue {
-            return  "Hello，我是新港生活规划师"
-        }
-        return ""
-    }
-    
-    func getHeaderDesc() -> String {
-        
-        if robotType == YHAIRobotType.education.rawValue {
-            return "有香港教育的问题尽管问我"
-
-        } else if robotType == YHAIRobotType.sale.rawValue {
-            return  "香港身份办理问题可以找我"
-        }
-        return ""
-    }
-    
-    func getBannerForRobotType(_ robotType: String) -> [YHAIChatBannerItem] {
+        let config = YHAIChatBannerViewConfig()
         
         if robotType == YHAIRobotType.sale.rawValue {
-            return  [YHAIChatBannerItem(id: 0, title: "了解银河集团", desc: "香港身份生活一站式服务平台", msg: "银河集团，能够为我提供什么？"),
+            config.title = "Hello，我是新港生活规划师"
+            config.desc = "香港身份办理问题可以找我"
+            config.bgColor = .init(hex: 0xE6F4FF)
+            config.indicatorColor = .brandMainColor
+            config.bgImageName = "ai_chat_header_bg_0"
+            config.bannerItems =  [YHAIChatBannerItem(id: 0, title: "了解银河集团", desc: "香港身份生活一站式服务平台", msg: "银河集团，能够为我提供什么？"),
                                    YHAIChatBannerItem(id: 1, title: "香港身份智能评估", desc: "20s快速评估，了解自身条件是否符合", msg: "开始身份办理评估"),
                                    YHAIChatBannerItem(id: 2, title: "银河产品矩阵", desc: "香港身份、生活多样产品", msg: "介绍一下银河的产品"),]
-           
-        }
-        if robotType == YHAIRobotType.education.rawValue {
-            return  [YHAIChatBannerItem(id: 0, title: "幼中小学升学", desc: "去香港插班需要考核哪些"),
+            config.flowMessages = ["优才学历加分要求是什么？", "优才现在要怎么申请？", "墨尔本大学是优才合资格大学吗？", "推荐一些优才产品"]
+            
+        } else if robotType == YHAIRobotType.education.rawValue {
+            config.title = "Hello，我是香港教育宝"
+            config.desc = "有香港教育的问题尽管问我"
+            config.bgColor = .init(hex: 0xDAF6FC)
+            config.indicatorColor = .init(hex: 0x00C77C)
+            config.bgImageName = "ai_chat_header_bg_1"
+            config.bannerItems =  [YHAIChatBannerItem(id: 0, title: "幼中小学升学", desc: "去香港插班需要考核哪些"),
                                    YHAIChatBannerItem(id: 1, title: "大学升学", desc: "DSE分数和Alevel的换算关系"),
                                    YHAIChatBannerItem(id: 2, title: "银河教育服务", desc: "银河教育插班成功率如何？"),]
+            config.flowMessages = ["香港教育有哪些优势？", "去香港读书有哪些条件？", "申请香港学校费用是多少？", "了解银河教育插班服务流程", "了解银河教育插班录取率"]
         }
-        return []
+        return config
     }
     
     func isNeedStopResonse() -> Bool {

galaxy/galaxy/Classes/Modules/AI/AI-Dialogue(AI对话)/V/YHAIChatBannerView.swift

@@ -23,6 +23,17 @@ class YHAIChatBannerItem {
     }
 }
 
+class YHAIChatBannerViewConfig {
+    var bgColor: UIColor = .init(hex: 0xE6F4FF)
+    var bgImageName: String = ""
+    var title: String = ""
+    var desc: String = ""
+    var indicatorColor: UIColor = .brandMainColor
+    var bannerItems:[YHAIChatBannerItem] = []
+    var flowMessages:[String] = []
+}
+
+
 class YHAIChatBannerView: UIView {
 
     static let bannersHeight = 95.0
@@ -30,6 +41,18 @@ class YHAIChatBannerView: UIView {
     var selectFlowMsgBlock:((String)->())?
     var selectBannerItemBlock:((YHAIChatBannerItem)->())?
     
+    var config = YHAIChatBannerViewConfig() {
+        didSet {
+            titleLabel.text = config.title
+            descLabel.text = config.desc
+            bgCardView.backgroundColor = config.bgColor
+            bgImgV.image = UIImage(named: config.bgImageName)
+            indicatorView.selectedColor = config.indicatorColor
+            bannerArr = config.bannerItems
+            messages = config.flowMessages
+        }
+    }
+
     var messages:[String] = [] {
         didSet {
             layout.dataSource = messages