修改bug

galaxy/galaxy/Classes/Modules/Plan(方案)/MakePlan/C/YHMakePlanViewController.swift

@@ -185,6 +185,7 @@ extension YHMakePlanViewController: UITableViewDelegate, UITableViewDataSource {
             self.viewModel.makePlanModel.toggleAddCartStatus(for: product)
             self.updataPrice()
             self.getScore()
+            self.tableView.reloadData()
         }
         return cell
     }

galaxy/galaxy/Classes/Modules/Plan(方案)/MakePlan/M/YHMakePlanModel.swift

@@ -142,27 +142,27 @@ extension YHMakePlanModel {
     }
     
     func calculateTotalPrice() -> String {
-            let allCategories = [basic, work, live, invest]
-            var total: Double = 0
-            
-            for category in allCategories {
-                for mainModel in category {
-                    // Check main product
-                    if mainModel.is_add_cart, let price = Double(mainModel.price) {
+        let allCategories = [basic, work, live, invest]
+        var total: Double = 0
+        
+        for category in allCategories {
+            for mainModel in category {
+                // Check main product
+                if mainModel.is_add_cart, let price = Double(mainModel.price), mainModel.sub_product.count == 0 {
+                    total += price
+                }
+                
+                // Check sub products
+                for subModel in mainModel.sub_product {
+                    if subModel.is_add_cart, let price = Double(subModel.price) {
                         total += price
                     }
-                    
-                    // Check sub products
-                    for subModel in mainModel.sub_product {
-                        if subModel.is_add_cart, let price = Double(subModel.price) {
-                            total += price
-                        }
-                    }
                 }
             }
-            
-            return "\(total)"
         }
+        
+        return "\(total)"
+    }
 }
 
 extension YHMakePlanMainModel {